diff --git a/science_access/crawl.py b/science_access/crawl.py index f4a9baef91006657119ac7e8be0ebe57ba52e6bc..39f420bc5093068f8fee1db0c767364e6bf23152 100644 --- a/science_access/crawl.py +++ b/science_access/crawl.py @@ -146,9 +146,9 @@ def get_driver(): options.add_argument("--headless") options.add_argument("--disable-dev-shm-usage") options.add_argument("--no-sandbox") - if not heroku: + try: driver = webdriver.Firefox(options=options) - else: + except: try: options.binary_location = "/app/vendor/firefox/firefox" driver = webdriver.Firefox(options=options) diff --git a/science_access/scrape.py b/science_access/scrape.py index 0630cde4d6e821d221e598a3167e79755fb3edd2..ce122bd54c15746343bcb49c089fa3b3d0b7dad0 100644 --- a/science_access/scrape.py +++ b/science_access/scrape.py @@ -65,9 +65,9 @@ def get_driver(): options.add_argument("--headless") options.add_argument("--disable-dev-shm-usage") options.add_argument("--no-sandbox") - if not heroku: + try: driver = webdriver.Firefox(options=options) - else: + except: try: options.binary_location = "/app/vendor/firefox/firefox" driver = webdriver.Firefox(options=options)