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-01 00:53:27 +08:00
parent 4c392aa707
commit c62b28d50e
1 changed files with 15 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
DrissionPage/drission.py
View File

@ -156,24 +156,30 @@ class Drission(object):
self._ensure_add_cookie(cookie) self._ensure_add_cookie(cookie)
def set_cookies(self, def set_cookies(self,
cookies: Union[RequestsCookieJar, list, tuple], cookies: Union[RequestsCookieJar, list, tuple, str],
set_session: bool = False, set_session: bool = False,
set_driver: bool = False): set_driver: bool = False) -> None:
if isinstance(cookies, (list, tuple, RequestsCookieJar)): if isinstance(cookies, (list, tuple, RequestsCookieJar)):
cookies = tuple(_cookie_to_dict(cookie) for cookie in cookies) cookies = tuple(_cookie_to_dict(cookie) for cookie in cookies)
elif isinstance(cookies, str):
cookies = tuple(dict([cookie.split("=", 1)]) for cookie in cookies.split("; "))
elif isinstance(cookies, dict):
cookies = tuple({'name': cookie, 'value': cookies[cookie]} for cookie in cookies)
else: else:
raise TypeError raise TypeError
for cookie in cookies:
if set_session: if set_session:
pass kwargs = {x: cookie[x] for x in cookie if x not in ('name', 'value')}
self.session.cookies.set(cookie['name'], cookie['value'], **kwargs)
if set_driver: if set_driver:
self.driver.add_cookie(cookie)
def add_a_cookie(self):
pass pass
def add_cookie(self): def remove_a_cookie(self):
pass
def remove_cookie(self):
pass pass
def clear_cookies(self): def clear_cookies(self):
@ -187,14 +193,7 @@ class Drission(object):
'cert', 'stream', 'trust_env', 'max_redirects'] # , 'adapters' 'cert', 'stream', 'trust_env', 'max_redirects'] # , 'adapters'
if 'cookies' in data: if 'cookies' in data:
if isinstance(data['cookies'], (list, tuple)): self.set_cookies(data['cookies'], set_driver=False)
for cookie in data['cookies']:
kwargs = {x: cookie[x] for x in cookie if x not in ('name', 'value')}
self._session.cookies.set(cookie['name'], cookie['value'], **kwargs)
elif isinstance(data['cookies'], RequestsCookieJar):
for cookie in data['cookies']:
self._session.cookies.set_cookie(cookie)
for i in attrs: for i in attrs:
if i in data: if i in data: