fix: 修复用户信息更新失败问题

lianweiping · Apr 25, 2024 · 972aac7 · 972aac7
1 parent 42cac24
commit 972aac7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/system/utils/signal_handler.py b/system/utils/signal_handler.py
@@ -174,7 +174,7 @@ def clean_cache_handler(sender, instance, **kwargs):
         logger.info(f"invalid cache {sender}")
 
     if issubclass(sender, UserInfo):
-        if {'roles', 'rules', 'dept', 'mode_type'} & set(update_fields):
+        if update_fields is None or {'roles', 'rules', 'dept', 'mode_type'} & set(update_fields):
             invalid_user_cache(instance.pk)
         else:
             cache_response.invalid_cache(f'UserInfoView_retrieve_{instance.pk}')

diff --git a/system/views/user/userinfo.py b/system/views/user/userinfo.py
@@ -21,6 +21,7 @@
 class UserInfoView(OwnerModelSet, UploadFileAction):
     serializer_class = UserInfoSerializer
     FILE_UPLOAD_FIELD = 'avatar'
+    choices_models = [UserInfo]
 
     def get_object(self):
         return self.request.user