🎨 use Arg to replace assert

nonebot_bison/config_manager/add_sub.py

@@ -37,8 +37,6 @@ def do_add_sub(add_sub: Type[Matcher]):
 
     @add_sub.got("platform", MessageTemplate("{_prompt}"), [handle_cancel])
     async def parse_platform(state: T_State, platform: str = ArgPlainText()) -> None:
-        if not isinstance(state["platform"], Message):
-            return
         if platform == "全部":
             message = "全部平台\n" + "\n".join(
                 [
@@ -152,9 +150,9 @@ def do_add_sub(add_sub: Type[Matcher]):
             state["tags"] = raw_tags_text.split()
 
     @add_sub.handle()
-    async def add_sub_process(state: T_State):
+    async def add_sub_process(
-        user = cast(PlatformTarget, state.get("target_user_info"))
+        state: T_State, user: PlatformTarget = Arg("target_user_info")
-        assert isinstance(user, PlatformTarget)
+    ):
         try:
             await config.add_subscribe(
                 user=user,

nonebot_bison/config_manager/del_sub.py

@@ -1,7 +1,7 @@
 from typing import Type
 
 from nonebot.matcher import Matcher
-from nonebot.params import EventPlainText
+from nonebot.params import Arg, EventPlainText
 from nonebot.typing import T_State
 from nonebot_plugin_saa import MessageFactory, PlatformTarget
 
@@ -18,15 +18,12 @@ def do_del_sub(del_sub: Type[Matcher]):
     del_sub.handle()(ensure_user_info(del_sub))
 
     @del_sub.handle()
-    async def send_list(state: T_State):
+    async def send_list(
-        user_info = state["target_user_info"]
+        state: T_State, user_info: PlatformTarget = Arg("target_user_info")
-        assert isinstance(user_info, PlatformTarget)
+    ):
-        try:
         sub_list = await config.list_subscribe(user_info)
-            assert sub_list
+        if not sub_list:
-        except AssertionError:
             await del_sub.finish("暂无已订阅账号\n请使用“添加订阅”命令添加订阅")
-        else:
         res = "订阅的帐号为：\n"
         state["sub_table"] = {}
         for index, sub in enumerate(sub_list, 1):
@@ -57,11 +54,13 @@ def do_del_sub(del_sub: Type[Matcher]):
         await MessageFactory(await parse_text(res)).send()
 
     @del_sub.receive(parameterless=[handle_cancel])
-    async def do_del(state: T_State, index_str: str = EventPlainText()):
+    async def do_del(
+        state: T_State,
+        index_str: str = EventPlainText(),
+        user_info: PlatformTarget = Arg("target_user_info"),
+    ):
         try:
             index = int(index_str)
-            user_info = state["target_user_info"]
-            assert isinstance(user_info, PlatformTarget)
             await config.del_subscribe(user_info, **state["sub_table"][index])
         except Exception as e:
             await del_sub.reject("删除错误")

nonebot_bison/config_manager/query_sub.py

@@ -1,7 +1,7 @@
 from typing import Type
 
 from nonebot.matcher import Matcher
-from nonebot.typing import T_State
+from nonebot.params import Arg
 from nonebot_plugin_saa import MessageFactory, PlatformTarget
 
 from ..config import config
@@ -15,9 +15,7 @@ def do_query_sub(query_sub: Type[Matcher]):
     query_sub.handle()(ensure_user_info(query_sub))
 
     @query_sub.handle()
-    async def _(state: T_State):
+    async def _(user_info: PlatformTarget = Arg("target_user_info")):
-        user_info = state["target_user_info"]
-        assert isinstance(user_info, PlatformTarget)
         sub_list = await config.list_subscribe(user_info)
         res = "订阅的帐号为：\n"
         for sub in sub_list: