瀏覽代碼

Add COOKIES_FILE support for singlefile extractor

Naomi Phillips 1 年之前
父節點
當前提交
a729480b75
共有 1 個文件被更改,包括 2 次插入0 次删除
  1. 2 0
      archivebox/extractors/singlefile.py

+ 2 - 0
archivebox/extractors/singlefile.py

@@ -19,6 +19,7 @@ from ..config import (
     SINGLEFILE_VERSION,
     SINGLEFILE_VERSION,
     SINGLEFILE_ARGS,
     SINGLEFILE_ARGS,
     CHROME_BINARY,
     CHROME_BINARY,
+    COOKIES_FILE,
 )
 )
 from ..logging_util import TimedProgress
 from ..logging_util import TimedProgress
 
 
@@ -48,6 +49,7 @@ def save_singlefile(link: Link, out_dir: Optional[Path]=None, timeout: int=TIMEO
     browser_args = '--browser-args={}'.format(json.dumps(browser_args[1:]))
     browser_args = '--browser-args={}'.format(json.dumps(browser_args[1:]))
     options = [
     options = [
         *SINGLEFILE_ARGS,
         *SINGLEFILE_ARGS,
+        *(["--browser-cookies-file={}".format(COOKIES_FILE)] if COOKIES_FILE else []),
         '--browser-executable-path={}'.format(CHROME_BINARY),
         '--browser-executable-path={}'.format(CHROME_BINARY),
         browser_args,
         browser_args,
     ]
     ]