diff --git a/backend/dbhelper/relation.py b/backend/dbhelper/relation.py deleted file mode 100644 index f2653e6..0000000 --- a/backend/dbhelper/relation.py +++ /dev/null @@ -1,22 +0,0 @@ -from dbhelper.menu import get_menu -from models import RoleMenuModel, UserRoleModel -from schemas import UserRole - - -async def user_assigned_role(data: UserRole): - """给用户分配角色""" - return await UserRoleModel.create(**data.dict()) - - -async def role_assigned_menu(data): - """给角色分配菜单""" - for mid in data.menus: - if await get_menu({"id": mid}) is None: - return mid - - # 1. 先把所有数据做删除 - await RoleMenuModel.filter(rid=data.rid).update(status=9) - # 2. 新增数据 - await RoleMenuModel.bulk_create( - [RoleMenuModel(rid=data.rid, mid=mid) for mid in data.menus] - )