From c0a3f5243b5762a766823647647a2a296d156d1f Mon Sep 17 00:00:00 2001 From: Rabbit <9531183+rabbit_test@user.noreply.gitee.com> Date: Thu, 9 Sep 2021 02:29:22 +0000 Subject: [PATCH] =?UTF-8?q?!14=20=E3=80=90=E8=BD=BB=E9=87=8F=E7=BA=A7=20PR?= =?UTF-8?q?=E3=80=91=EF=BC=9Aupdate=20applications/view/admin/user.py.=20M?= =?UTF-8?q?erge=20pull=20request=20!14=20from=20Rabbit/N/A?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- applications/view/admin/user.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/applications/view/admin/user.py b/applications/view/admin/user.py index f2556ba..2a6d9e0 100644 --- a/applications/view/admin/user.py +++ b/applications/view/admin/user.py @@ -119,15 +119,10 @@ def update(): role_ids = a.split(',') User.query.filter_by(id=id).update({'username': username, 'realname': real_name, 'dept_id': dept_id}) u = User.query.filter_by(id=id).first() - roles_id = [] - for role in u.role: - roles_id.append(role.id) - roles = Role.query.filter(Role.id.in_(roles_id)).all() - for r in roles: - u.role.remove(r) + roles = Role.query.filter(Role.id.in_(role_ids)).all() - for r in roles: - u.role.append(r) + u.role = roles + db.session.commit() return success_api(msg="更新成功")