diff options
Diffstat (limited to 'python/report_ops/scotia.py')
| -rw-r--r-- | python/report_ops/scotia.py | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/python/report_ops/scotia.py b/python/report_ops/scotia.py index c81dad5c..2a3b48f9 100644 --- a/python/report_ops/scotia.py +++ b/python/report_ops/scotia.py @@ -1,6 +1,7 @@ import re from selenium import webdriver from selenium.webdriver.firefox.options import Options +from selenium.webdriver import Firefox, FirefoxProfile from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC from selenium.webdriver.common.by import By @@ -67,14 +68,15 @@ def get_attachment_element(driver): def create_driver(download_dir): options = Options() - # options.add_argument("--headless") - fp = webdriver.FirefoxProfile() - fp.set_preference("browser.download.dir", str(download_dir)) - fp.set_preference("browser.download.folderList", 2) - fp.set_preference( + options.add_argument("--headless") + profile = FirefoxProfile() + profile.set_preference("browser.download.dir", str(download_dir)) + profile.set_preference("browser.download.folderList", 2) + profile.set_preference( "browser.helperApps.neverAsk.saveToDisk", "application/octet-stream" ) - return webdriver.Firefox(firefox_profile=fp, options=options) + profile.set_preference("browser.helperApps.neverAsk.saveToDisk", "text/csv") + return Firefox(profile, options=options) def download_scotia_report(workdate): |