suyiiyii/nonebot-bison
1
0
Fork 0
mirror of https://github.com/suyiiyii/nonebot-bison.git synced 2025-06-02 09:26:12 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

♻️ 将 refresh_anonymous_cookie 改为内部方法， 同时外部使用 refresh_client 方法进行刷新匿名 cookie

Browse Source
This commit is contained in:
suyiiyii 2024-10-29 09:38:08 +08:00
parent b130627d7e
commit 81e53419a3
Signed by: suyiiyii
GPG Key ID: 044704CB29B8AD85
4 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nonebot_bison/platform/bilibili/scheduler.py
View File

@ -78,7 +78,7 @@ class BilibiliClientManager(CookieClientManager):
@override @override
async def refresh_client(self): async def refresh_client(self):
self.refresh_anonymous_cookie() await self._refresh_anonymous_cookie()
logger.debug("刷新B站客户端的cookie") logger.debug("刷新B站客户端的cookie")
@override @override

2
nonebot_bison/scheduler/manager.py
View File

@ -35,7 +35,7 @@ async def init_scheduler():
for site, target_list in _schedule_class_dict.items(): for site, target_list in _schedule_class_dict.items():
if is_cookie_client_manager(site.client_mgr): if is_cookie_client_manager(site.client_mgr):
client_mgr = cast(CookieClientManager, site.client_mgr) client_mgr = cast(CookieClientManager, site.client_mgr)
await client_mgr.refresh_anonymous_cookie() await client_mgr.refresh_client()
if not plugin_config.bison_use_browser and site.require_browser: if not plugin_config.bison_use_browser and site.require_browser:
logger.warning(f"{site.name} requires browser, it will not schedule.") logger.warning(f"{site.name} requires browser, it will not schedule.")
continue continue

4
nonebot_bison/utils/site.py
View File

@ -60,7 +60,7 @@ class CookieClientManager(ClientManager):
) )
@classmethod @classmethod
async def refresh_anonymous_cookie(cls): async def _refresh_anonymous_cookie(cls):
"""更新已有的匿名cookie，若不存在则添加""" """更新已有的匿名cookie，若不存在则添加"""
existing_anonymous_cookies = await config.get_cookie(cls._site_name, is_anonymous=True) existing_anonymous_cookies = await config.get_cookie(cls._site_name, is_anonymous=True)
if existing_anonymous_cookies: if existing_anonymous_cookies:
@ -135,7 +135,7 @@ class CookieClientManager(ClientManager):
return http_client() return http_client()
async def refresh_client(self): async def refresh_client(self):
self.refresh_anonymous_cookie() await self._refresh_anonymous_cookie()
def is_cookie_client_manager(manger: type[ClientManager]) -> bool: def is_cookie_client_manager(manger: type[ClientManager]) -> bool:

2
tests/config/test_cookie.py
View File

@ -30,7 +30,7 @@ async def test_cookie(app: App, init_scheduler):
client_mgr = cast(CookieClientManager, site.client_mgr) client_mgr = cast(CookieClientManager, site.client_mgr)
# 刷新匿名cookie # 刷新匿名cookie
await client_mgr.refresh_anonymous_cookie() await client_mgr.refresh_client()
cookies = await config.get_cookie(site_name=site.name) cookies = await config.get_cookie(site_name=site.name)
assert len(cookies) == 1 assert len(cookies) == 1