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

对Drission对象设置session_options属性时可即使改变内置的Session

Browse Source
This commit is contained in:
g1879 2020-11-25 23:13:15 +08:00
parent 2f019fb4ba
commit 161bbb92ad
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
DrissionPage/drission.py
View File

@ -70,13 +70,7 @@ class Drission(object):
def session(self) -> Session:
"""返回Session对象，如未初始化则按配置信息创建"""
if self._session is None:
self._session = Session()
attrs = ['headers', 'cookies', 'auth', 'proxies', 'hooks', 'params', 'verify',
'cert', 'adapters', 'stream', 'trust_env', 'max_redirects']
for i in attrs:
if i in self._session_options:
self._session.__setattr__(i, self._session_options[i])
self._set_session(self._session_options)
if self._proxy:
self._session.proxies = self._proxy
@ -131,6 +125,7 @@ class Drission(object):
:return: None
"""
self._session_options = _session_options_to_dict(options)
self._set_session(self._session_options)
@property
def proxy(self) -> Union[None, dict]:
@ -159,6 +154,17 @@ class Drission(object):
for cookie in cookies:
self._ensure_add_cookie(cookie)
def _set_session(self, data: dict) -> None:
if self._session is None:
self._session = Session()
attrs = ['headers', 'cookies', 'auth', 'proxies', 'hooks', 'params', 'verify',
'cert', 'adapters', 'stream', 'trust_env', 'max_redirects']
for i in attrs:
if i in data:
self._session.__setattr__(i, data[i])
def cookies_to_session(self, copy_user_agent: bool = False,
driver: WebDriver = None,
session: Session = None) -> None: