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

修正创建session时写入cookies失败的问题，未完成

Browse Source
This commit is contained in:
g1879 2020-11-26 17:29:30 +08:00
parent ee95a8b45c
commit a91b22df49
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
DrissionPage/drission.py
View File

@ -9,6 +9,7 @@ from typing import Union
from urllib.parse import urlparse from urllib.parse import urlparse
from requests import Session from requests import Session
from requests.cookies import RequestsCookieJar
from selenium import webdriver from selenium import webdriver
from selenium.common.exceptions import WebDriverException, SessionNotCreatedException from selenium.common.exceptions import WebDriverException, SessionNotCreatedException
from selenium.webdriver.chrome.options import Options from selenium.webdriver.chrome.options import Options
@ -158,9 +159,19 @@ class Drission(object):
if self._session is None: if self._session is None:
self._session = Session() self._session = Session()
attrs = ['headers', 'cookies', 'auth', 'proxies', 'hooks', 'params', 'verify', attrs = ['headers', 'auth', 'proxies', 'hooks', 'params', 'verify',
'cert', 'adapters', 'stream', 'trust_env', 'max_redirects'] 'cert', 'adapters', 'stream', 'trust_env', 'max_redirects']
if 'cookies' in data:
if isinstance(data['cookies'], (list, tuple)):
pass
# 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:
self._session.__setattr__(i, data[i]) self._session.__setattr__(i, data[i])