Switch to firefox driver

Dockerfile

@@ -1,12 +1,21 @@
 # https://github.com/joyzoursky/docker-python-chromedriver
-FROM joyzoursky/python-chromedriver:3.9
+# https://github.com/TomRoush/python-selenium-firefox-docker
+# FROM tomroush/python-selenium-firefox-docker:latest
+FROM ubuntu:20.04
 
-ENV SELENIUM_HEADLESS=true
+ARG DEBIAN_FRONTEND=noninteractive
+RUN apt-get update && apt-get install -y \
+  python3 python3-pip \
+  firefox-geckodriver \
+  xvfb \
+  && rm -rf /var/lib/apt/lists/*
+
+RUN pip3 install --no-cache-dir selenium
 
-RUN pip install selenium
+ENV SELENIUM_HEADLESS=true
 
 ADD page_perf_timer.py /opt/page_timer/
 
 WORKDIR /opt/page_timer
 
-ENTRYPOINT ["python", "page_perf_timer.py"]
+ENTRYPOINT ["python3", "page_perf_timer.py"]

page_perf_timer.py

@@ -84,12 +84,12 @@ def __init__(
         self.timings = {}
 
         """Start web driver"""
-        chrome_options = webdriver.ChromeOptions()
+        options = webdriver.FirefoxOptions()
         if os.environ.get("SELENIUM_HEADLESS"):
-            chrome_options.add_argument("--no-sandbox")
-            chrome_options.add_argument("--headless")
-            chrome_options.add_argument("--disable-gpu")
-        self.driver = webdriver.Chrome(options=chrome_options)
+            options.add_argument("--no-sandbox")
+            options.add_argument("--headless")
+            options.add_argument("--disable-gpu")
+        self.driver = webdriver.Firefox(options=options)
         # self.driver = webdriver.Firefox()
         self.driver.implicitly_wait(180)
         self.wait = WebDriverWait(self.driver, 180)