♻️ 更新部分方法使用 scheduler_dict 进行cookie操作

This commit is contained in:
2024-10-29 12:24:31 +08:00
parent 9d985eb3c8
commit 29c2eb456d
4 changed files with 8 additions and 5 deletions
+2 -1
View File
@@ -12,6 +12,7 @@ async def test_cookie(app: App, init_scheduler):
from nonebot_bison.platform import site_manager
from nonebot_bison.config.db_config import config
from nonebot_bison.scheduler import scheduler_dict
from nonebot_bison.types import Target as T_Target
from nonebot_bison.utils.site import CookieClientManager
from nonebot_bison.config.utils import DuplicateCookieTargetException
@@ -27,7 +28,7 @@ async def test_cookie(app: App, init_scheduler):
tags=[],
)
site = site_manager["weibo.com"]
client_mgr = cast(CookieClientManager, site.client_mgr)
client_mgr = cast(CookieClientManager, scheduler_dict[site].client_mgr)
# 刷新匿名cookie
await client_mgr.refresh_client()
+1 -1
View File
@@ -32,7 +32,7 @@ def bilibili(app: App) -> "Bilibili":
from nonebot_bison.platform import platform_manager
from nonebot_bison.platform.bilibili import BilibiliClientManager
return platform_manager["bilibili"](ProcessContext(BilibiliClientManager())) # type: ignore
return platform_manager["bilibili"](ProcessContext(BilibiliClientManager(platform_manager["bilibili"].site))) # type: ignore
@pytest.fixture