diff --git a/backend/dbhelper/relation.py b/backend/dbhelper/relation.py index 24d9927..f2653e6 100644 --- a/backend/dbhelper/relation.py +++ b/backend/dbhelper/relation.py @@ -1,5 +1,3 @@ -from tortoise import connections - from dbhelper.menu import get_menu from models import RoleMenuModel, UserRoleModel from schemas import UserRole @@ -16,15 +14,8 @@ async def role_assigned_menu(data): if await get_menu({"id": mid}) is None: return mid - # todo 性能优化 - db = connections.get("default") # 1. 先把所有数据做删除 - await db.execute_query_dict( - """ - update sys_role_menu set status = 9 where rid = (?) - """, - [data.rid], - ) + 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] diff --git a/backend/dbhelper/user.py b/backend/dbhelper/user.py index 741c8cf..e979ec8 100644 --- a/backend/dbhelper/user.py +++ b/backend/dbhelper/user.py @@ -105,15 +105,7 @@ async def put_user(uid: int, data: UserPut): ) # 2. 将先有的数据标记 删除 - [ - await db.execute_query_dict( - """ - update sys_user_role set status = 9 where rid = (?) - """, - [role["id"]], - ) - for role in has_roles - ] + [await UserRoleModel.filter(rid=role["id"]).update(status=9) for role in has_roles] # 2. 新增次此更新的数据 await UserRoleModel.bulk_create(