From 2669d0e0a93e06983dade724146bd0e1b6da6ba7 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期二, 12 十一月 2024 18:42:05 +0800
Subject: [PATCH] 增加接口

---
 app/api/role.py |   18 ++++++++++--------
 1 files changed, 10 insertions(+), 8 deletions(-)

diff --git a/app/api/role.py b/app/api/role.py
index dfcc7bb..1d54a6f 100644
--- a/app/api/role.py
+++ b/app/api/role.py
@@ -25,7 +25,7 @@
         return Response(code=200, msg="role already created")
     is_create = await create_role(db, role.roleName, role.remark, current_user.id)
     if not is_create:
-        return Response(code=200, msg="role create failure", data={})
+        return Response(code=500, msg="role create failure", data={})
     return Response(code=200, msg="role create successfully", data={"roleName": role.roleName})
 
 @role_router.delete("/delete_role/{role_id}", response_model=Response)
@@ -35,7 +35,7 @@
         return Response(code=200, msg="role does not exist")
     is_edit = await delete_role_data(db, role_id)
     if not is_edit:
-        return Response(code=200, msg="role delete failure", data={})
+        return Response(code=500, msg="role delete failure", data={})
     return Response(code=200, msg="role delete successfully", data={})
 
 
@@ -48,19 +48,21 @@
         return Response(code=200, msg="role does not exist")
     is_edit = await edit_role_status(db, role.status,role.role_id)
     if not is_edit:
-        return Response(code=200, msg="role status edit failure", data={})
+        return Response(code=500, msg="role status edit failure", data={})
     return Response(code=200, msg="role status edit successfully", data={})
 
 @role_router.put("/edit_role", response_model=Response)
 async def edit_role_data(role: RoleEdit, current_user: UserModel = Depends(get_current_user), db=Depends(get_db)):
+    if not role.roleName:
+        return Response(code=400, msg="The roleName cannot be empty!")
     db_role = db.query(RoleModel).filter(RoleModel.id == role.role_id).first()
     if not db_role:
         return Response(code=200, msg="role does not exist")
-    if role.roleName:
-        db_role = db.query(RoleModel).filter(RoleModel.name == role.roleName).first()
-        if db_role:
-            return Response(code=200, msg="role already created")
+
+    db_role = db.query(RoleModel).filter(RoleModel.name == role.roleName).first()
+    if db_role and db_role.id != role.role_id:
+        return Response(code=200, msg="role already created")
     is_edit = await edit_role_resource(db, role.role_id,role.roleName, role.remark, role.resources)
     if not is_edit:
-        return Response(code=200, msg="role edit failure", data={})
+        return Response(code=500, msg="role edit failure", data={})
     return Response(code=200, msg="role edit successfully", data={})
\ No newline at end of file

--
Gitblit v1.8.0