python-cq 0.2.2__tar.gz → 0.3.0__tar.gz

{python_cq-0.2.2 → python_cq-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-cq
-Version: 0.2.2
+Version: 0.3.0
 Summary: Lightweight CQRS library.
 Home-page: https://github.com/100nm/python-cq
 License: MIT

{python_cq-0.2.2 → python_cq-0.3.0}/cq/__init__.py

@@ -12,9 +12,9 @@ from ._core.message import (
     QueryBus,
     command_handler,
     event_handler,
-    find_command_bus,
-    find_event_bus,
-    find_query_bus,
+    get_command_bus,
+    get_event_bus,
+    get_query_bus,
     query_handler,
 )
 from ._core.middleware import Middleware, MiddlewareResult
@@ -35,8 +35,8 @@ __all__ = (
     "QueryBus",
     "command_handler",
     "event_handler",
-    "find_command_bus",
-    "find_event_bus",
-    "find_query_bus",
+    "get_command_bus",
+    "get_event_bus",
+    "get_query_bus",
     "query_handler",
 )

{python_cq-0.2.2 → python_cq-0.3.0}/cq/_core/dispatcher/bus.py

@@ -1,11 +1,11 @@
 import asyncio
-from abc import abstractmethod
+from abc import ABC, abstractmethod
 from collections import defaultdict
-from collections.abc import Callable
+from collections.abc import Awaitable, Callable
 from dataclasses import dataclass, field
 from inspect import isclass
 from types import GenericAlias
-from typing import Protocol, Self, TypeAliasType, runtime_checkable
+from typing import Any, Protocol, Self, TypeAliasType, runtime_checkable
 
 import injection
 
@@ -14,6 +14,8 @@ from cq._core.dispatcher.base import BaseDispatcher, Dispatcher
 type HandlerType[**P, T] = type[Handler[P, T]]
 type HandlerFactory[**P, T] = Callable[..., Handler[P, T]]
 
+type Listener[T] = Callable[[T], Awaitable[Any]]
+
 type BusType[I, O] = type[Bus[I, O]]
 
 
@@ -34,6 +36,10 @@ class Bus[I, O](Dispatcher[I, O], Protocol):
     def subscribe(self, input_type: type[I], factory: HandlerFactory[[I], O]) -> Self:
         raise NotImplementedError
 
+    @abstractmethod
+    def add_listeners(self, *listeners: Listener[I]) -> Self:
+        raise NotImplementedError
+
 
 @dataclass(eq=False, frozen=True, slots=True)
 class SubscriberDecorator[I, O]:
@@ -59,7 +65,24 @@ class SubscriberDecorator[I, O]:
         return self.injection_module.find_instance(self.bus_type)
 
 
-class SimpleBus[I, O](BaseDispatcher[I, O], Bus[I, O]):
+class BaseBus[I, O](BaseDispatcher[I, O], Bus[I, O], ABC):
+    __slots__ = ("__listeners",)
+
+    __listeners: list[Listener[I]]
+
+    def __init__(self) -> None:
+        super().__init__()
+        self.__listeners = []
+
+    def add_listeners(self, *listeners: Listener[I]) -> Self:
+        self.__listeners.extend(listeners)
+        return self
+
+    async def _trigger_listeners(self, input_value: I, /) -> None:
+        await asyncio.gather(*(listener(input_value) for listener in self.__listeners))
+
+
+class SimpleBus[I, O](BaseBus[I, O]):
     __slots__ = ("__handlers",)
 
     __handlers: dict[type[I], HandlerFactory[[I], O]]
@@ -69,6 +92,7 @@ class SimpleBus[I, O](BaseDispatcher[I, O], Bus[I, O]):
         self.__handlers = {}
 
     async def dispatch(self, input_value: I, /) -> O:
+        await self._trigger_listeners(input_value)
         input_type = type(input_value)
 
         try:
@@ -91,7 +115,7 @@ class SimpleBus[I, O](BaseDispatcher[I, O], Bus[I, O]):
         return self
 
 
-class TaskBus[I](BaseDispatcher[I, None], Bus[I, None]):
+class TaskBus[I](BaseBus[I, None]):
     __slots__ = ("__handlers",)
 
     __handlers: dict[type[I], list[HandlerFactory[[I], None]]]
@@ -101,6 +125,7 @@ class TaskBus[I](BaseDispatcher[I, None], Bus[I, None]):
         self.__handlers = defaultdict(list)
 
     async def dispatch(self, input_value: I, /) -> None:
+        await self._trigger_listeners(input_value)
         handler_factories = self.__handlers.get(type(input_value))
 
         if not handler_factories:

{python_cq-0.2.2 → python_cq-0.3.0}/cq/_core/message.py

@@ -39,13 +39,16 @@ injection.set_constant(TaskBus(), EventBus, alias=True)
 injection.set_constant(SimpleBus(), QueryBus, alias=True)
 
 
-def find_command_bus[T]() -> CommandBus[T]:
-    return injection.find_instance(CommandBus)
+@injection.inject
+def get_command_bus[T](bus: CommandBus[T] = NotImplemented, /) -> CommandBus[T]:
+    return bus
 
 
-def find_event_bus() -> EventBus:
-    return injection.find_instance(EventBus)
+@injection.inject
+def get_event_bus(bus: EventBus = NotImplemented, /) -> EventBus:
+    return bus
 
 
-def find_query_bus[T]() -> QueryBus[T]:
-    return injection.find_instance(QueryBus)
+@injection.inject
+def get_query_bus[T](bus: QueryBus[T] = NotImplemented, /) -> QueryBus[T]:
+    return bus

{python_cq-0.2.2 → python_cq-0.3.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "python-cq"
-version = "0.2.2"
+version = "0.3.0"
 description = "Lightweight CQRS library."
 license = "MIT"
 authors = ["remimd"]