DRF 和 Django 管理员的通用自定义验证

Question

我有一个用户表，其中必须对员工用户设置唯一的电子邮件限制，对非员工用户设置唯一的电话限制。现在我想在 DRF API 和 Django 管理员中维护这个约束，如何在不重复代码的情况下实现这一点。

正如我从 DRF 3.0 开始看到的那样，所有验证都是为序列化程序显式执行的（如果我覆盖了 unique_validation 或 clean 方法来执行条件验证，我也必须为序列化程序单独编写这个自定义验证）所以如何编写适用于 DRF API 和 Django 管理员的自定义验证，无需重复代码？

Answer 1

如果您的模型没有嵌套关系，这应该可以工作。

class ValidateModelMixin(object)
    def validate(self, attrs):
        attrs = super().validate(attrs)
        obj = self.Meta.model(**attrs)
        obj.clean()
        return attrs

class SomeModelSerializer(ValidateModelMixin, serializers.ModelSerializer):
    #...
    class Meta:
        model = SomeModel

但就我而言，我有嵌套模型，我需要覆盖序列化程序中的 create() 和 update() 方法。我将验证保存在一个地方的解决方案：

def create(self, validated_data):
        # Remove nested and M2m relationships from validated_data
        firstmodel_set = validated_data.pop('firstmodel_set') if 'firstmodel_set' in validated_data else []
        .....
        # Get instance of project
        instance = Project(**validated_data)
        instance.clean()

        # If clean method doesn't raise any exception, create.
        project = instance.save()
        ......

如果你想序列化验证消息..也许这样的东西会帮助你......

def validate_instance(instance):
    try:
        # Call model validation
        instance.clean()
    except ValidationError as e:
        raise serializers.ValidationError(e)

或者您可以使用 Signals 并在 pre_save() 上进行验证。 看看https://docs.djangoproject.com/en/1.10/topics/signals/

Answer 2

对于 API： 为 UserRegistration 创建一个序列化程序，并使用两个条件实现验证方法，一个用于唯一电子邮件，另一个用于电话号码。 类似于：

class UserCreateSerializer(ModelSerializer):
   class Meta:
        model = User
        fields = ['email','phone','...']
   def validate(self, data):
        """user email phone validator """

        user_qs = User.objects.filter(email=data['email'], is_staff=True)
        if user_qs.exists():
            raise ValidationError({"email":"This user has already registered."})
        user_phone = User.objects.filter(phone=data['phone'], is_active=True, is_staff=False)
        if user_phone.exists():
            raise ValidationError({"phone":"Phone number already exists."})
        return data

网页版： 如果您已经创建了注册表单，那么在您的表单中实现 clean 方法作为序列化程序中的验证，上面。

class UserCreateForm(ModelForm):

    class Meta:
            model = User
            fields = ['email','phone','...']
    def clean(self):
            """user email phone validator """

            user_qs = User.objects.filter(email=self.cleaned_data['email'], is_staff=True)
            if user_qs.exists():
                raise ValidationError({"email":"This user has already registered."})
            user_phone = User.objects.filter(phone=self.cleaned_data['phone'], is_active=True, is_staff=False)
            if user_phone.exists():
                raise ValidationError({"phone":"Phone number already exists."})
            return data