add '--cookies' command-line option

pull/30/head
Mike Fährmann 7 years ago
parent 58e95a7487
commit a804a42e23
No known key found for this signature in database
GPG Key ID: 5680CA389D365A88

@ -15,6 +15,7 @@ import queue
import logging
import requests
import threading
import http.cookiejar
from .message import Message
from .. import config
@ -30,6 +31,15 @@ class Extractor():
self.session = requests.Session()
self.log = logging.getLogger(self.category)
cookies = self.config("cookies")
if cookies:
try:
cj = http.cookiejar.MozillaCookieJar()
cj.load(cookies)
self.session.cookies = cj
except OSError as exc:
self.log.warning("cookies: %s", exc)
def __iter__(self):
return self.items()

@ -125,6 +125,11 @@ def build_parser():
metavar="SECONDS", action=ConfigAction, dest="timeout", type=float,
help="Timeout for HTTP connections (defaut: no timeout)",
)
parser.add_argument(
"--cookies",
metavar="FILE", action=ConfigAction, dest="cookies",
help="File to load additional cookies from",
)
parser.add_argument(
"-c", "--config",
metavar="CFG", dest="cfgfiles", action="append",

Loading…
Cancel
Save