rb-commons 0.5.0__tar.gz → 0.5.1__tar.gz

{rb_commons-0.5.0 → rb_commons-0.5.1}/PKG-INFO

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

{rb_commons-0.5.0 → rb_commons-0.5.1}/rb_commons/orm/managers.py

@@ -267,6 +267,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
+
     @with_transaction_error_handling
     async def update(self, instance: ModelType, **fields):
         if not fields:

{rb_commons-0.5.0 → rb_commons-0.5.1}/rb_commons.egg-info/PKG-INFO

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

{rb_commons-0.5.0 → rb_commons-0.5.1}/setup.py

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