diff options
Diffstat (limited to 'python/download_scotia.py')
| -rw-r--r-- | python/download_scotia.py | 17 |
1 files changed, 15 insertions, 2 deletions
diff --git a/python/download_scotia.py b/python/download_scotia.py index 883ad6e5..a25c445c 100644 --- a/python/download_scotia.py +++ b/python/download_scotia.py @@ -5,6 +5,10 @@ from selenium.webdriver.support import expected_conditions as EC from selenium.webdriver.firefox.options import Options import time +import os + +from serenitas.utils.env import DAILY_DIR + # # create Firefox options object # options = Options() @@ -18,11 +22,12 @@ def wait(): ) +report_dir = "/home/serenitas/Daily/Selene/Scotia_reports" options = Options() options.add_argument("--headless") fp = webdriver.FirefoxProfile() fp.set_preference("browser.download.folderList", 2) -fp.set_preference("browser.download.dir", "/home/serenitas/Daily/Selene/Scotia_reports") +fp.set_preference("browser.download.dir", report_dir) fp.set_preference("browser.download.useDownloadDir", True) fp.set_preference("browser.helperApps.neverAsk.saveToDisk", "application/octet-stream") fp.set_preference("browser.download.manager.showWhenStarting", False) @@ -71,9 +76,17 @@ login_button = driver.find_element_by_id("leftNavInboxFolderLink") login_button.click() time.sleep(1) + elements = driver.find_elements_by_class_name("zfolder-msg-clickable")[0] elements.click() time.sleep(1) elements = driver.find_elements_by_class_name("zmessage-attachment-link")[0] -elements.click() + +REPORT_DIR = DAILY_DIR / "Selene" / "Scotia_reports" +fname = elements.text.split()[0] +# check if the file exists in the directory +if not os.path.exists(os.path.join(report_dir, fname)): + elements.click() +else: + print(f"{fname} exists already {report_dir}") |