python-event-sourcery 0.5.1a4__tar.gz → 0.5.1a5__tar.gz

{python_event_sourcery-0.5.1a4 → python_event_sourcery-0.5.1a5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: python-event-sourcery
-Version: 0.5.1a4
+Version: 0.5.1a5
 Summary: A library for event-based systems in Python.
 License: MIT
 Author: Sebastian Buczyński

{python_event_sourcery-0.5.1a4 → python_event_sourcery-0.5.1a5}/event_sourcery/_event_store/event/dto.py

@@ -73,6 +73,7 @@ class Event(BaseModel, extra="forbid"):
 
 
 TEvent = TypeVar("TEvent", bound=Event)
+TContext = TypeVar("TContext", bound=BaseModel)
 
 
 class Context(BaseModel, extra="allow"):
@@ -127,8 +128,18 @@ class WrappedEvent(Generic[TEvent]):
     context: Context = dataclasses.field(default_factory=Context)
 
     @classmethod
-    def wrap(cls, event: TEvent, version: int | None) -> "WrappedEvent[TEvent]":
-        return WrappedEvent[TEvent](event=event, version=version)
+    def wrap(
+        cls,
+        event: TEvent,
+        version: int | None,
+        context: Context | None = None,
+    ) -> "WrappedEvent[TEvent]":
+        return WrappedEvent[TEvent](
+            event=event, version=version, context=context or Context()
+        )
+
+    def get_context(self, context_type: type[TContext]) -> TContext:
+        return context_type.model_validate(self.context.model_dump())
 
 
 @dataclasses.dataclass(frozen=True)

{python_event_sourcery-0.5.1a4 → python_event_sourcery-0.5.1a5}/event_sourcery/event_sourcing/repository.py

@@ -3,7 +3,7 @@ from contextlib import contextmanager
 from typing import Generic, TypeVar, cast
 
 from event_sourcery import EventStore, StreamId, StreamUUID
-from event_sourcery.event import Event, WrappedEvent
+from event_sourcery.event import Context, Event, WrappedEvent
 from event_sourcery.event_sourcing import Aggregate
 
 TAggregate = TypeVar("TAggregate", bound=Aggregate)
@@ -27,6 +27,7 @@ class Repository(Generic[TAggregate]):
         self,
         uuid: StreamUUID,
         aggregate: TAggregate,
+        context: Context | None = None,
     ) -> Iterator[TAggregate]:
         """
         Context manager for loading an aggregate instance.
@@ -38,6 +39,7 @@ class Repository(Generic[TAggregate]):
         Args:
             uuid (StreamUUID): The unique identifier of the aggregate's stream.
             aggregate (TAggregate): The aggregate initial instance to load state into.
+            context (Context | None): Optional context to attach to all emitted events.
 
         Yields:
             TAggregate: The loaded and ready-to-use aggregate instance.
@@ -45,7 +47,7 @@ class Repository(Generic[TAggregate]):
         stream_id = StreamId(uuid=uuid, name=uuid.name, category=aggregate.category)
         old_version = self._load(stream_id, aggregate)
         yield aggregate
-        self._save(aggregate, old_version, stream_id)
+        self._save(aggregate, old_version, stream_id, context)
 
     def _load(self, stream_id: StreamId, aggregate: TAggregate) -> int:
         stream = self._event_store.load_stream(stream_id)
@@ -61,11 +63,12 @@ class Repository(Generic[TAggregate]):
         aggregate: TAggregate,
         old_version: int,
         stream_id: StreamId,
+        context: Context | None = None,
     ) -> None:
         with aggregate.__persisting_changes__() as pending:
             start_from = old_version + 1
             events = [
-                WrappedEvent.wrap(event, version)
+                WrappedEvent.wrap(event, version, context=context)
                 for version, event in enumerate(pending, start=start_from)
             ]
 

{python_event_sourcery-0.5.1a4 → python_event_sourcery-0.5.1a5}/event_sourcery_sqlalchemy/models/__init__.py

@@ -1,7 +1,7 @@
 from typing import Protocol
 
 from sqlalchemy import MetaData
-from sqlalchemy.orm import registry
+from sqlalchemy.orm import DeclarativeBase, registry
 from sqlalchemy.orm.clsregistry import ClsRegistryToken
 
 from event_sourcery_sqlalchemy.models.base import (
@@ -28,7 +28,7 @@ _class_registry: dict[str, type | ClsRegistryToken] = {}
 
 
 def configure_models(
-    base: type[BaseProto],
+    base: type[BaseProto | DeclarativeBase],
     event_model: type[BaseEvent] = DefaultEvent,
     stream_model: type[BaseStream] = DefaultStream,
     snapshot_model: type[BaseSnapshot] = DefaultSnapshot,

{python_event_sourcery-0.5.1a4 → python_event_sourcery-0.5.1a5}/pyproject.toml

@@ -3,7 +3,7 @@ name = "python-event-sourcery"
 requires-python = ">=3.10"
 
 [tool.poetry]
-version = "0.5.1a4"
+version = "0.5.1a5"
 description = "A library for event-based systems in Python."
 readme = "README.md"
 authors = [