mirror of
https://gitee.com/g1879/DrissionPage.git
synced 2024-12-10 04:00:23 +08:00
111 lines
3.6 KiB
Python
111 lines
3.6 KiB
Python
# -*- coding:utf-8 -*-
|
||
"""
|
||
@Author : g1879
|
||
@Contact : g1879@qq.com
|
||
@Copyright: (c) 2024 by g1879, Inc. All Rights Reserved.
|
||
@License : BSD 3-Clause.
|
||
"""
|
||
from http.cookiejar import Cookie
|
||
|
||
from .._functions.web import set_browser_cookies, set_session_cookies
|
||
|
||
|
||
class CookiesSetter(object):
|
||
def __init__(self, owner):
|
||
"""
|
||
:param owner: ChromiumBase对象
|
||
"""
|
||
self._owner = owner
|
||
|
||
def __call__(self, cookies):
|
||
"""设置一个或多个cookie
|
||
:param cookies: cookies信息
|
||
:return: None
|
||
"""
|
||
if (isinstance(cookies, dict) and 'name' in cookies and 'value' in cookies) or isinstance(cookies, Cookie):
|
||
cookies = [cookies]
|
||
set_browser_cookies(self._owner, cookies)
|
||
|
||
def remove(self, name, url=None, domain=None, path=None):
|
||
"""删除一个cookie
|
||
:param name: cookie的name字段
|
||
:param url: cookie的url字段，可选
|
||
:param domain: cookie的domain字段，可选
|
||
:param path: cookie的path字段，可选
|
||
:return: None
|
||
"""
|
||
d = {'name': name}
|
||
if url is not None:
|
||
d['url'] = url
|
||
if domain is not None:
|
||
d['domain'] = domain
|
||
if not url and not domain:
|
||
d['url'] = self._owner.url
|
||
if path is not None:
|
||
d['path'] = path
|
||
self._owner.run_cdp('Network.deleteCookies', **d)
|
||
|
||
def clear(self):
|
||
"""清除cookies"""
|
||
self._owner.run_cdp('Network.clearBrowserCookies')
|
||
|
||
|
||
class SessionCookiesSetter(object):
|
||
def __init__(self, owner):
|
||
self._owner = owner
|
||
|
||
def __call__(self, cookies):
|
||
"""设置多个cookie，注意不要传入单个
|
||
:param cookies: cookies信息
|
||
:return: None
|
||
"""
|
||
if (isinstance(cookies, dict) and 'name' in cookies and 'value' in cookies) or isinstance(cookies, Cookie):
|
||
cookies = [cookies]
|
||
set_session_cookies(self._owner.session, cookies)
|
||
|
||
def remove(self, name):
|
||
"""删除一个cookie
|
||
:param name: cookie的name字段
|
||
:return: None
|
||
"""
|
||
self._owner.session.cookies.set(name, None)
|
||
|
||
def clear(self):
|
||
"""清除cookies"""
|
||
self._owner.session.cookies.clear()
|
||
|
||
|
||
class WebPageCookiesSetter(CookiesSetter, SessionCookiesSetter):
|
||
|
||
def __call__(self, cookies):
|
||
"""设置多个cookie，注意不要传入单个
|
||
:param cookies: cookies信息
|
||
:return: None
|
||
"""
|
||
if self._owner.mode == 'd' and self._owner._has_driver:
|
||
super().__call__(cookies)
|
||
elif self._owner.mode == 's' and self._owner._has_session:
|
||
super(CookiesSetter, self).__call__(cookies)
|
||
|
||
def remove(self, name, url=None, domain=None, path=None):
|
||
"""删除一个cookie
|
||
:param name: cookie的name字段
|
||
:param url: cookie的url字段，可选，d模式时才有效
|
||
:param domain: cookie的domain字段，可选，d模式时才有效
|
||
:param path: cookie的path字段，可选，d模式时才有效
|
||
:return: None
|
||
"""
|
||
if self._owner.mode == 'd' and self._owner._has_driver:
|
||
super().remove(name, url, domain, path)
|
||
elif self._owner.mode == 's' and self._owner._has_session:
|
||
if url or domain or path:
|
||
raise AttributeError('url、domain、path参数只有d模式下有效。')
|
||
super(CookiesSetter, self).remove(name)
|
||
|
||
def clear(self):
|
||
"""清除cookies"""
|
||
if self._owner.mode == 'd' and self._owner._has_driver:
|
||
super().clear()
|
||
elif self._owner.mode == 's' and self._owner._has_session:
|
||
super(CookiesSetter, self).clear()
|