完善cookies方面功能,未完成

This commit is contained in:
g1879 2020-11-30 06:56:49 +08:00
parent d10985030c
commit 4c392aa707
2 changed files with 34 additions and 5 deletions

View File

@ -726,10 +726,15 @@ def _session_options_to_dict(options: Union[dict, SessionOptions, None]) -> Unio
def _cookie_to_dict(cookie: Cookie) -> dict:
# TODO: 其它值?
cookie_dict = {'name': cookie.name, 'value': str(cookie.value), 'path': cookie.path, 'domain': cookie.domain}
# cookie_dict = {'name': cookie.name, 'value': str(cookie.value), 'path': cookie.path, 'domain': cookie.domain}
if cookie.expires:
cookie_dict['expiry'] = cookie.expires
# if cookie.expires:
# cookie_dict['expiry'] = cookie.expires
return cookie_dict
# return cookie_dict
if isinstance(cookie, Cookie):
return cookie.__dict__
elif isinstance(cookie, dict):
return cookie
else:
raise TypeError

View File

@ -155,6 +155,30 @@ class Drission(object):
for cookie in cookies:
self._ensure_add_cookie(cookie)
def set_cookies(self,
cookies: Union[RequestsCookieJar, list, tuple],
set_session: bool = False,
set_driver: bool = False):
if isinstance(cookies, (list, tuple, RequestsCookieJar)):
cookies = tuple(_cookie_to_dict(cookie) for cookie in cookies)
else:
raise TypeError
if set_session:
pass
if set_driver:
pass
def add_cookie(self):
pass
def remove_cookie(self):
pass
def clear_cookies(self):
pass
def _set_session(self, data: dict) -> None:
if self._session is None:
self._session = Session()