当前位置

首页 > 博客 > qyt的博客

Django REST framework serializer 跟 model 无关的只写（write_only） field 的处理

qyt 在 周五, 2016-04-22 23:13 提交

有时候想在 DRF 的 api 里塞一些跟 model 无关的，但又是必须要处理的数据，比如设密码时，为了保证输入无误第二次输入的 password，这种信息一般设为只写。
这个时候 serializers.py 里，把 validate 方法重载一下，在里面删掉 request.data 里实际上 model 不需要的数据就 ok 了。

  1. class AccountListSerializer(serializers.HyperlinkedModelSerializer):
  2.     password = serializers.CharField(write_only=True, style={'input_type': 'password'})
  3.     confirm_password = serializers.CharField(write_only=True, style={'input_type': 'password'})
  4.     def validate(self, data):
  5.         del data['confirm_password']
  6.         return data
  7.     class Meta:
  8.         model = Account
  9.         fields = ('password', 'confirm_password')

当然，在 views.py 里，需要校验一下 password 和 confirm_password 是否输入了，如果输入了，是否一致

  1. if not request.data.get('password') or not request.data.get('confirm_password'):
  2.     raise serializers.ValidationError("Please input password and confirm it!")
  3. if request.data.get('password') != request.data.get('confirm_password'):
  4.     raise serializers.ValidationError("Passwords do NOT match!")
Topic: 
技术