#20 add download directory

lugensa · Jan 13, 2021 · 6d72931 · 6d72931
1 parent 831c975
commit 6d72931
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/gocept/selenium/webdriver.py b/src/gocept/selenium/webdriver.py
@@ -105,6 +105,17 @@ def get_chrome_webdriver_args(self):
             raise NotImplementedError(
                 'Chromedriver currently only works headless.')
 
+        # Setup download dir.
+        self['selenium_download_dir'] = pathlib.Path(tempfile.mkdtemp(
+            prefix='gocept.selenium.download-dir'))
+
+        prefs = {
+            'download.default_directory': str(self['selenium_download_dir']),
+            'download.prompt_for_download': False,
+        }
+
+        options.add_experimental_option('prefs', prefs)
+
         return {
             'options': options,
             'service_args': ['--log-path=chromedriver.log']