Changes for Pydantic v2 (model_dump() and Config.frozen).

pyeventsourcing · Oct 16, 2023 · befbc46 · befbc46
1 parent e34dd84
commit befbc46
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 9 deletions.
diff --git a/eventsourcing/examples/aggregate7/domainmodel.py b/eventsourcing/examples/aggregate7/domainmodel.py
@@ -16,7 +16,7 @@ class DomainEvent(BaseModel):
     timestamp: datetime
 
     class Config:
-        allow_mutation = False
+        frozen = True
 
 
 def create_timestamp() -> datetime:
@@ -30,24 +30,21 @@ class Aggregate(BaseModel):
     modified_on: datetime
 
     class Config:
-        allow_mutation = False
+        frozen = True
 
 
 class Snapshot(DomainEvent):
     topic: str
     state: Dict[str, Any]
 
-    class Config:
-        allow_mutation = False
-
     @classmethod
     def take(cls, aggregate: Aggregate) -> Snapshot:
         return Snapshot(
             originator_id=aggregate.id,
             originator_version=aggregate.version,
             timestamp=create_timestamp(),
             topic=get_topic(type(aggregate)),
-            state=aggregate.dict(),
+            state=aggregate.model_dump(),
         )
 
 

diff --git a/eventsourcing/examples/aggregate7/persistence.py b/eventsourcing/examples/aggregate7/persistence.py
@@ -17,7 +17,7 @@
 class PydanticMapper(Mapper):
     def to_stored_event(self, domain_event: DomainEventProtocol) -> StoredEvent:
         topic = get_topic(domain_event.__class__)
-        event_state = cast(BaseModel, domain_event).dict()
+        event_state = cast(BaseModel, domain_event).model_dump()
         stored_state = self.transcoder.encode(event_state)
         if self.compressor:
             stored_state = self.compressor.compress(stored_state)

diff --git a/eventsourcing/examples/aggregate8/domainmodel.py b/eventsourcing/examples/aggregate8/domainmodel.py
@@ -19,7 +19,7 @@ class DomainEvent(BaseModel):
     timestamp: datetime
 
     class Config:
-        allow_mutation = False
+        frozen = True
 
 
 class Aggregate(BaseAggregate):

diff --git a/eventsourcing/examples/aggregate8/persistence.py b/eventsourcing/examples/aggregate8/persistence.py
@@ -17,7 +17,7 @@
 class PydanticMapper(Mapper):
     def to_stored_event(self, domain_event: DomainEventProtocol) -> StoredEvent:
         topic = get_topic(domain_event.__class__)
-        event_state = cast(BaseModel, domain_event).dict()
+        event_state = cast(BaseModel, domain_event).model_dump()
         stored_state = self.transcoder.encode(event_state)
         if self.compressor:
             stored_state = self.compressor.compress(stored_state)