From 96ab3ab2f586be54e6af367fc0bbbbbd8ee13926 Mon Sep 17 00:00:00 2001
From: Rosyid Haryadi <rosyid_haryadi@protonmail.com>
Date: Sun, 28 Jul 2024 00:24:41 +0700
Subject: fix handle connection problem

---
 downloader.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

(limited to 'downloader.py')

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__':
-- 
cgit v1.2.3-70-g09d2