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

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

View File

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

View File

@ -35,7 +35,7 @@ async def init_scheduler():
for site, target_list in _schedule_class_dict.items():
if is_cookie_client_manager(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:
logger.warning(f"{site.name} requires browser, it will not schedule.")
continue

View File

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

View File

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