Response封装

app/api/users/views.py

@@ -6,6 +6,7 @@ from sqlalchemy.orm import Session
 import models
 from app.api.users import crud, schemas
 from core.storage.db import SessionLocal
+from libs.res import HttpResultResponse, HttpMessage
 
 router = APIRouter()
 
@@ -60,11 +61,12 @@ def login():
     }
 
 
-@router.post("/create", response_model=schemas.User)
+@router.post("/create")
 def create_user(data: schemas.UserCreate, db: Session = Depends(get_db)):
     """添加用户"""
     db_user = crud.get_user_by_name(db, username=data.username)
     if db_user:
-        raise HTTPException(status_code=400, detail="用户已存在")
-    return crud.create_user(db=db, user=data)
+        return HttpResultResponse(code=400, msg=HttpMessage.USER_EXIST)
+    res = crud.create_user(db=db, user=data)
+    return HttpResultResponse(data=res.id)
 

libs/res.py

+import json
+from fastapi import status
+from fastapi.responses import Response
+
+
+class HttpMessage(object):
+    SUCCESS = "操作成功"
+    USER_EXIST = "用户已存在"
+
+
+class HttpResultResponse(Response):
+    def __init__(self, code=None, msg=None, **kwargs):
+        result_dict = {
+            'code': code if code else status.HTTP_200_OK,
+            'msg': msg if code else HttpMessage.SUCCESS
+        }
+
+        result_dict = {**result_dict, **kwargs}
+        super().__init__(
+            content=json.dumps(result_dict),
+            media_type="application/json",
+            status_code=result_dict.get('code')
+        )