nonebot2/tests/plugins/param/param_state.py

56 lines
1.1 KiB
Python
Raw Normal View History

2021-12-20 00:28:02 +08:00
from typing import List, Tuple
from nonebot.typing import T_State
from nonebot.adapters import Message
from nonebot.params import (
State,
Command,
RegexDict,
CommandArg,
RawCommand,
RegexGroup,
RegexMatched,
ShellCommandArgs,
ShellCommandArgv,
)
async def state(x: T_State = State()) -> T_State:
return x
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 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