Update main.py

main.py

@@ -173,57 +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:
-        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)
+        
+        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:
-        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)
+
+        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