🐛 为cookie相关操作添加非私聊拒绝提示

nonebot_bison/sub_manager/add_cookie.py

@@ -6,6 +6,7 @@ from nonebot.typing import T_State
 from nonebot.matcher import Matcher
 from nonebot.params import Arg, ArgPlainText
 from nonebot.adapters import Message, MessageTemplate
+from nonebot.adapters.onebot.v11 import MessageEvent, PrivateMessageEvent
 
 from ..platform import platform_manager
 from .utils import common_platform, gen_handle_cancel
@@ -16,7 +17,9 @@ def do_add_cookie(add_cookie: type[Matcher]):
     handle_cancel = gen_handle_cancel(add_cookie, "已中止添加cookie")
 
     @add_cookie.handle()
-    async def init_promote(state: T_State):
+    async def init_promote(state: T_State, event: MessageEvent):
+        if not issubclass(PrivateMessageEvent, event.__class__):
+            await add_cookie.finish("请在私聊中使用此命令")
         state["_prompt"] = (
             "请输入想要添加 Cookie 的平台，目前支持，请输入冒号左边的名称：\n"
             + "".join(

nonebot_bison/sub_manager/add_cookie_target.py

@@ -3,6 +3,7 @@ from nonebot.matcher import Matcher
 from nonebot.params import ArgPlainText
 from nonebot_plugin_saa import MessageFactory
 from nonebot.internal.adapter import MessageTemplate
+from nonebot.adapters.onebot.v11 import MessageEvent, PrivateMessageEvent
 
 from ..config import config
 from ..utils import parse_text
@@ -14,7 +15,9 @@ def do_add_cookie_target(add_cookie_target_matcher: type[Matcher]):
     handle_cancel = gen_handle_cancel(add_cookie_target_matcher, "已中止关联 cookie")
 
     @add_cookie_target_matcher.handle()
-    async def init_promote(state: T_State):
+    async def init_promote(state: T_State, event: MessageEvent):
+        if not issubclass(PrivateMessageEvent, event.__class__):
+            await add_cookie_target_matcher.finish("请在私聊中使用此命令")
         res = await generate_sub_list_text(
             add_cookie_target_matcher, state, is_index=True, is_show_cookie=True, is_hide_no_cookie_platfrom=True
         )

nonebot_bison/sub_manager/del_cookie.py

@@ -2,6 +2,7 @@ from nonebot.typing import T_State
 from nonebot.matcher import Matcher
 from nonebot.params import EventPlainText
 from nonebot_plugin_saa import MessageFactory
+from nonebot.adapters.onebot.v11 import MessageEvent, PrivateMessageEvent
 
 from ..config import config
 from ..utils import parse_text
@@ -12,7 +13,9 @@ def do_del_cookie(del_cookie: type[Matcher]):
     handle_cancel = gen_handle_cancel(del_cookie, "删除中止")
 
     @del_cookie.handle()
-    async def send_list(state: T_State):
+    async def send_list(state: T_State, event: MessageEvent):
+        if not issubclass(PrivateMessageEvent, event.__class__):
+            await del_cookie.finish("请在私聊中使用此命令")
         cookies = await config.get_cookie(is_anonymous=False)
         if not cookies:
             await del_cookie.finish("暂无已添加的 Cookie\n请使用“添加cookie”命令添加")

nonebot_bison/sub_manager/del_cookie_target.py

@@ -2,6 +2,7 @@ from nonebot.typing import T_State
 from nonebot.matcher import Matcher
 from nonebot.params import EventPlainText
 from nonebot_plugin_saa import MessageFactory
+from nonebot.adapters.onebot.v11 import MessageEvent, PrivateMessageEvent
 
 from ..config import config
 from ..utils import parse_text
@@ -12,7 +13,9 @@ def do_del_cookie_target(del_cookie_target: type[Matcher]):
     handle_cancel = gen_handle_cancel(del_cookie_target, "取消关联中止")
 
     @del_cookie_target.handle()
-    async def send_list(state: T_State):
+    async def send_list(state: T_State, event: MessageEvent):
+        if not issubclass(PrivateMessageEvent, event.__class__):
+            await del_cookie_target.finish("请在私聊中使用此命令")
         cookie_targets = await config.get_cookie_target()
         if not cookie_targets:
             await del_cookie_target.finish("暂无已关联 Cookie\n请使用“添加cookie”命令添加关联")