zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: user

Browse Source
This commit is contained in:
wxg0103 2025-06-18 14:01:43 +08:00
parent 4205dc902f
commit b16d1d68dc
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
apps/users/serializers/user.py
View File

@ -202,13 +202,13 @@ class UserManageSerializer(serializers.Serializer):
user_role_relations = ( user_role_relations = (
user_role_relation_model.objects user_role_relation_model.objects
.filter(user_id__in=user_ids) .filter(user_id__in=user_ids)
.select_related('role_id') # 预加载外键数据 .select_related('role') # 预加载外键数据
) )
# 构建用户ID到角色名称列表的映射 # 构建用户ID到角色名称列表的映射
user_role_mapping = defaultdict(list) user_role_mapping = defaultdict(list)
for relation in user_role_relations: for relation in user_role_relations:
user_role_mapping[relation.user_id].append(relation.role_id.name) user_role_mapping[relation.user_id].append(relation.role.role_name)
return user_role_mapping return user_role_mapping
@ -450,7 +450,7 @@ class UserManageSerializer(serializers.Serializer):
'id': user_id, 'id': user_id,
'nick_name': relation.user.nick_name, 'nick_name': relation.user.nick_name,
'email': relation.user.email, 'email': relation.user.email,
'roles': [relation.role.name] 'roles': [relation.role.role_name]
} }
else: else:
user_dict[user_id]['roles'].append(relation.role.name) user_dict[user_id]['roles'].append(relation.role.name)