Merge pull request #12 from GSI-Xapiens-CSIRO/fix/return-quota-data

Fix return if data does not exists
GSI-Xapiens-CSIRO · Dec 19, 2024 · f2653cb · f2653cb
2 parents 1b956b9 + 1361df1
commit f2653cb
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/lambda/dataPortal/quota_function.py b/lambda/dataPortal/quota_function.py
@@ -72,17 +72,13 @@ def store_quota(event, context):
 @router.attach("/dportal/quota/{userIdentity}", "get")
 def get_quota(event, context):
     uId = event["pathParameters"]["userIdentity"]
-
     try:
         myQuota = Quota.get(uId)
 
     except Quota.DoesNotExist:
-        raise PortalError(
-            error_code=409,
-            error_message="No data available here.",
-        )
+        return {'success': False, 'data': None}
 
-    return myQuota.to_dict()
+    return {'success': True, 'data': myQuota.to_dict()}
 
 @router.attach("/dportal/quota/{userIdentity}/increment_usagecount", "post")
 def increment_usagecount(event, context):