suyiiyii/nonebot-bison
1
0
Fork 0
mirror of https://github.com/suyiiyii/nonebot-bison.git synced 2025-06-04 02:26:11 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

:recycles: add_cookie

Browse Source
This commit is contained in:
suyiiyii 2024-09-09 11:35:39 +08:00
parent 65a5976897
commit d43d042618
3 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nonebot_bison/sub_manager/__init__.py
View File

@ -14,7 +14,7 @@ from nonebot.adapters.onebot.v11.event import PrivateMessageEvent
from .add_sub import do_add_sub from .add_sub import do_add_sub
from .del_sub import do_del_sub from .del_sub import do_del_sub
from .query_sub import do_query_sub from .query_sub import do_query_sub
# from .add_cookie import do_add_cookie from .add_cookie import do_add_cookie
# from .del_cookie import do_del_cookie # from .del_cookie import do_del_cookie
# from .add_cookie_target import do_add_cookie_target # from .add_cookie_target import do_add_cookie_target
# from .del_cookie_target import do_del_cookie_target # from .del_cookie_target import do_del_cookie_target
@ -51,7 +51,7 @@ add_cookie_matcher = on_command(
priority=5, priority=5,
block=True, block=True,
) )
# do_add_cookie(add_cookie_matcher) do_add_cookie(add_cookie_matcher)
# add_cookie_target_matcher = on_command( # add_cookie_target_matcher = on_command(
# "关联cookie", # "关联cookie",

3
nonebot_bison/sub_manager/add_cookie.py
View File

@ -44,6 +44,7 @@ def do_add_cookie(add_cookie: type[Matcher]):
await add_cookie.finish("已中止添加cookie") await add_cookie.finish("已中止添加cookie")
elif platform in platform_manager: elif platform in platform_manager:
state["platform"] = platform state["platform"] = platform
state["site"] = platform_manager[platform].site
else: else:
await add_cookie.reject("平台输入错误") await add_cookie.reject("平台输入错误")
@ -62,7 +63,7 @@ def do_add_cookie(add_cookie: type[Matcher]):
@add_cookie.handle() @add_cookie.handle()
async def add_cookie_process(state: T_State): async def add_cookie_process(state: T_State):
cookie = Cookie(platform_name=state["platform"], content=state["cookie"]) cookie = Cookie(site_name=state["site"].name, content=state["cookie"])
client_mgr = cast(CookieClientManager, platform_manager[state["platform"]].site.client_mgr) client_mgr = cast(CookieClientManager, platform_manager[state["platform"]].site.client_mgr)
cookie = await client_mgr.init_cookie(cookie) cookie = await client_mgr.init_cookie(cookie)
await config.add_cookie(cookie) await config.add_cookie(cookie)

5
nonebot_bison/utils/site.py
View File

@ -156,11 +156,6 @@ def create_cookie_client_manager(site_name: str) -> type[CookieClientManager]:
(CookieClientManager,), (CookieClientManager,),
{"_site_name": site_name}, {"_site_name": site_name},
) )
#
# def create_cookie_client_manager(platform_name: str):
# """创建一个平台特化的 CookieClientManger"""
# return DefaultClientManager
#
class Site(metaclass=RegistryMeta,base=True): class Site(metaclass=RegistryMeta,base=True):