mirror of
https://github.com/suyiiyii/nonebot-bison.git
synced 2025-05-31 16:16:11 +08:00
♻️ 将 refresh_anonymous_cookie 改为内部方法, 同时外部使用 refresh_client 方法进行刷新匿名 cookie
This commit is contained in:
parent
b130627d7e
commit
81e53419a3
@ -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
|
||||
|
@ -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
|
||||
|
@ -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:
|
||||
|
@ -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
|
||||
|
Loading…
x
Reference in New Issue
Block a user