vbatts/maubot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Update type hints and allow InternalEventTypes for @event.on

Browse source
This commit is contained in:
Tulir Asokan 2019-09-27 00:45:30 +03:00
parent ed6055744f
commit 7fcb7cbf0a
2 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
maubot/handlers/event.py
View file

@ -16,19 +16,20 @@
from typing import Callable, Union, NewType
from mautrix.types import EventType
from mautrix.client import EventHandler
from mautrix.client import EventHandler, InternalEventType
EventHandlerDecorator = NewType("EventHandlerDecorator", Callable[[EventHandler], EventHandler])
def on(var: Union[EventType, EventHandler]) -> Union[EventHandlerDecorator, EventHandler]:
def on(var: Union[EventType, InternalEventType, EventHandler]
) -> Union[EventHandlerDecorator, EventHandler]:
def decorator(func: EventHandler) -> EventHandler:
func.__mb_event_handler__ = True
if isinstance(var, EventType):
if isinstance(var, (EventType, InternalEventType)):
func.__mb_event_type__ = var
else:
func.__mb_event_type__ = EventType.ALL
return func
return decorator if isinstance(var, EventType) else decorator(var)
return decorator if isinstance(var, (EventType, InternalEventType)) else decorator(var)

2
maubot/matrix.py
View file

@ -73,7 +73,7 @@ class MaubotMessageEvent(MessageEvent):
def mark_read(self) -> Awaitable[None]:
return self.client.send_receipt(self.room_id, self.event_id, "m.read")
def react(self, key: str) -> Awaitable[None]:
def react(self, key: str) -> Awaitable[EventID]:
return self.client.react(self.room_id, self.event_id, key)
def edit(self, content: Union[str, MessageEventContent],