From 743b04aebff50e5b3282e3d4f6f2bc48fcbe1595 Mon Sep 17 00:00:00 2001 From: wxg0103 <727495428@qq.com> Date: Wed, 28 May 2025 17:45:02 +0800 Subject: [PATCH] refactor: user role --- apps/users/serializers/user.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/apps/users/serializers/user.py b/apps/users/serializers/user.py index 398b08d29..29699b1ba 100644 --- a/apps/users/serializers/user.py +++ b/apps/users/serializers/user.py @@ -373,6 +373,7 @@ class UserManageSerializer(serializers.Serializer): users = User.objects.filter(id__in=user_ids).values('id', 'nick_name') return list(users) + def update_user_role(instance, user): workspace_user_role_mapping_model = DatabaseModelManage.get_model("workspace_user_role_mapping") role_setting_model = DatabaseModelManage.get_model("role_model") @@ -386,7 +387,7 @@ def update_user_role(instance, user): role_ids = {role_id for item in role_setting for role_id in item} role_ids_is_system = role_setting_model.objects.filter(id__in=role_ids, - type='SYSTEM_ADMIN').values_list( + type=RoleConstants.ADMIN.name).values_list( 'id', flat=True) if role_ids_is_system: relations = {(role_id, 'SYSTEM') if role_id in role_ids_is_system else (role_id, workspace_id) @@ -398,4 +399,3 @@ def update_user_role(instance, user): workspace_id=workspace_id, user_id=user.id ) -