Merge pull request #13 from Klavionik/support-async-callables

Support async callable classes

pymitter/__init__.py

@@ -193,7 +193,7 @@ def _emit(self: EventEmitter, event: str, *args, **kwargs) -> list[Awaitable]:
                 self.off(listener.event, func=listener.func)
 
             res = listener(*args, **kwargs)
-            if listener.is_coroutine:
+            if listener.is_coroutine or listener.is_async_callable:
                 awaitables.append(res)
 
         return awaitables
@@ -410,6 +410,10 @@ def __init__(self: Listener, func: Callable, event: str, ttl: int) -> None:
     def is_coroutine(self: Listener) -> bool:
         return asyncio.iscoroutinefunction(self.func)
 
+    @property
+    def is_async_callable(self):
+        return hasattr(self.func, "__call__") and asyncio.iscoroutinefunction(self.func.__call__)
+
     def __call__(self: Listener, *args, **kwargs) -> Any:
         """
         Invokes the wrapped function when ttl is non-zero, decreases the ttl value when positive and

tests/test_all.py

@@ -398,3 +398,16 @@ async def test():
                 self.assertEqual(tuple(stack), ("emit_future_bar",))
 
             await test()
+
+        def test_supports_async_callables(self):
+            ee = EventEmitter()
+            stack = []
+
+            class EventHandler:
+                async def __call__(self, arg):
+                    stack.append(arg)
+
+            ee.on("event", EventHandler())
+
+            ee.emit("event", "arg")
+            self.assertEqual(stack, ["arg"])