[Bugfix] 修复小bug

pull/1147/head
ibuler 2018-03-28 14:35:27 +08:00
parent 5f0b4a4b63
commit 829f57e2d7
1 changed files with 2 additions and 4 deletions

View File

@ -16,13 +16,11 @@ class AuthSerializer(serializers.ModelSerializer):
return private_key, public_key
def save(self, **kwargs):
password = self.validated_data.pop('password') or None
private_key = self.validated_data.pop('private_key') or None
password = self.validated_data.pop('password', None) or None
private_key = self.validated_data.pop('private_key', None) or None
self.instance = super().save(**kwargs)
if password or private_key:
private_key, public_key = self.gen_keys(private_key, password)
self.instance.set_auth(password=password, private_key=private_key,
public_key=public_key)
return self.instance