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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: qena-shared-lib
-Version: 0.1.21
+Version: 0.1.23
 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.21 → qena_shared_lib-0.1.23}/pyproject.toml

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

{qena_shared_lib-0.1.21 → qena_shared_lib-0.1.23}/src/qena_shared_lib/http/_exception_handlers.py

@@ -94,20 +94,12 @@ class HTTPServiceExceptionHandler(
 
             if exception.response_code is not None:
                 content["code"] = exception.response_code
-                str_response_code = str(exception.response_code)
-                extra["responseCode"] = str_response_code
-
-                tags.append(str_response_code)
 
             if exception.corrective_action is not None:
                 content["correctiveAction"] = exception.corrective_action
 
             if exception.status_code is not None:
                 status_code = exception.status_code
-                str_status_code = str(status_code)
-                extra["statusCode"] = str_status_code
-
-                tags.append(str_status_code)
 
             if exception.headers is not None:
                 headers = exception.headers

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

@@ -33,8 +33,8 @@ from .alias import CamelCaseAliasedBaseModel
 from .logging import LoggerFactory
 
 __all__ = [
-    "AggregatedDocument",
     "ASCENDING",
+    "AggregatedDocument",
     "AsyncClientSession",
     "DESCENDING",
     "Document",
@@ -47,6 +47,7 @@ __all__ = [
     "IndexModel",
     "MongoDBManager",
     "MongoDBObjectId",
+    "ObjectId",
     "ProjectedDocument",
     "RepositoryBase",
     "TEXT",
@@ -587,16 +588,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)