add logger

1 files changed, 9 insertions, 0 deletions

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()