♻️ 整理代码

nonebot_bison/sub_manager/__init__.py

@@ -15,7 +15,9 @@ 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 .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
 
 add_sub_matcher = on_command(
@@ -45,23 +47,40 @@ do_del_sub(del_sub_matcher)
 add_cookie_matcher = on_command(
     "添加cookie",
     rule=configurable_to_me,
-    permission=admin_permission(),
+    permission=SUPERUSER,
     priority=5,
     block=True,
 )
-add_cookie_matcher.handle()(set_target_user_info)
 do_add_cookie(add_cookie_matcher)
 
 add_cookie_target_matcher = on_command(
     "关联cookie",
     rule=configurable_to_me,
-    permission=admin_permission(),
+    permission=SUPERUSER,
     priority=5,
     block=True,
 )
-add_cookie_target_matcher.handle()(set_target_user_info)
 do_add_cookie_target(add_cookie_target_matcher)
 
+del_cookie_target_matcher = on_command(
+    "取消关联cookie",
+    rule=configurable_to_me,
+    permission=SUPERUSER,
+    priority=5,
+    block=True,
+)
+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)
+
+
 group_manage_matcher = on_command("群管理", rule=to_me(), permission=SUPERUSER, priority=4, block=True)
 
 group_handle_cancel = gen_handle_cancel(group_manage_matcher, "已取消")

nonebot_bison/sub_manager/del_cookie.py

@@ -0,0 +1,9 @@
+from nonebot.matcher import Matcher
+
+from .utils import ensure_user_info, gen_handle_cancel
+
+
+def do_del_cookie(del_cookie: type[Matcher]):
+    handle_cancel = gen_handle_cancel(del_cookie, "删除中止")
+
+    del_cookie.handle()(ensure_user_info(del_cookie))

nonebot_bison/sub_manager/del_cookie_target.py

@@ -0,0 +1,9 @@
+from nonebot.matcher import Matcher
+
+from .utils import ensure_user_info, gen_handle_cancel
+
+
+def do_del_cookie_target(del_cookie_target: type[Matcher]):
+    handle_cancel = gen_handle_cancel(del_cookie_target, "删除中止")
+
+    del_cookie_target.handle()(ensure_user_info(del_cookie_target))