Update main.py

main.py

@@ -173,55 +173,57 @@ async def endpoint_output(endpoint_result: ResultItem, code: int = 0, error: str
 # Caching Post requests is challenging
 async def sms_posts(instance: ResultItem, idx: str = None, action: str = "add") -> Union[ErrorResponse, EndpointResponse]:
     async with AsyncSession(sms_resource["engine"]) as session:
-        code = 1
-        error = None
-        result = None
-        existing = await session.get(instance.__class__, idx)
-
-        # For add action, do db operation if instance is not existing. Other actions, do db operation if instance exists in db
-        checker = existing is None if action == "add" else existing is not None
-
-        try:
-            if checker:
-                if action == "delete":
-                    await session.delete(existing)  # Asynchronous
-                    await session.commit()                    
-                else:  # add or update use add
-                    session.add(instance)  # Not asynchronous
-                    await session.commit()
-                    await session.refresh(instance)
-                    result = instance
-        except Exception as e:
-            code = 0
-            error = str(e)
-
-        finally:
-            return await endpoint_output(result, code, error)
+        async with session.begin():
+            code = 1
+            error = None
+            result = None
+            existing = await session.get(instance.__class__, idx)
+    
+            # For add action, do db operation if instance is not existing. Other actions, do db operation if instance exists in db
+            checker = existing is None if action == "add" else existing is not None
+    
+            try:
+                if checker:
+                    if action == "delete":
+                        await session.delete(existing)  # Asynchronous
+                        await session.commit()                    
+                    else:  # add or update use add
+                        session.add(instance)  # Not asynchronous
+                        await session.commit()
+                        await session.refresh(instance)
+                        result = instance
+            except Exception as e:
+                code = 0
+                error = str(e)
+    
+            finally:
+                return await endpoint_output(result, code, error)
 
 
 # @cache(expire=ONE_DAY_SEC, namespace='sms_gets')  # Cache for 1 day
 async def sms_gets(sms_class: Type[Result], action: str = "first", idx: str = None, stmt: SelectOfScalar[Type[Result]] = None) -> Union[ErrorResponse, EndpointResponse]:
     async with AsyncSession(sms_resource["engine"]) as session:
-        result = None
-        error = None
-        code = 1
-        try:
-            if action == "all":
-                statement = select(sms_class) if stmt is None else stmt
-                instance_list = (await session.exec(statement)).all()
-                if instance_list:
-                    result = {
-                        str(instance.id): instance for instance in instance_list}
-            elif action == "first":
-                statement = select(sms_class).where(
-                    sms_class.id == idx) if stmt is None else stmt
-                result = (await session.exec(statement)).first()
-
-        except Exception as e:
-            code = 0
-            error = str(e)
-        finally:
-            return await endpoint_output(result, code, error)
+        async with session.begin():
+            result = None
+            error = None
+            code = 1
+            try:
+                if action == "all":
+                    statement = select(sms_class) if stmt is None else stmt
+                    instance_list = (await session.exec(statement)).all()
+                    if instance_list:
+                        result = {
+                            str(instance.id): instance for instance in instance_list}
+                elif action == "first":
+                    statement = select(sms_class).where(
+                        sms_class.id == idx) if stmt is None else stmt
+                    result = (await session.exec(statement)).first()
+    
+            except Exception as e:
+                code = 0
+                error = str(e)
+            finally:
+                return await endpoint_output(result, code, error)
 
 
 # Student Routes