diff options
Diffstat (limited to 'python/report_ops/scotia.py')
| -rw-r--r-- | python/report_ops/scotia.py | 14 |
1 files changed, 6 insertions, 8 deletions
diff --git a/python/report_ops/scotia.py b/python/report_ops/scotia.py index 2a3b48f9..c81dad5c 100644 --- a/python/report_ops/scotia.py +++ b/python/report_ops/scotia.py @@ -1,7 +1,6 @@ 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 @@ -68,15 +67,14 @@ def get_attachment_element(driver): def create_driver(download_dir): options = Options() - 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( + # 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( "browser.helperApps.neverAsk.saveToDisk", "application/octet-stream" ) - profile.set_preference("browser.helperApps.neverAsk.saveToDisk", "text/csv") - return Firefox(profile, options=options) + return webdriver.Firefox(firefox_profile=fp, options=options) def download_scotia_report(workdate): |