From de45d8dc19cd16b85344a3c3bbb243a12e8811a0 Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期三, 13 十一月 2024 15:16:59 +0800 Subject: [PATCH] 增加功能 --- app/api/role.py | 24 +++++++++++++----------- 1 files changed, 13 insertions(+), 11 deletions(-) diff --git a/app/api/role.py b/app/api/role.py index 1d54a6f..516619a 100644 --- a/app/api/role.py +++ b/app/api/role.py @@ -13,7 +13,8 @@ @role_router.get("/list", response_model=Response) async def user_group_list(paras: PageParameter, current_user: UserModel = Depends(get_current_user), db=Depends(get_db)): - return Response(code=200, msg="", data=await role_list(db, paras.page_size, paras.page_index, paras.keyword)) + + return Response(code=200, msg="", data=await role_list(db, paras.page_size, paras.page_index, paras.keyword, current_user.permission, current_user.id)) @role_router.post("/add_role", response_model=Response) @@ -23,7 +24,7 @@ db_role = db.query(RoleModel).filter(RoleModel.name == role.roleName).first() if db_role: return Response(code=200, msg="role already created") - is_create = await create_role(db, role.roleName, role.remark, current_user.id) + is_create = await create_role(db, role.roleName, role.remark, role.roleKey, role.dataScope,current_user.id) if not is_create: return Response(code=500, msg="role create failure", data={}) return Response(code=200, msg="role create successfully", data={"roleName": role.roleName}) @@ -53,16 +54,17 @@ @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.editType == 1: + 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") - 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) + 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.roleKey, role.dataScope, role.resources, role.editType) if not is_edit: 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