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

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

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

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

@@ -1,16 +1,23 @@
-from ._core.command import (
+from ._core.dispatcher.bus import Bus
+from ._core.dispatcher.pipe import Pipe
+from ._core.dto import DTO
+from ._core.message import (
     AnyCommandBus,
     Command,
     CommandBus,
+    Event,
+    EventBus,
+    Message,
+    Query,
+    QueryBus,
     command_handler,
-    find_command_bus,
+    event_handler,
+    get_command_bus,
+    get_event_bus,
+    get_query_bus,
+    query_handler,
 )
-from ._core.dispatcher.bus import Bus
-from ._core.dispatcher.pipe import Pipe
-from ._core.dto import DTO
-from ._core.event import Event, EventBus, event_handler, find_event_bus
 from ._core.middleware import Middleware, MiddlewareResult
-from ._core.query import Query, QueryBus, find_query_bus, query_handler
 
 __all__ = (
     "AnyCommandBus",
@@ -20,6 +27,7 @@ __all__ = (
     "DTO",
     "Event",
     "EventBus",
+    "Message",
     "Middleware",
     "MiddlewareResult",
     "Pipe",
@@ -27,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.1 → python_cq-0.3.0}/cq/_core/dispatcher/base.py

@@ -1,7 +1,7 @@
 import asyncio
 from abc import ABC, abstractmethod
-from collections.abc import Awaitable
-from typing import Callable, Protocol, Self, runtime_checkable
+from collections.abc import Awaitable, Callable
+from typing import Protocol, Self, runtime_checkable
 
 from cq._core.middleware import Middleware, MiddlewareGroup
 

{python_cq-0.2.1 → 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.1 → python_cq-0.3.0}/cq/_core/dispatcher/pipe.py

@@ -1,6 +1,6 @@
-from collections.abc import Callable
+from collections.abc import Awaitable, Callable
 from dataclasses import dataclass, field
-from typing import Any, Awaitable, Self
+from typing import Any, Self
 
 from cq._core.dispatcher.base import BaseDispatcher, Dispatcher
 

python_cq-0.3.0/cq/_core/message.py

@@ -0,0 +1,54 @@
+from abc import ABC
+from typing import Any
+
+import injection
+
+from cq._core.dispatcher.bus import Bus, SimpleBus, SubscriberDecorator, TaskBus
+from cq._core.dto import DTO
+
+
+class Message(DTO, ABC):
+    __slots__ = ()
+
+
+class Command(Message, ABC):
+    __slots__ = ()
+
+
+class Event(Message, ABC):
+    __slots__ = ()
+
+
+class Query(Message, ABC):
+    __slots__ = ()
+
+
+type CommandBus[T] = Bus[Command, T]
+type EventBus = Bus[Event, None]
+type QueryBus[T] = Bus[Query, T]
+
+AnyCommandBus = CommandBus[Any]
+
+
+command_handler: SubscriberDecorator[Command, Any] = SubscriberDecorator(CommandBus)
+event_handler: SubscriberDecorator[Event, None] = SubscriberDecorator(EventBus)
+query_handler: SubscriberDecorator[Query, Any] = SubscriberDecorator(QueryBus)
+
+injection.set_constant(SimpleBus(), CommandBus, alias=True)
+injection.set_constant(TaskBus(), EventBus, alias=True)
+injection.set_constant(SimpleBus(), QueryBus, alias=True)
+
+
+@injection.inject
+def get_command_bus[T](bus: CommandBus[T] = NotImplemented, /) -> CommandBus[T]:
+    return bus
+
+
+@injection.inject
+def get_event_bus(bus: EventBus = NotImplemented, /) -> EventBus:
+    return bus
+
+
+@injection.inject
+def get_query_bus[T](bus: QueryBus[T] = NotImplemented, /) -> QueryBus[T]:
+    return bus

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

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

python_cq-0.2.1/cq/_core/command.py

@@ -1,22 +0,0 @@
-from abc import ABC
-from typing import Any
-
-import injection
-
-from cq._core.dispatcher.bus import Bus, SimpleBus, SubscriberDecorator
-from cq._core.dto import DTO
-
-
-class Command(DTO, ABC):
-    __slots__ = ()
-
-
-type CommandBus[T] = Bus[Command, T]
-AnyCommandBus = CommandBus[Any]
-command_handler: SubscriberDecorator[Command, Any] = SubscriberDecorator(CommandBus)
-
-injection.set_constant(SimpleBus(), CommandBus, alias=True)
-
-
-def find_command_bus[T]() -> CommandBus[T]:
-    return injection.find_instance(CommandBus)

python_cq-0.2.1/cq/_core/event.py

@@ -1,20 +0,0 @@
-from abc import ABC
-
-import injection
-
-from cq._core.dispatcher.bus import Bus, SubscriberDecorator, TaskBus
-from cq._core.dto import DTO
-
-
-class Event(DTO, ABC):
-    __slots__ = ()
-
-
-type EventBus = Bus[Event, None]
-event_handler: SubscriberDecorator[Event, None] = SubscriberDecorator(EventBus)
-
-injection.set_constant(TaskBus(), EventBus, alias=True)
-
-
-def find_event_bus() -> EventBus:
-    return injection.find_instance(EventBus)

python_cq-0.2.1/cq/_core/query.py

@@ -1,21 +0,0 @@
-from abc import ABC
-from typing import Any
-
-import injection
-
-from cq._core.dispatcher.bus import Bus, SimpleBus, SubscriberDecorator
-from cq._core.dto import DTO
-
-
-class Query(DTO, ABC):
-    __slots__ = ()
-
-
-type QueryBus[T] = Bus[Query, T]
-query_handler: SubscriberDecorator[Query, Any] = SubscriberDecorator(QueryBus)
-
-injection.set_constant(SimpleBus(), QueryBus, alias=True)
-
-
-def find_query_bus[T]() -> QueryBus[T]:
-    return injection.find_instance(QueryBus)