[Screenshot] fix finding geckodriver

japandotorg · Sep 19, 2024 · dff1740 · dff1740
1 parent 86cbb30
commit dff1740
Showing 1 changed file with 18 additions and 2 deletions.
diff --git a/screenshot/common/downloader.py b/screenshot/common/downloader.py
@@ -75,8 +75,24 @@ def location(self) -> Optional[pathlib.Path]:
         return (
             loc[0]
             if (
-                loc := list(self.data_directory.glob("geckodriver-{}*".format(self.get_os())))
-                or (loc := list(self.data_directory.glob("geckodrive-{}*".format(self.get_os()))))
+                loc := list(
+                    self.data_directory.glob(
+                        "geckodriver-{}*".format(
+                            "linux-aarch64" if platform.machine() == "aarch64" else self.get_os()
+                        )
+                    )
+                )
+                or (
+                    loc := list(
+                        self.data_directory.glob(
+                            "geckodrive-{}*".format(
+                                "linux-aarch64"
+                                if platform.machine() == "aarch64"
+                                else self.get_os()
+                            )
+                        )
+                    )
+                )
             )
             else None
         )