From b222642821faa50c191ea69f3adf73726314f179 Mon Sep 17 00:00:00 2001
From: Rosyid Haryadi <rosyid_haryadi@protonmail.com>
Date: Fri, 26 Jul 2024 22:05:23 +0700
Subject: add logger

---
 downloader.py | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'downloader.py')

diff --git a/downloader.py b/downloader.py
index 8b74611..58b5f42 100644
--- a/downloader.py
+++ b/downloader.py
@@ -6,12 +6,17 @@ import time
 
 from dotenv import load_dotenv
 from selenium import webdriver
+from selenium.common import TimeoutException
 from selenium.webdriver import ActionChains
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.webdriver.support.wait import WebDriverWait
 
+from util import setupLogger
+
 load_dotenv()
 
+logger = setupLogger()
+
 
 def driverSetup(url):
     op = webdriver.ChromeOptions()
@@ -49,6 +54,10 @@ def downloadCsv():
         downloadButton.click()
 
         time.sleep(5)
+        logger.info("REPORT FILE DOWNLOADED")
+
+    except TimeoutException:
+        logger.error("Timed out while downloading report")
 
     finally:
         driver.quit()
-- 
cgit v1.2.3-70-g09d2