rb-commons 0.4.5__tar.gz → 0.4.7__tar.gz

{rb_commons-0.4.5 → rb_commons-0.4.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rb-commons
-Version: 0.4.5
+Version: 0.4.7
 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.5 → rb_commons-0.4.7}/rb_commons/orm/managers.py

@@ -166,18 +166,23 @@ class BaseManager(Generic[ModelType]):
         if not self._filtered:
             raise RuntimeError("You must call `filter()` before using this method.")
 
-    async def all(self, load_all_relations: bool = False) -> List[ModelType]:
-        self._ensure_filtered()
-
-        stmt = select(self.model).filter(and_(*self.filters))
+    async def _execute_query_and_unique_data(self, stmt, load_all_relations: bool) -> List[ModelType]:
         stmt = self._apply_eager_loading(stmt, load_all_relations)
-
         result = await self.session.execute(stmt)
         rows = result.scalars().all()
-
         unique_by_pk = {obj.id: obj for obj in rows}
         return list(unique_by_pk.values())
 
+    async def all(self, load_all_relations: bool = False) -> List[ModelType]:
+        self._ensure_filtered()
+        stmt = select(self.model).filter(and_(*self.filters))
+        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]:
+        self._ensure_filtered()
+        stmt = select(self.model).filter(and_(*self.filters)).limit(limit).offset(offset)
+        return await self._execute_query_and_unique_data(stmt, load_all_relations)
+
     async def first(self, load_relations: Sequence[str] = None) -> Optional[ModelType]:
         """Return the first matching object, or None."""
         self._ensure_filtered()
@@ -382,6 +387,40 @@ class BaseManager(Generic[ModelType]):
         if not self.session.in_transaction():
             await self.session.commit()
 
+    @with_transaction_error_handling
+    def has_relation(self, relation_name: str) -> 'BaseManager[ModelType]':
+        """
+           Check if a relationship exists between models using an EXISTS subquery.
+           
+           :param relation_name Name of the relationship to check. Must be a valid relationship
+                   defined in the model.
+
+           :return BaseManager[ModelType]: Self instance for method chaining.
+
+           :raise DatabaseException: If there's an error constructing the subquery.
+           :raise InternalException: If there's an unexpected error in relationship handling.
+           
+           Notes:
+               - The relationship must be properly defined in the SQLAlchemy model
+               - Uses a non-correlated EXISTS subquery for better performance
+               - Silently continues if the relationship doesn't exist
+           """
+        # Get the relationship property
+        relationship = getattr(self.model, relation_name)
+
+        # Create subquery using select
+        subquery = (
+            select(1)
+            .select_from(relationship.property.mapper.class_)
+            .where(relationship.property.primaryjoin)
+            .exists()
+        )
+
+        # Add the exists condition to filters
+        self.filters.append(subquery)
+
+        return self
+
     def model_to_dict(self, instance: ModelType, exclude: set = None):
         exclude = exclude or set()
 

{rb_commons-0.4.5 → rb_commons-0.4.7}/rb_commons.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rb-commons
-Version: 0.4.5
+Version: 0.4.7
 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.5 → rb_commons-0.4.7}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="rb-commons",
-    version="0.4.5",
+    version="0.4.7",
     author="Abdulvoris",
     author_email="erkinovabdulvoris101@gmail.com",
     description="Commons of project and simplified orm based on sqlalchemy.",