🐛 add skip handle for process

This commit is contained in:
yanyongyu 2021-12-06 00:17:52 +08:00
parent c25ee1cb56
commit ca4d7397f8
4 changed files with 76 additions and 32 deletions

View File

@ -156,7 +156,7 @@ async def solve_dependencies(
f"type {type(value)} not match depends {_dependent.func} "
f"annotation {field._type_display()}, ignored"
)
raise SkippedException
raise SkippedException(field, value)
else:
values[field.name] = value

View File

@ -8,6 +8,8 @@
from typing import Any, Optional
from pydantic.fields import ModelField
class NoneBotException(Exception):
"""
@ -124,6 +126,16 @@ class SkippedException(MatcherException):
可以在 ``Handler`` 中通过 ``Matcher.skip()`` 抛出
"""
def __init__(self, param: ModelField, value: Any):
self.param = param
self.value = value
def __repr__(self):
return f"<SkippedException, param={self.param}, value={self.value}>"
def __str__(self):
self.__repr__()
class PausedException(MatcherException):
"""

View File

@ -669,8 +669,11 @@ class Matcher(metaclass=MatcherMeta):
_stack=stack,
_dependency_cache=dependency_cache,
)
except SkippedException:
pass
except SkippedException as e:
logger.debug(
f"Handler {handler} param {e.param.name} value {e.value} "
f"mismatch type {e.param._type_display()}, skipped"
)
except RejectedException:
self.handlers.insert(0, handler) # type: ignore

View File

@ -8,7 +8,16 @@ NoneBot 内部处理并按优先级分发事件给所有事件响应器,提供
import asyncio
from datetime import datetime
from contextlib import AsyncExitStack
from typing import TYPE_CHECKING, Any, Set, Dict, Type, Optional
from typing import (
TYPE_CHECKING,
Any,
Set,
Dict,
List,
Type,
Optional,
Coroutine,
)
from nonebot import params
from nonebot.log import logger
@ -16,7 +25,12 @@ from nonebot.rule import TrieRule
from nonebot.handler import Handler
from nonebot.utils import escape_tag
from nonebot.matcher import Matcher, matchers
from nonebot.exception import NoLogException, StopPropagation, IgnoredException
from nonebot.exception import (
NoLogException,
StopPropagation,
IgnoredException,
SkippedException,
)
from nonebot.typing import (
T_State,
T_DependencyCache,
@ -100,6 +114,13 @@ def run_postprocessor(func: T_RunPostProcessor) -> T_RunPostProcessor:
# FIXME: run handler with try/except skipped exception
async def _run_coro_with_catch(coro: Coroutine[Any, Any, Any]) -> Any:
try:
return await coro
except SkippedException:
pass
async def _check_matcher(
priority: int,
Matcher: Type[Matcher],
@ -150,13 +171,15 @@ async def _run_matcher(
coros = list(
map(
lambda x: x(
lambda x: _run_coro_with_catch(
x(
matcher=matcher,
bot=bot,
event=event,
state=state,
_stack=stack,
_dependency_cache=dependency_cache,
)
),
_run_preprocessors,
)
@ -189,7 +212,8 @@ async def _run_matcher(
coros = list(
map(
lambda x: x(
lambda x: _run_coro_with_catch(
x(
matcher=matcher,
exception=exception,
bot=bot,
@ -197,6 +221,7 @@ async def _run_matcher(
state=state,
_stack=stack,
_dependency_cache=dependency_cache,
)
),
_run_postprocessors,
)
@ -247,12 +272,14 @@ async def handle_event(bot: "Bot", event: "Event") -> None:
async with AsyncExitStack() as stack:
coros = list(
map(
lambda x: x(
lambda x: _run_coro_with_catch(
x(
bot=bot,
event=event,
state=state,
_stack=stack,
_dependency_cache=dependency_cache,
)
),
_event_preprocessors,
)
@ -307,12 +334,14 @@ async def handle_event(bot: "Bot", event: "Event") -> None:
coros = list(
map(
lambda x: x(
lambda x: _run_coro_with_catch(
x(
bot=bot,
event=event,
state=state,
_stack=stack,
_dependency_cache=dependency_cache,
)
),
_event_postprocessors,
)