From 5dd88353e93104c00fd9dc8d6ed46c878b763af4 Mon Sep 17 00:00:00 2001
From: zy7y <13271962515@163.com>
Date: Wed, 21 Sep 2022 15:25:52 +0800
Subject: [PATCH] =?UTF-8?q?ref=EF=BC=9A=E5=B0=86=E9=83=A8=E5=88=86?=
 =?UTF-8?q?=E5=8E=9F=E7=94=9FSQL=E6=9B=BF=E6=8D=A2=E6=88=90ORM=E7=9A=84API?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 backend/dbhelper/relation.py | 11 +----------
 backend/dbhelper/user.py     | 10 +---------
 2 files changed, 2 insertions(+), 19 deletions(-)

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(