from __future__ import annotations
from typing import TYPE_CHECKING, Any
from examples.aggregate7.orjsonpydantic import PydanticApplication
from examples.aggregate8.domainmodel import Dog, Trick
if TYPE_CHECKING:
from uuid import UUID
[docs]
class DogSchool(PydanticApplication):
is_snapshotting_enabled = True
[docs]
def register_dog(self, name: str) -> UUID:
dog = Dog(name)
self.save(dog)
return dog.id
[docs]
def add_trick(self, dog_id: UUID, trick: str) -> None:
dog: Dog = self.repository.get(dog_id)
dog.add_trick(Trick(name=trick))
self.save(dog)
[docs]
def get_dog(self, dog_id: UUID) -> dict[str, Any]:
dog: Dog = self.repository.get(dog_id)
return {
"name": dog.name,
"tricks": tuple([t.name for t in dog.tricks]),
"created_on": dog.created_on,
"modified_on": dog.modified_on,
}