allow 'cookies' config option to be a dictionary

This commit is contained in:
Mike Fährmann
2017-07-07 18:01:46 +02:00
parent a804a42e23
commit 726c6f01ae

View File

@@ -33,12 +33,15 @@ class Extractor():
cookies = self.config("cookies") cookies = self.config("cookies")
if cookies: if cookies:
try: if isinstance(cookies, dict):
cj = http.cookiejar.MozillaCookieJar() cj = cookies
cj.load(cookies) else:
self.session.cookies = cj try:
except OSError as exc: cj = http.cookiejar.MozillaCookieJar()
self.log.warning("cookies: %s", exc) cj.load(cookies)
except OSError as exc:
self.log.warning("cookies: %s", exc)
self.session.cookies.update(cj)
def __iter__(self): def __iter__(self):
return self.items() return self.items()