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

🐛 fix test 2

Browse Source
This commit is contained in:
suyiiyii 2024-11-26 01:18:26 +08:00
parent d75fe36c3b
commit 202b8df0a6
Signed by: suyiiyii
GPG Key ID: 044704CB29B8AD85
2 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
nonebot_bison/sub_manager/add_cookie_target.py
View File

@ -24,10 +24,9 @@ def do_add_cookie_target(add_cookie_target_matcher: type[Matcher]):
await MessageFactory(await parse_text(res)).send() await MessageFactory(await parse_text(res)).send()
@add_cookie_target_matcher.got("target_idx", parameterless=[handle_cancel]) @add_cookie_target_matcher.got("target_idx", parameterless=[handle_cancel])
async def got_target_idx(state: T_State, target_idx_str: str = ArgPlainText()): async def got_target_idx(state: T_State, target_idx: str = ArgPlainText()):
try: try:
target_idx = int(target_idx_str) state["target"] = state["sub_table"][int(target_idx)]
state["target"] = state["sub_table"][target_idx]
state["site"] = platform_manager[state["target"]["platform_name"]].site state["site"] = platform_manager[state["target"]["platform_name"]].site
except Exception: except Exception:
await add_cookie_target_matcher.reject("序号错误") await add_cookie_target_matcher.reject("序号错误")
@ -55,10 +54,9 @@ def do_add_cookie_target(add_cookie_target_matcher: type[Matcher]):
) )
@add_cookie_target_matcher.got("cookie_idx", MessageTemplate("{_prompt}"), [handle_cancel]) @add_cookie_target_matcher.got("cookie_idx", MessageTemplate("{_prompt}"), [handle_cancel])
async def got_cookie_idx(state: T_State, cookie_idx_str: str = ArgPlainText()): async def got_cookie_idx(state: T_State, cookie_idx: str = ArgPlainText()):
try: try:
cookie_idx = int(cookie_idx_str) state["cookie"] = state["cookies"][int(cookie_idx) - 1]
state["cookie"] = state["cookies"][cookie_idx - 1]
except Exception: except Exception:
await add_cookie_target_matcher.reject("序号错误") await add_cookie_target_matcher.reject("序号错误")

2
nonebot_bison/sub_manager/utils.py
View File

@ -127,5 +127,5 @@ async def only_allow_private(
event: Event, event: Event,
matcher: type[Matcher], matcher: type[Matcher],
): ):
if event.__getattr__("message_type") != "private": if not (hasattr(event, "message_type") and getattr(event, "message_type") == "private"):
await matcher.finish("请在私聊中使用此命令") await matcher.finish("请在私聊中使用此命令")