项目初次提交

utils/response.py

@@ -0,0 +1,32 @@
+# 依赖安装：pip install orjson
+from fastapi.responses import ORJSONResponse as Response
+from fastapi import status as http_status
+from utils import status as http
+
+
+class SuccessResponse(Response):
+    """
+    成功响应
+    """
+    def __init__(self, data=None, msg="success", code=http.HTTP_SUCCESS, status=http_status.HTTP_200_OK, **kwargs):
+        self.data = {
+            "code": code,
+            "message": msg,
+            "data": data
+        }
+        self.data.update(kwargs)
+        super().__init__(content=self.data, status_code=status)
+
+
+class ErrorResponse(Response):
+    """
+    失败响应
+    """
+    def __init__(self, msg=None, code=http.HTTP_ERROR, status=http_status.HTTP_200_OK, **kwargs):
+        self.data = {
+            "code": code,
+            "message": msg,
+            "data": []
+        }
+        self.data.update(kwargs)
+        super().__init__(content=self.data, status_code=status)