blackteay/DrissionPage
1
0
Fork 0
mirror of https://gitee.com/g1879/DrissionPage.git synced 2024-12-10 04:00:23 +08:00
Code Issues Projects Releases Wiki Activity

优化cookies相关功能

Browse Source
This commit is contained in:
g1879 2020-12-04 18:03:34 +08:00
parent b9365a9361
commit 99fc7d650a
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
DrissionPage/drission.py
View File

@ -184,13 +184,23 @@ class Drission(object):
if 'expiry' in cookie: if 'expiry' in cookie:
cookie['expiry'] = int(cookie['expiry']) cookie['expiry'] = int(cookie['expiry'])
cookie_domain = cookie['domain'] if cookie['domain'][0] != '.' else cookie['domain'][1:]
try: try:
browser_domain = extract(self.driver.current_url).fqdn browser_domain = extract(self.driver.current_url).fqdn
except AttributeError: except AttributeError:
browser_domain = '' browser_domain = ''
if not cookie.get('domain', None):
if browser_domain:
url = extract(browser_domain)
cookie_domain = f'{url.domain}.{url.suffix}'
else:
raise ValueError('There is no domain name in the cookie or the browser has not visited a URL.')
cookie['domain'] = cookie_domain
else:
cookie_domain = cookie['domain'] if cookie['domain'][0] != '.' else cookie['domain'][1:]
if cookie_domain not in browser_domain: if cookie_domain not in browser_domain:
self.driver.get(cookie_domain if cookie_domain.startswith('http://') self.driver.get(cookie_domain if cookie_domain.startswith('http://')
else f'http://{cookie_domain}') else f'http://{cookie_domain}')