diff options
Diffstat (limited to 'downloader.py')
| -rw-r--r-- | downloader.py | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/downloader.py b/downloader.py index 7ab3c79..5515b66 100644 --- a/downloader.py +++ b/downloader.py @@ -4,6 +4,7 @@ Ini nanti periodik import os import time +import selenium from dotenv import load_dotenv from selenium import webdriver from selenium.common import TimeoutException @@ -31,7 +32,12 @@ class Browser: op.add_argument("--no-sandbox") op.add_argument("--disable-dev-shm-usage") self.driver = webdriver.Chrome(options=op) - self.driver.get(url) + try: + self.driver.get(url) + self.connectionSuccess = True + except selenium.common.exceptions.WebDriverException: + self.connectionSuccess = False + logger.error(f"Selenium couldn't connect to {url}") def getDriver(self): return self.driver @@ -42,7 +48,10 @@ class Browser: def downloadCsv(): + downloadSuccess = False browser = Browser(os.getenv('URL')) + if not browser.connectionSuccess: + return False driver = browser.getDriver() action = ActionChains(driver) @@ -66,12 +75,15 @@ def downloadCsv(): time.sleep(5) logger.info("REPORT FILE DOWNLOADED") + downloadSuccess = True except TimeoutException: logger.error("Timed out while downloading report") + downloadSuccess = False finally: browser.close() + return downloadSuccess if __name__ == '__main__': |