mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-11-25 17:45:05 +08:00
27 lines
833 B
Python
27 lines
833 B
Python
|
from typing import Union, Callable
|
||
|
|
|
||
|
|
from nonebot.command import on_command
|
||
|
|
from nonebot.typing import CommandName_T
|
||
|
|
|
||
|
|
|
||
|
|
class CommandGroup:
|
||
|
|
"""
|
||
|
|
Group a set of commands with same name prefix.
|
||
|
|
"""
|
||
|
|
|
||
|
|
__slots__ = ('basename', 'base_kwargs')
|
||
|
|
|
||
|
|
def __init__(self, name: Union[str, CommandName_T], **kwargs):
|
||
|
|
self.basename = (name,) if isinstance(name, str) else name
|
||
|
|
if 'aliases' in kwargs:
|
||
|
|
del kwargs['aliases'] # ensure there is no aliases here
|
||
|
|
self.base_kwargs = kwargs
|
||
|
|
|
||
|
|
def command(self, name: Union[str, CommandName_T], **kwargs) -> Callable:
|
||
|
|
sub_name = (name,) if isinstance(name, str) else name
|
||
|
|
name = self.basename + sub_name
|
||
|
|
|
||
|
|
final_kwargs = self.base_kwargs.copy()
|
||
|
|
final_kwargs.update(kwargs)
|
||
|
|
return on_command(name, **final_kwargs)
|