qena-shared-lib 0.1.19__py3-none-any.whl → 0.1.21__py3-none-any.whl

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/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/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",

{qena_shared_lib-0.1.19.dist-info → qena_shared_lib-0.1.21.dist-info}/METADATA

@@ -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.dist-info → qena_shared_lib-0.1.21.dist-info}/RECORD

@@ -10,7 +10,7 @@ qena_shared_lib/enums.py,sha256=qYOZUD8Or0O1uGoNRShMp-SzmkXDQ31djMVTG_yGxL4,107
 qena_shared_lib/eventbus.py,sha256=Qk4TiJHGlnUNQkLTIWEuHe3MPyRmwUmkjbk5Y4dNSxE,11501
 qena_shared_lib/exception_handling.py,sha256=oiPVaxD74y-dbB8hJsncPNd-Fphot60Q60tZQTbCZ50,13392
 qena_shared_lib/exceptions.py,sha256=pByxZnxesxLVyHF2ARd0C0ioWLYZZVOtxPytJnF5EDo,14567
-qena_shared_lib/http/__init__.py,sha256=nZmSw7Al8OawmDPsQ4i33DCbLOEcrSY7gMo1JArt7Ug,2014
+qena_shared_lib/http/__init__.py,sha256=F6CqzPsRx55fa5ozRkk2lSNJu28KPw9uI8zQwoo4lVU,2055
 qena_shared_lib/http/_base.py,sha256=5jPAjiDmJIxjtMdhxNHpcTM_xbXC5eBMvy3U0tpB25Q,25199
 qena_shared_lib/http/_exception_handlers.py,sha256=fkxW7KqyCy923f4f6s1qmd8gO-Oaoi-mLR9gaIWC-PU,6265
 qena_shared_lib/http/_request.py,sha256=kppCnS5zZfvKl-hNO6ysqkVdtoXshFp6Ue_0O5apqXQ,426
@@ -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=RH0wa0XopJlFdQVZdZVyknnh26vsLRv-rKR-2t_cGD0,16630
+qena_shared_lib/mongodb.py,sha256=J39ttzyhL1umxIT9lQvkUOV_1uHYZQcQvrYrO9Okr8I,17109
 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
@@ -31,7 +31,7 @@ qena_shared_lib/rabbitmq/_listener.py,sha256=BwZ3YVTGI2sIMJWMoL7SnYN_RGfjBr1tYxa
 qena_shared_lib/rabbitmq/_pool.py,sha256=eEJswSoWoWzjh290doFYccn1cwQW6DgbdpHeU-7nZSg,2006
 qena_shared_lib/rabbitmq/_publisher.py,sha256=57WruxVlZemA7XVDNHSNB3_2gI-ycIKDobepLGa_QoM,2511
 qena_shared_lib/rabbitmq/_rpc_client.py,sha256=DKYvyPnmaMfa8Ymw1nLKinuFeG_H0Lc1OL8EUjN8M6U,9353
-qena_shared_lib/rabbitmq/message/__init__.py,sha256=HsTg8uzDFUVAnAa-HYHdl0dSwZ2gZXYrKjp7IgAXSR8,398
+qena_shared_lib/rabbitmq/message/__init__.py,sha256=J0wKjSuwOC7urUffWdxGZ7x_F9S5S6EryC9NxuysOhY,439
 qena_shared_lib/rabbitmq/message/_inbound.py,sha256=v744Vr4JZn9iE4GnD8uoQQLiD9TuKGZgLIiMwIcLGp4,272
 qena_shared_lib/rabbitmq/message/_outbound.py,sha256=axf7OF2rA05ixdjM3PiHG33I3dJU2gS8GP-ooefX4Jk,274
 qena_shared_lib/redis.py,sha256=N15rYGX5iMbFvFzCnSTtOBD_EIrdZ9xkoS1Sw9s-3tI,1262
@@ -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.19.dist-info/WHEEL,sha256=k57ZwB-NkeM_6AsPnuOHv5gI5KM5kPD6Vx85WmGEcI0,78
-qena_shared_lib-0.1.19.dist-info/METADATA,sha256=bUaVtmWTCPymqMS9MqOdXfE48Xi3Xrww_eNaKrcQowQ,19281
-qena_shared_lib-0.1.19.dist-info/RECORD,,
+qena_shared_lib-0.1.21.dist-info/WHEEL,sha256=k57ZwB-NkeM_6AsPnuOHv5gI5KM5kPD6Vx85WmGEcI0,78
+qena_shared_lib-0.1.21.dist-info/METADATA,sha256=kAbusp4enrIcpkobDr6LX6wna8JGcYjCtz35Q9LAs1E,19281
+qena_shared_lib-0.1.21.dist-info/RECORD,,