23 lines
647 B
Python
23 lines
647 B
Python
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]
|
|
)
|