Fix waiting for raw events again

nexy7574 · Dec 24, 2024 · c0092c0 · c0092c0
1 parent ca18bb8
commit c0092c0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/niobot/client.py b/src/niobot/client.py
@@ -716,6 +716,7 @@ def add_event_listener(self, event_type: typing.Union[str, nio.Event, Type[nio.E
             @functools.wraps(func)
             async def event_safety_wrapper(*args):
                 # This is necessary to stop callbacks crashing the process
+                self.log.debug("Raw event received: %r", args)
                 try:
                     return await func(*args)
                 except Exception as e:
@@ -724,6 +725,7 @@ async def event_safety_wrapper(*args):
             func = event_safety_wrapper
             self._raw_events.setdefault(event_type, [])
             self._raw_events[event_type].append(func)
+            self.add_event_callback(func, event_type)
             self.log.debug("Added raw event listener %r for %r", func, event_type)
             return func
         else: