set_cookies()增加refresh参数

This commit is contained in:
g1879 2020-12-08 20:24:23 +08:00
parent 4d6ca7f699
commit b9a1930ff8

View File

@ -140,9 +140,10 @@ class MixPage(Null, SessionPage, DriverPage):
elif self._mode == 'd':
return super(SessionPage, self).title
def set_cookies(self, cookies: Union[RequestsCookieJar, list, tuple, str, dict]) -> None:
def set_cookies(self, cookies: Union[RequestsCookieJar, list, tuple, str, dict], refresh: bool = True) -> None:
"""设置cookies \n
:param cookies: cookies信息可为CookieJar, list, tuple, str, dict
:param refresh: 设置cookies后是否刷新页面
:return: None
"""
if self._mode == 's':
@ -150,6 +151,9 @@ class MixPage(Null, SessionPage, DriverPage):
elif self._mode == 'd':
self.drission.set_cookies(cookies, set_driver=True)
if refresh:
self.refresh()
def get_cookies(self, as_dict: bool = False, all_domains: bool = False) -> Union[dict, list]:
"""返回cookies \n
:param as_dict: 是否以字典方式返回