rb-commons 0.4.8__py3-none-any.whl → 0.4.10__py3-none-any.whl

rb_commons/orm/managers.py

@@ -33,6 +33,7 @@ class BaseManager(Generic[ModelType]):
         self.data = None
         self.filters = []
         self._filtered = False
+        self._limit = None
 
     async def _smart_commit(self, instance: Optional[ModelType] = None) -> Optional[ModelType]:
         if not self.session.in_transaction():
@@ -176,6 +177,11 @@ class BaseManager(Generic[ModelType]):
     async def all(self, load_all_relations: bool = False) -> List[ModelType]:
         self._ensure_filtered()
         stmt = select(self.model).filter(and_(*self.filters))
+
+        if self._limit:
+            stmt = stmt.limit(self._limit)
+            self._limit = None
+
         return await self._execute_query_and_unique_data(stmt, load_all_relations)
 
     async def paginate(self, limit: int = 10, offset: int = 0, load_all_relations: bool = False) -> List[ModelType]:
@@ -183,6 +189,13 @@ class BaseManager(Generic[ModelType]):
         stmt = select(self.model).filter(and_(*self.filters)).limit(limit).offset(offset)
         return await self._execute_query_and_unique_data(stmt, load_all_relations)
 
+    def limit(self, value: int) -> 'BaseManager[ModelType]':
+        """
+        Set a limit on the number of results returned by queries like `all()` or `first()`.
+        """
+        self._limit = value
+        return self
+
     async def first(self, load_relations: Sequence[str] = None) -> Optional[ModelType]:
         """Return the first matching object, or None."""
         self._ensure_filtered()

{rb_commons-0.4.8.dist-info → rb_commons-0.4.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rb-commons
-Version: 0.4.8
+Version: 0.4.10
 Summary: Commons of project and simplified orm based on sqlalchemy.
 Home-page: https://github.com/RoboSell-organization/rb-commons
 Author: Abdulvoris

{rb_commons-0.4.8.dist-info → rb_commons-0.4.10.dist-info}/RECORD

@@ -11,7 +11,7 @@ rb_commons/http/consul.py,sha256=Ioq72VD1jGwoC96set7n2SgxN40olzI-myA2lwKkYi4,186
 rb_commons/http/exceptions.py,sha256=EGRMr1cRgiJ9Q2tkfANbf0c6-zzXf1CD6J3cmCaT_FA,1885
 rb_commons/orm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rb_commons/orm/exceptions.py,sha256=1aMctiEwrPjyehoXVX1l6ML5ZOhmDkmBISzlTD5ey1Y,509
-rb_commons/orm/managers.py,sha256=2nezz9ZNtvJIeaDt00K8QNNWwPtx21GwjIXmopWJKXo,15920
+rb_commons/orm/managers.py,sha256=Q5nNAOPeTJvxlsG_OBrkBXiY2kfOhzrncXNap960Pw4,16286
 rb_commons/orm/services.py,sha256=71eRcJ4TxZvzNz-hLXo12X4U7PGK54ZfbLAb27AjZi8,1589
 rb_commons/permissions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rb_commons/permissions/role_permissions.py,sha256=HL50s5RHW1nk2U3-YTESg0EDGkJDu1vN4QchcfK-L5g,988
@@ -19,7 +19,7 @@ rb_commons/schemes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuF
 rb_commons/schemes/jwt.py,sha256=F66JJDhholuOPPzlKeoC6f1TL4gXg4oRUrV5yheNpyo,1675
 rb_commons/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rb_commons/utils/media.py,sha256=KNY_9SdRa3Rp7d3B1tZaXkhmzVa65RcS62BYwZP1bVM,332
-rb_commons-0.4.8.dist-info/METADATA,sha256=8-1yWgE9JHuRtXZoEGfi6jdbpE1LOheL-iB7ciygrpw,6570
-rb_commons-0.4.8.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-rb_commons-0.4.8.dist-info/top_level.txt,sha256=HPx_WAYo3_fbg1WCeGHsz3wPGio1ucbnrlm2lmqlJog,11
-rb_commons-0.4.8.dist-info/RECORD,,
+rb_commons-0.4.10.dist-info/METADATA,sha256=PUkyY0LiPIUcknO3_-4i_7rL4Yci4v-GYaCLn5FiVh4,6571
+rb_commons-0.4.10.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+rb_commons-0.4.10.dist-info/top_level.txt,sha256=HPx_WAYo3_fbg1WCeGHsz3wPGio1ucbnrlm2lmqlJog,11
+rb_commons-0.4.10.dist-info/RECORD,,