diff --git a/nonebot_bison/platform/bilibili/scheduler.py b/nonebot_bison/platform/bilibili/scheduler.py index 928b8cf..b117dcd 100644 --- a/nonebot_bison/platform/bilibili/scheduler.py +++ b/nonebot_bison/platform/bilibili/scheduler.py @@ -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 diff --git a/nonebot_bison/scheduler/manager.py b/nonebot_bison/scheduler/manager.py index 27a294d..ca02a56 100644 --- a/nonebot_bison/scheduler/manager.py +++ b/nonebot_bison/scheduler/manager.py @@ -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 diff --git a/nonebot_bison/utils/site.py b/nonebot_bison/utils/site.py index 7a6af53..e570e10 100644 --- a/nonebot_bison/utils/site.py +++ b/nonebot_bison/utils/site.py @@ -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: diff --git a/tests/config/test_cookie.py b/tests/config/test_cookie.py index 6a03e12..549cb7f 100644 --- a/tests/config/test_cookie.py +++ b/tests/config/test_cookie.py @@ -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