From 81e53419a396340304853987ce9b7ab01887a271 Mon Sep 17 00:00:00 2001
From: suyiiyii <suyiiyii@gmail.com>
Date: Tue, 29 Oct 2024 09:38:08 +0800
Subject: [PATCH] =?UTF-8?q?:recycle:=20=E5=B0=86=20refresh=5Fanonymous=5Fc?=
 =?UTF-8?q?ookie=20=E6=94=B9=E4=B8=BA=E5=86=85=E9=83=A8=E6=96=B9=E6=B3=95?=
 =?UTF-8?q?=EF=BC=8C=20=E5=90=8C=E6=97=B6=E5=A4=96=E9=83=A8=E4=BD=BF?=
 =?UTF-8?q?=E7=94=A8=20refresh=5Fclient=20=E6=96=B9=E6=B3=95=E8=BF=9B?=
 =?UTF-8?q?=E8=A1=8C=E5=88=B7=E6=96=B0=E5=8C=BF=E5=90=8D=20cookie?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 nonebot_bison/platform/bilibili/scheduler.py | 2 +-
 nonebot_bison/scheduler/manager.py           | 2 +-
 nonebot_bison/utils/site.py                  | 4 ++--
 tests/config/test_cookie.py                  | 2 +-
 4 files changed, 5 insertions(+), 5 deletions(-)

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