nonebot2/nonebot/params.py

import inspect
from typing import Any, Dict, Optional

from nonebot.typing import T_State
from nonebot.dependencies import Param
from nonebot.adapters import Bot, Event
from nonebot.utils import generic_check_issubclass


class BotParam(Param):

    @classmethod
    def _check(cls, name: str, param: inspect.Parameter) -> bool:
        return generic_check_issubclass(
            param.annotation, Bot) or (param.annotation == param.empty and
                                       name == "bot")

    def _solve(self, bot: Bot, **kwargs: Any) -> Any:
        return bot


class EventParam(Param):

    @classmethod
    def _check(cls, name: str, param: inspect.Parameter) -> bool:
        return generic_check_issubclass(
            param.annotation, Event) or (param.annotation == param.empty and
                                         name == "event")

    def _solve(self, event: Event, **kwargs: Any) -> Any:
        return event


class StateParam(Param):

    @classmethod
    def _check(cls, name: str, param: inspect.Parameter) -> bool:
        return generic_check_issubclass(
            param.annotation, Dict) or (param.annotation == param.empty and
                                        name == "state")

    def _solve(self, state: T_State, **kwargs: Any) -> Any:
        return state


class MatcherParam(Param):

    @classmethod
    def _check(cls, name: str, param: inspect.Parameter) -> bool:
        return generic_check_issubclass(
            param.annotation, Matcher) or (param.annotation == param.empty and
                                           name == "matcher")

    def _solve(self, matcher: Optional["Matcher"] = None, **kwargs: Any) -> Any:
        return matcher


class ExceptionParam(Param):

    @classmethod
    def _check(cls, name: str, param: inspect.Parameter) -> bool:
        return generic_check_issubclass(
            param.annotation, Exception) or (param.annotation == param.empty and
                                             name == "exception")

    def _solve(self,
               exception: Optional[Exception] = None,
               **kwargs: Any) -> Any:
        return exception


from nonebot.matcher import Matcher