diff options
| -rw-r--r-- | python/report_ops/scotia.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/python/report_ops/scotia.py b/python/report_ops/scotia.py index fccb45a7..4b7ac7bc 100644 --- a/python/report_ops/scotia.py +++ b/python/report_ops/scotia.py @@ -49,6 +49,10 @@ def navigate_to_inbox(driver, workdate): EC.presence_of_element_located((By.ID, "leftNavFolderLink")) ) login_button.click() + inbox_link = wait.until( + EC.presence_of_element_located((By.ID, "leftNavInboxFolderLink")) + ) + inbox_link.click() wait.until(EC.presence_of_element_located((By.CLASS_NAME, "zfolder-msg"))) for msg in driver.find_elements_by_class_name("zfolder-msg"): date_string = re.search( @@ -66,7 +70,7 @@ def get_attachment_element(driver): attach = wait.until( EC.presence_of_element_located((By.CLASS_NAME, "zmessage-attachment-link")) ) - return attach + return driver.find_elements_by_class_name("zmessage-attachment-link")[0] def create_driver(download_dir): |