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 17:54:15 +08:00
parent c62b28d50e
commit 7a8558befc
2 changed files with 16 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
DrissionPage/config.py
View File

@ -726,14 +726,14 @@ def _session_options_to_dict(options: Union[dict, SessionOptions, None]) -> Unio
def _cookie_to_dict(cookie: Cookie) -> dict: def _cookie_to_dict(cookie: Cookie) -> dict:
# cookie_dict = {'name': cookie.name, 'value': str(cookie.value), 'path': cookie.path, 'domain': cookie.domain} """把Cookie对象转为dict格式"""
# if cookie.expires:
# cookie_dict['expiry'] = cookie.expires
# return cookie_dict
if isinstance(cookie, Cookie): if isinstance(cookie, Cookie):
return cookie.__dict__ # return cookie.__dict__
cookie_dict = {'name': cookie.name, 'value': str(cookie.value), 'path': cookie.path, 'domain': cookie.domain}
if cookie.expires:
cookie_dict['expiry'] = cookie.expires
elif isinstance(cookie, dict): elif isinstance(cookie, dict):
return cookie return cookie
else: else:

12
DrissionPage/drission.py
View File

@ -162,7 +162,7 @@ class Drission(object):
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): elif isinstance(cookies, str):
cookies = tuple(dict([cookie.split("=", 1)]) for cookie in cookies.split("; ")) cookies = tuple(dict([cookie.lstrip().split("=", 1)]) for cookie in cookies.split(";"))
elif isinstance(cookies, dict): elif isinstance(cookies, dict):
cookies = tuple({'name': cookie, 'value': cookies[cookie]} for cookie in cookies) cookies = tuple({'name': cookie, 'value': cookies[cookie]} for cookie in cookies)
else: else:
@ -176,10 +176,16 @@ class Drission(object):
if set_driver: if set_driver:
self.driver.add_cookie(cookie) self.driver.add_cookie(cookie)
def add_a_cookie(self): def add_a_cookie(self,
cookie: str,
set_session: bool = False,
set_driver: bool = False):
pass pass
def remove_a_cookie(self): def remove_a_cookie(self,
name: str,
set_session: bool = False,
set_driver: bool = False):
pass pass
def clear_cookies(self): def clear_cookies(self):