fastapi-factory-utilities 0.2.6__tar.gz → 0.2.8__tar.gz

{fastapi_factory_utilities-0.2.6 → fastapi_factory_utilities-0.2.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: fastapi_factory_utilities
-Version: 0.2.6
+Version: 0.2.8
 Summary: Consolidate libraries and utilities to create microservices in Python with FastAPI, Beanie, Httpx, AioPika and OpenTelemetry.
 License: MIT
 Keywords: python,fastapi,beanie,httpx,opentelemetry,microservices

{fastapi_factory_utilities-0.2.6 → fastapi_factory_utilities-0.2.8}/pyproject.toml

@@ -3,7 +3,7 @@ name = "fastapi_factory_utilities"
 homepage = "https://github.com/miragecentury/fastapi_factory_utilities"
 repository = "https://github.com/miragecentury/fastapi_factory_utilities"
 keywords = ["python", "fastapi", "beanie", "httpx", "opentelemetry", "microservices"]
-version = "0.2.6"
+version = "0.2.8"
 description = "Consolidate libraries and utilities to create microservices in Python with FastAPI, Beanie, Httpx, AioPika and OpenTelemetry."
 authors = ["miragecentury <victorien.vanroye@gmail.com>"]
 maintainers = ["miragecentury <victorien.vanroye@gmail.com>"]

{fastapi_factory_utilities-0.2.6 → fastapi_factory_utilities-0.2.8}/src/fastapi_factory_utilities/core/plugins/odm_plugin/documents.py

@@ -17,6 +17,7 @@ class BaseDocument(Document):
         default_factory=uuid4, description="The document ID."
     )
 
+    revision_id: UUID | None = Field(default=None, exclude=False)
     created_at: Annotated[datetime.datetime, Indexed(index_type=DESCENDING)] = Field(  # pyright: ignore
         default_factory=lambda: datetime.datetime.now(tz=datetime.UTC), description="Creation timestamp."
     )

{fastapi_factory_utilities-0.2.6 → fastapi_factory_utilities-0.2.8}/src/fastapi_factory_utilities/core/plugins/odm_plugin/repositories.py

@@ -1,10 +1,12 @@
 """Provides the abstract classes for the repositories."""
 
+import datetime
 from abc import ABC
 from collections.abc import AsyncGenerator, Callable
 from contextlib import asynccontextmanager
 from typing import Any, Generic, TypeVar, get_args
 from uuid import UUID
+from venv import create
 
 from motor.motor_asyncio import AsyncIOMotorClientSession, AsyncIOMotorDatabase
 from pydantic import BaseModel
@@ -77,8 +79,13 @@ class AbstractRepository(ABC, Generic[DocumentGenericType, EntityGenericType]):
             UnableToCreateEntityDueToDuplicateKeyError: If the entity cannot be created due to a duplicate key error.
             OperationError: If the operation fails.
         """
+        insert_time: datetime.datetime = datetime.datetime.now(tz=datetime.timezone.utc)
         try:
-            document: DocumentGenericType = self._document_type(**entity.model_dump())
+            entity_dump: dict[str, Any] = entity.model_dump()
+            entity_dump["created_at"] = insert_time
+            entity_dump["updated_at"] = insert_time
+            document: DocumentGenericType = self._document_type(**entity_dump)
+
         except ValueError as error:
             raise ValueError(f"Failed to create document from entity: {error}") from error
 
@@ -96,6 +103,44 @@ class AbstractRepository(ABC, Generic[DocumentGenericType, EntityGenericType]):
 
         return entity_created
 
+    @managed_session()
+    async def update(
+        self, entity: EntityGenericType, session: AsyncIOMotorClientSession | None = None
+    ) -> EntityGenericType:
+        """Update the entity in the database.
+
+        Args:
+            entity (EntityGenericType): The entity to update.
+            session (AsyncIOMotorClientSession | None): The session to use. Defaults to None. (managed by decorator)
+
+        Returns:
+            EntityGenericType: The updated entity.
+
+        Raises:
+            ValueError: If the entity cannot be created from the document.
+            OperationError: If the operation fails.
+        """
+        update_time: datetime.datetime = datetime.datetime.now(tz=datetime.timezone.utc)
+        try:
+            entity_dump: dict[str, Any] = entity.model_dump()
+            entity_dump["updated_at"] = update_time
+            document: DocumentGenericType = self._document_type(**entity_dump)
+
+        except ValueError as error:
+            raise ValueError(f"Failed to create document from entity: {error}") from error
+
+        try:
+            document_updated: DocumentGenericType = await document.save(session=session)
+        except PyMongoError as error:
+            raise OperationError(f"Failed to update document: {error}") from error
+
+        try:
+            entity_updated: EntityGenericType = self._entity_type(**document_updated.model_dump())
+        except ValueError as error:
+            raise ValueError(f"Failed to create entity from document: {error}") from error
+
+        return entity_updated
+
     @managed_session()
     async def get_one_by_id(
         self,