nonebot2/tests/plugins/param/param_state.py

from typing import List, Tuple

from nonebot.typing import T_State
from nonebot.adapters import Message
from nonebot.params import (
    Command,
    RegexDict,
    CommandArg,
    RawCommand,
    RegexGroup,
    CommandStart,
    RegexMatched,
    ShellCommandArgs,
    ShellCommandArgv,
)


async def state(x: T_State) -> T_State:
    return x


async def legacy_state(state):
    return state


async def not_legacy_state(state: int):
    ...


async def command(cmd: Tuple[str, ...] = Command()) -> Tuple[str, ...]:
    return cmd


async def raw_command(raw_cmd: str = RawCommand()) -> str:
    return raw_cmd


async def command_arg(cmd_arg: Message = CommandArg()) -> Message:
    return cmd_arg


async def command_start(start: str = CommandStart()) -> str:
    return start


async def shell_command_args(
    shell_command_args: dict = ShellCommandArgs(),
) -> dict:
    return shell_command_args


async def shell_command_argv(
    shell_command_argv: List[str] = ShellCommandArgv(),
) -> List[str]:
    return shell_command_argv


async def regex_dict(regex_dict: dict = RegexDict()) -> dict:
    return regex_dict


async def regex_group(regex_group: Tuple = RegexGroup()) -> Tuple:
    return regex_group


async def regex_matched(regex_matched: str = RegexMatched()) -> str:
    return regex_matched