Adjust

nonebot/command/argfilter/__init__.py

@@ -1,50 +1,8 @@
-from typing import Awaitable, List, TYPE_CHECKING
+from typing import Optional
 
-from nonebot.helpers import render_expression
-from nonebot.typing import Filter_T
-
-if TYPE_CHECKING:
-    from nonebot.command import CommandSession
+from nonebot.typing import Message_T
 
 
 class ValidateError(ValueError):
-    def __init__(self, message=None):
+    def __init__(self, message: Optional[Message_T] = None):
         self.message = message
-
-
-async def run_arg_filters(session: 'CommandSession',
-                          arg_filters: List[Filter_T]) -> None:
-    """
-    Run a specific list of argument filters on a command session.
-
-    This will call all argument filter functions successively,
-    with `session.current_arg` as the argument.
-
-    If all filters are passed, the final result will be put into
-    `session.state` with the key `session.current_key`.
-
-    If some validation failed, the session will be paused and
-    failure message will be sent to the user.
-
-    :param session: command session to run on
-    :param arg_filters: argument filters
-    """
-    arg = session.current_arg
-    for f in arg_filters:
-        try:
-            res = f(arg)
-            if isinstance(res, Awaitable):
-                res = await res
-            arg = res
-        except ValidateError as e:
-            # validation failed
-            failure_message = e.message
-            if failure_message is None:
-                failure_message = render_expression(
-                    session.bot.config.DEFAULT_VALIDATION_FAILURE_EXPRESSION
-                )
-            # noinspection PyProtectedMember
-            session.pause(failure_message, **session._current_send_kwargs)
-
-    # passed all filters
-    session.state[session.current_key] = arg

nonebot/command/argfilter/validators.py

@@ -1,7 +1,8 @@
 import re
 from typing import Callable, Any
 
-from nonebot.command.argfilter import ValidateError, Filter_T
+from nonebot.command.argfilter import ValidateError
+from nonebot.typing import Filter_T
 
 
 class BaseValidator: