cda/maubot
1
0
Fork 0
forked from vbatts/maubot
Code Pull requests Activity

Some sort of command handling system

Browse source
This commit is contained in:
Tulir Asokan 2018-12-18 00:53:39 +02:00
parent f104595217
commit 682eab348d
5 changed files with 107 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

19
maubot/handlers/event.py
View file

@ -23,20 +23,21 @@ from mautrix.client import EventHandler
EventHandlerDecorator = NewType("EventHandlerDecorator", Callable[[EventHandler], EventHandler])
def handler(var: Union[EventType, EventHandler]) -> Union[EventHandlerDecorator, EventHandler]:
def on(var: Union[EventType, EventHandler]) -> Union[EventHandlerDecorator, EventHandler]:
def decorator(func: EventHandler) -> EventHandler:
func.__mb_event_handler__ = True
@functools.wraps(func)
async def wrapper(event: Event) -> None:
pass
wrapper.__mb_event_handler__ = True
if isinstance(var, EventType):
func.__mb_event_type__ = var
wrapper.__mb_event_type__ = var
else:
func.__mb_event_type__ = EventType.ALL
wrapper.__mb_event_type__ = EventType.ALL
return func
return wrapper
if isinstance(var, EventType):
return decorator
else:
decorator(var)
return decorator if isinstance(var, EventType) else decorator(var)
class Field: