完善get_cookies(),没有domain的cookie也能获取到

This commit is contained in:
g1879 2020-12-08 20:56:12 +08:00
parent f294a777ed
commit 54d734089a

View File

@ -77,9 +77,12 @@ class SessionPage(object):
if all_domains:
cookies = self.session.cookies
else:
url = extract(self.url)
domain = f'{url.domain}.{url.suffix}'
cookies = tuple(x for x in self.session.cookies if domain in x.domain)
if self.url:
url = extract(self.url)
domain = f'{url.domain}.{url.suffix}'
cookies = tuple(x for x in self.session.cookies if domain in x.domain or x.domain == '')
else:
cookies = tuple(x for x in self.session.cookies)
if as_dict:
return {x.name: x.value for x in cookies}