qena-shared-lib 0.1.19__tar.gz → 0.1.21__tar.gz

{qena_shared_lib-0.1.19 → qena_shared_lib-0.1.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: qena-shared-lib
-Version: 0.1.19
+Version: 0.1.21
 Summary: A shared tools for other services
 Requires-Dist: fastapi[all]==0.115.6
 Requires-Dist: prometheus-client==0.21.1

{qena_shared_lib-0.1.19 → qena_shared_lib-0.1.21}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "qena-shared-lib"
-version = "0.1.19"
+version = "0.1.21"
 description = "A shared tools for other services"
 readme = "README.md"
 requires-python = ">=3.10"

{qena_shared_lib-0.1.19 → qena_shared_lib-0.1.21}/src/qena_shared_lib/http/__init__.py

@@ -1,3 +1,5 @@
+from pydantic import Field
+
 from ._base import (
     ROUTE_HANDLER_ATTRIBUTE,
     APIRouter,
@@ -68,6 +70,7 @@ __all__ = [
     "Cookie",
     "delete",
     "Depends",
+    "Field",
     "File",
     "FileResponse",
     "Form",

{qena_shared_lib-0.1.19 → qena_shared_lib-0.1.21}/src/qena_shared_lib/mongodb.py

@@ -14,7 +14,16 @@ from typing import (
 
 from bson.objectid import ObjectId
 from pydantic import BeforeValidator, Field, field_serializer
-from pymongo import AsyncMongoClient, IndexModel
+from pymongo import (
+    ASCENDING,
+    DESCENDING,
+    GEO2D,
+    GEOSPHERE,
+    HASHED,
+    TEXT,
+    AsyncMongoClient,
+    IndexModel,
+)
 from pymongo.asynchronous.client_session import AsyncClientSession
 from pymongo.asynchronous.collection import AsyncCollection
 from pymongo.asynchronous.database import AsyncDatabase
@@ -25,14 +34,22 @@ from .logging import LoggerFactory
 
 __all__ = [
     "AggregatedDocument",
+    "ASCENDING",
+    "AsyncClientSession",
+    "DESCENDING",
     "Document",
     "EmbeddedDocument",
+    "Field",
+    "GEO2D",
+    "GEOSPHERE",
+    "HASHED",
     "IndexManager",
     "IndexModel",
     "MongoDBManager",
     "MongoDBObjectId",
     "ProjectedDocument",
     "RepositoryBase",
+    "TEXT",
     "validate_object_id",
 ]
 
@@ -235,11 +252,9 @@ S = TypeVar("S")
 
 
 class RepositoryBase(Generic[T]):
-    def __init__(
-        self, db: MongoDBManager, session: AsyncClientSession | None = None
-    ) -> None:
+    def __init__(self, db: MongoDBManager) -> None:
         self._db = db
-        self._session = session
+        self._session = None
         self._document_type = None
 
     @property
@@ -254,6 +269,15 @@ class RepositoryBase(Generic[T]):
     def session(self) -> AsyncClientSession | None:
         return self._session
 
+    @session.setter
+    def session(self, session: AsyncClientSession) -> None:
+        if session.has_ended:
+            raise RuntimeError(
+                f"session with id {session.session_id} has already ended"
+            )
+
+        self._session = session
+
     @property
     def document_type(self) -> type[T]:
         document_type = self._document_type
@@ -364,25 +388,28 @@ class RepositoryBase(Generic[T]):
         session: AsyncClientSession | None = None,
     ) -> T | P | None:
         if projection is not None:
-            return projection.from_raw_projected_document(
-                await self.collection.find_one(
-                    filter=filter,
-                    projection=projection.get_projection(),
-                    skip=skip,
-                    sort=sort,
-                    session=session or self.session,
-                )
+            document = await self.collection.find_one(
+                filter=filter,
+                projection=projection.get_projection(),
+                skip=skip,
+                sort=sort,
+                session=session or self.session,
             )
-
-        return self.document_type.from_raw_document(
-            await self.collection.find_one(
+        else:
+            document = await self.collection.find_one(
                 filter=filter,
-                projection=projection,
                 skip=skip,
                 sort=sort,
                 session=session or self.session,
             )
-        )
+
+        if document is None:
+            return None
+
+        if projection is not None:
+            return projection.from_raw_projected_document(document)
+
+        return self.document_type.from_raw_document(document)
 
     async def replace(
         self, replacement: T, session: AsyncClientSession | None = None

{qena_shared_lib-0.1.19 → qena_shared_lib-0.1.21}/src/qena_shared_lib/rabbitmq/message/__init__.py

@@ -1,3 +1,5 @@
+from pydantic import Field
+
 from ._inbound import (
     CamelCaseInboundMessage,
     InboundMessage,
@@ -12,6 +14,7 @@ from ._outbound import (
 __all__ = [
     "CamelCaseInboundMessage",
     "CamelCaseOutboundMessage",
+    "Field",
     "InboundMessage",
     "OutboundMessage",
     "SnakeCaseInboundMessage",