rb-commons 0.3.2__py3-none-any.whl → 0.3.4__py3-none-any.whl

rb_commons/orm/managers.py

@@ -181,19 +181,28 @@ class BaseManager(Generic[ModelType]):
         return await self._smart_commit(obj)
 
     @with_transaction_error_handling
-    async def delete(self):
+    async def delete(self, instance: ModelType) -> bool:
         """
         Delete object(s) with flexible filtering options
+            - If `instance` is provided, delete that single instance.
+            - If `instance` is not provided, delete according to self.filters.
 
+        :arg instance: Model instance to delete.
         :return: Number of deleted records or None
         """
+
+        if instance is not None:
+            await self.session.delete(instance)
+            await self.session.commit()
+            return True
+
         self._ensure_filtered()
 
         try:
             delete_stmt = delete(self.model).where(and_(*self.filters))
-            result = await self.session.execute(delete_stmt)
+            await self.session.execute(delete_stmt)
             await self.session.commit()
-            return result.rowcount
+            return True
         except NoResultFound:
             return False
 
@@ -273,6 +282,36 @@ class BaseManager(Generic[ModelType]):
         await self.session.flush()
         return await self._smart_commit(instance)
 
+    @with_transaction_error_handling
+    async def lazy_save(self, instance: ModelType, load_relations: Sequence[str] = None) -> Optional[ModelType]:
+        self.session.add(instance)
+        await self.session.flush()
+        await self._smart_commit(instance)
+
+        if load_relations is None:
+            mapper = inspect(self.model)
+            load_relations = [rel.key for rel in mapper.relationships]
+
+        if not load_relations:
+            return instance
+
+        stmt = select(self.model).filter_by(id=instance.id)
+
+        for rel in load_relations:
+            stmt = stmt.options(selectinload(getattr(self.model, rel)))
+
+        result = await self.session.execute(stmt)
+        loaded_instance = result.scalar_one_or_none()
+
+        if loaded_instance is None:
+            raise NotFoundException(
+                message="Object saved but could not be retrieved with relationships",
+                status=404,
+                code="0001",
+            )
+
+        return loaded_instance
+
     async def is_exists(self, **kwargs) -> bool:
         stmt = select(self.model).filter_by(**kwargs)
         result = await self.session.execute(stmt)

{rb_commons-0.3.2.dist-info → rb_commons-0.3.4.dist-info}/METADATA

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

{rb_commons-0.3.2.dist-info → rb_commons-0.3.4.dist-info}/RECORD

@@ -11,14 +11,14 @@ 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=LyskLC802SanjtETgdw7vX1MwXPj2hzScGVnfGvDWj0,11273
+rb_commons/orm/managers.py,sha256=PyvJqlDo4eK_89XzTgrmOQByTTO88pPHyCdNA5bLLa8,12675
 rb_commons/permissions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rb_commons/permissions/role_permissions.py,sha256=3ZTwKclavAKSheO58fybo2uLDeTkd-iluTY7hUEjRPk,957
 rb_commons/schemes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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.3.2.dist-info/METADATA,sha256=4X0sjVssW-hdgOJZ6L9a95ftAvJD0_ORAPhIEDN9mhc,6570
-rb_commons-0.3.2.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-rb_commons-0.3.2.dist-info/top_level.txt,sha256=HPx_WAYo3_fbg1WCeGHsz3wPGio1ucbnrlm2lmqlJog,11
-rb_commons-0.3.2.dist-info/RECORD,,
+rb_commons-0.3.4.dist-info/METADATA,sha256=My_CBWU0ri8LI2MkU6eO7VCuI3g24pWoILzw1yZ3C4I,6570
+rb_commons-0.3.4.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+rb_commons-0.3.4.dist-info/top_level.txt,sha256=HPx_WAYo3_fbg1WCeGHsz3wPGio1ucbnrlm2lmqlJog,11
+rb_commons-0.3.4.dist-info/RECORD,,