!14 【轻量级 PR】：update applications/view/admin/user.py.

Merge pull request !14 from Rabbit/N/A
2021-09-09 02:29:22 +00:00
parent d22eb22f74
commit c0a3f5243b
1 changed files with 3 additions and 8 deletions
@@ -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="更新成功")