From 2c2c9a091c45be831da38ef310ec9cf62675f44f Mon Sep 17 00:00:00 2001 From: suyiiyii Date: Mon, 9 Sep 2024 18:36:18 +0800 Subject: [PATCH] :recycle: del_cookie --- nonebot_bison/sub_manager/__init__.py | 18 +++++++++--------- nonebot_bison/sub_manager/del_cookie.py | 6 +++--- 2 files changed, 12 insertions(+), 12 deletions(-) diff --git a/nonebot_bison/sub_manager/__init__.py b/nonebot_bison/sub_manager/__init__.py index aba8b33..943efd1 100644 --- a/nonebot_bison/sub_manager/__init__.py +++ b/nonebot_bison/sub_manager/__init__.py @@ -15,7 +15,7 @@ from .add_sub import do_add_sub from .del_sub import do_del_sub from .query_sub import do_query_sub 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 .del_cookie_target import do_del_cookie_target from .utils import common_platform, admin_permission, gen_handle_cancel, configurable_to_me, set_target_user_info @@ -71,14 +71,14 @@ del_cookie_target_matcher = on_command( ) do_del_cookie_target(del_cookie_target_matcher) -# del_cookie_matcher = on_command( -# "删除cookie", -# rule=configurable_to_me, -# permission=SUPERUSER, -# priority=5, -# block=True, -# ) -# do_del_cookie(del_cookie_matcher) +del_cookie_matcher = on_command( + "删除cookie", + rule=configurable_to_me, + permission=SUPERUSER, + priority=5, + block=True, +) +do_del_cookie(del_cookie_matcher) group_manage_matcher = on_command("群管理", rule=to_me(), permission=SUPERUSER, priority=4, block=True) diff --git a/nonebot_bison/sub_manager/del_cookie.py b/nonebot_bison/sub_manager/del_cookie.py index 4054078..6b3056f 100644 --- a/nonebot_bison/sub_manager/del_cookie.py +++ b/nonebot_bison/sub_manager/del_cookie.py @@ -5,8 +5,8 @@ from nonebot_plugin_saa import MessageFactory from ..config import config from ..utils import parse_text +from ..platform import site_manager from .utils import gen_handle_cancel -from ..platform import platform_manager def do_del_cookie(del_cookie: type[Matcher]): @@ -21,9 +21,9 @@ def do_del_cookie(del_cookie: type[Matcher]): state["cookie_table"] = {} for index, cookie in enumerate(cookies, 1): state["cookie_table"][index] = cookie - client_mgr = platform_manager[cookie.platform_name].site.client_mgr + client_mgr = site_manager[cookie.site_name].client_mgr friendly_name = await client_mgr.get_cookie_friendly_name(cookie) - res += f"{index} {cookie.platform_name} {friendly_name} {len(cookie.targets)}个关联\n" + res += f"{index} {cookie.site_name} {friendly_name} {len(cookie.targets)}个关联\n" if res[-1] != "\n": res += "\n" res += "请输入要删除的 Cookie 的序号\n输入'取消'中止"