qena-shared-lib 0.1.20__py3-none-any.whl → 0.1.22__py3-none-any.whl

qena_shared_lib/mongodb.py

@@ -388,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
@@ -584,16 +587,40 @@ class RepositoryBase(Generic[T]):
 
         return cast(int, await self.collection.estimated_document_count())
 
-    async def aggregate(
+    @overload
+    def aggregate(
         self,
+        *,
         aggregation: type[A],
         let: dict[str, Any] | None = None,
         session: AsyncClientSession | None = None,
     ) -> AsyncGenerator[A, None]:
+        pass
+
+    @overload
+    def aggregate(
+        self,
+        *,
+        aggregation: type[A],
+        pipeline: list[Any],
+        let: dict[str, Any] | None = None,
+        session: AsyncClientSession | None = None,
+    ) -> AsyncGenerator[A, None]:
+        pass
+
+    async def aggregate(
+        self, *_: Any, **kwarg: Any
+    ) -> AsyncGenerator[Any, None]:
+        aggregation = cast(AggregatedDocument, kwarg.get("aggregation"))
+        pipeline = kwarg.get("pipeline")
+
+        if pipeline is None:
+            pipeline = aggregation.get_pipeline()
+
         async with await self.collection.aggregate(
-            pipeline=aggregation.get_pipeline(),
-            let=let,
-            session=session or self.session,
+            pipeline=pipeline,
+            let=kwarg.get("let"),
+            session=kwarg.get("session") or self.session,
         ) as cursor:
             async for document in cursor:
                 yield aggregation.from_raw_aggregated_document(document)

{qena_shared_lib-0.1.20.dist-info → qena_shared_lib-0.1.22.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: qena-shared-lib
-Version: 0.1.20
+Version: 0.1.22
 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.20.dist-info → qena_shared_lib-0.1.22.dist-info}/RECORD

@@ -21,7 +21,7 @@ qena_shared_lib/kafka/_consumer.py,sha256=aEOtAqMhFPIMYbKK_BMDwR2rd_eaDzBJ6xR3Ud
 qena_shared_lib/kafka/_exception_handlers.py,sha256=D7Xh5eSpSkumQKvPfBmVB_E6vQxD58y41bFpP5Jzhqk,4084
 qena_shared_lib/kafka/_producer.py,sha256=QtYfKzNtxeIT1EtGpS-tvFC342oPoleF1AOl20LfNXs,3941
 qena_shared_lib/logging.py,sha256=HZiqbXpmJsAI7hheAaQ9EA8uj9yU0pMVCg64d_V0wzA,1748
-qena_shared_lib/mongodb.py,sha256=CHkegKThenGB4A9bb0wNMLYGdrGo8-jDMx_3wRngIJ4,17056
+qena_shared_lib/mongodb.py,sha256=tlrmXxDae5IOZz-alL559ySKpRrTTQvOGFGB2QdJ5Ww,17701
 qena_shared_lib/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qena_shared_lib/rabbitmq/__init__.py,sha256=g15fP9XQ1CHFBXZmeAxQcsppQ0Nx62gJ4VOK8nzW7T4,1254
 qena_shared_lib/rabbitmq/_base.py,sha256=pKJL0PNrhUOXw4XUJ1iZsuCPVY3ykvO8ebIfeX9HwXo,21502
@@ -45,6 +45,6 @@ qena_shared_lib/scheduler.py,sha256=Spp5PnpFJ0U4aLeFCNmje9zUA4zwkL-b7X8SSGkGssM,
 qena_shared_lib/security.py,sha256=RAXBkfhytJl-2SOXYQTJMnUb789-8DNZKRjRtpod16Y,6372
 qena_shared_lib/sync.py,sha256=79aGGWGYg_oGY9jTav1kghD8BSRY-_oNYdx3tuExr20,2541
 qena_shared_lib/utils.py,sha256=crD8NJD1Se7ZHPCaQBCi3c7Y46lMvk8OsN_bZTbcyto,1184
-qena_shared_lib-0.1.20.dist-info/WHEEL,sha256=k57ZwB-NkeM_6AsPnuOHv5gI5KM5kPD6Vx85WmGEcI0,78
-qena_shared_lib-0.1.20.dist-info/METADATA,sha256=NXMw-mncYxD5lp76H1gsdJpqoOTmwYldBIkxyXMYNfo,19281
-qena_shared_lib-0.1.20.dist-info/RECORD,,
+qena_shared_lib-0.1.22.dist-info/WHEEL,sha256=k57ZwB-NkeM_6AsPnuOHv5gI5KM5kPD6Vx85WmGEcI0,78
+qena_shared_lib-0.1.22.dist-info/METADATA,sha256=e-RJxza3kT6NvTVgU4z-ZPNppAjKd35vCjkBWeNwjNI,19281
+qena_shared_lib-0.1.22.dist-info/RECORD,,