AI butler Jarvis

example_services/demo_service2/response.py

@@ -0,0 +1,34 @@
+from typing import Union, Tuple, Dict, List
+from pydantic import BaseModel
+
+
+class Response(BaseModel):
+    code: int
+    message: str
+    data: Union[Dict, List]
+
+
+def build_success_response(data: Union[Dict, List]):
+    return {
+        "code": RESPONSE_SUCCESS[0],
+        "message": RESPONSE_SUCCESS[1],
+        "data": data
+    }
+
+
+def build_failure_response(result_tuple: Tuple, data: Union[Dict, List]):
+    return build_failure_response(result_tuple[0], result_tuple[1], data)
+
+
+def build_failure_response(code: int, message: str, data: Union[Dict, List]):
+    return {
+        "code": code,
+        "message": message,
+        "data": data
+    }
+
+
+RESPONSE_SUCCESS = 200, "SUCCESS"
+RESPONSE_UNKNOWN_ERROR = 100001, "UNKNOWN ERROR"
+RESPONSE_UNAUTHORIZED = 100002, "UNAUTHORIZED"
+RESPONSE_TASK_NOT_FOUND = 100003, "TASK NOT FOUND"