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

避免s模式设置cookie时sameSite字段导致的错误

Browse Source
This commit is contained in:
g1879 2021-01-06 17:45:10 +08:00
parent f2ca4fad46
commit 25c59cf950
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
DrissionPage/drission.py
View File

@ -4,7 +4,6 @@
@Contact : g1879@qq.com @Contact : g1879@qq.com
@File : drission.py @File : drission.py
""" """
from sys import exit
from typing import Union from typing import Union
from requests import Session from requests import Session
@ -13,6 +12,7 @@ from selenium import webdriver
from selenium.common.exceptions import SessionNotCreatedException, WebDriverException from selenium.common.exceptions import SessionNotCreatedException, WebDriverException
from selenium.webdriver.chrome.options import Options from selenium.webdriver.chrome.options import Options
from selenium.webdriver.chrome.webdriver import WebDriver from selenium.webdriver.chrome.webdriver import WebDriver
from sys import exit
from tldextract import extract from tldextract import extract
from .config import (_dict_to_chrome_options, _session_options_to_dict, from .config import (_dict_to_chrome_options, _session_options_to_dict,
@ -194,7 +194,8 @@ class Drission(object):
# 添加cookie到session # 添加cookie到session
if set_session: if set_session:
kwargs = {x: cookie[x] for x in cookie if x not in ('name', 'value', 'httpOnly', 'expiry')} kwargs = {x: cookie[x] for x in cookie
if x.lower() not in ('name', 'value', 'httponly', 'expiry', 'samesite')}
if 'expiry' in cookie: if 'expiry' in cookie:
kwargs['expires'] = cookie['expiry'] kwargs['expires'] = cookie['expiry']