[Update] 修复用户更新页面会清空用户public_key的问题

whq123-blog · Jan 20, 2020 · da6a0c2 · da6a0c2
1 parent 13a042b
commit da6a0c2
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/apps/users/serializers/user.py b/apps/users/serializers/user.py
@@ -74,8 +74,17 @@ def change_password_to_raw(attrs):
             attrs['password_raw'] = password
         return attrs
 
+    @staticmethod
+    def clean_auth_fields(attrs):
+        for field in ('password', 'public_key'):
+            value = attrs.get(field)
+            if not value:
+                attrs.pop(field, None)
+        return attrs
+
     def validate(self, attrs):
         attrs = self.change_password_to_raw(attrs)
+        attrs = self.clean_auth_fields(attrs)
         return attrs