Allow multiple event types in the same handler
This commit is contained in:
parent
ed046bcbfe
commit
61f154876d
3 changed files with 11 additions and 7 deletions
|
@ -72,7 +72,7 @@ class CommandHandler:
|
||||||
self.__mb_must_consume_args__: bool = True
|
self.__mb_must_consume_args__: bool = True
|
||||||
self.__mb_arg_fallthrough__: bool = True
|
self.__mb_arg_fallthrough__: bool = True
|
||||||
self.__mb_event_handler__: bool = True
|
self.__mb_event_handler__: bool = True
|
||||||
self.__mb_event_type__: EventType = EventType.ROOM_MESSAGE
|
self.__mb_event_types__: set[EventType] = {EventType.ROOM_MESSAGE}
|
||||||
self.__mb_msgtypes__: Iterable[MessageType] = (MessageType.TEXT,)
|
self.__mb_msgtypes__: Iterable[MessageType] = (MessageType.TEXT,)
|
||||||
self.__bound_copies__: Dict[Any, CommandHandler] = {}
|
self.__bound_copies__: Dict[Any, CommandHandler] = {}
|
||||||
self.__bound_instance__: Any = None
|
self.__bound_instance__: Any = None
|
||||||
|
@ -95,7 +95,7 @@ class CommandHandler:
|
||||||
"must_consume_args",
|
"must_consume_args",
|
||||||
"arg_fallthrough",
|
"arg_fallthrough",
|
||||||
"event_handler",
|
"event_handler",
|
||||||
"event_type",
|
"event_types",
|
||||||
"msgtypes",
|
"msgtypes",
|
||||||
]
|
]
|
||||||
for key in keys:
|
for key in keys:
|
||||||
|
@ -315,7 +315,7 @@ def new(
|
||||||
func.__mb_require_subcommand__ = require_subcommand
|
func.__mb_require_subcommand__ = require_subcommand
|
||||||
func.__mb_arg_fallthrough__ = arg_fallthrough
|
func.__mb_arg_fallthrough__ = arg_fallthrough
|
||||||
func.__mb_must_consume_args__ = must_consume_args
|
func.__mb_must_consume_args__ = must_consume_args
|
||||||
func.__mb_event_type__ = event_type
|
func.__mb_event_types__ = {event_type}
|
||||||
if msgtypes:
|
if msgtypes:
|
||||||
func.__mb_msgtypes__ = msgtypes
|
func.__mb_msgtypes__ = msgtypes
|
||||||
return func
|
return func
|
||||||
|
|
|
@ -27,9 +27,12 @@ def on(var: EventType | InternalEventType | EventHandler) -> EventHandlerDecorat
|
||||||
def decorator(func: EventHandler) -> EventHandler:
|
def decorator(func: EventHandler) -> EventHandler:
|
||||||
func.__mb_event_handler__ = True
|
func.__mb_event_handler__ = True
|
||||||
if isinstance(var, (EventType, InternalEventType)):
|
if isinstance(var, (EventType, InternalEventType)):
|
||||||
func.__mb_event_type__ = var
|
if hasattr(func, "__mb_event_types__"):
|
||||||
|
func.__mb_event_types__.add(var)
|
||||||
|
else:
|
||||||
|
func.__mb_event_types__ = {var}
|
||||||
else:
|
else:
|
||||||
func.__mb_event_type__ = EventType.ALL
|
func.__mb_event_types__ = {EventType.ALL}
|
||||||
|
|
||||||
return func
|
return func
|
||||||
|
|
||||||
|
|
|
@ -76,8 +76,9 @@ class Plugin(ABC):
|
||||||
val = getattr(obj, key)
|
val = getattr(obj, key)
|
||||||
try:
|
try:
|
||||||
if val.__mb_event_handler__:
|
if val.__mb_event_handler__:
|
||||||
self._handlers_at_startup.append((val, val.__mb_event_type__))
|
for event_type in val.__mb_event_types__:
|
||||||
self.client.add_event_handler(val.__mb_event_type__, val)
|
self._handlers_at_startup.append((val, event_type))
|
||||||
|
self.client.add_event_handler(event_type, val)
|
||||||
except AttributeError:
|
except AttributeError:
|
||||||
pass
|
pass
|
||||||
try:
|
try:
|
||||||
|
|
Loading…
Reference in a new issue