在 django-rest-framework 中验证模型

Question

我有一个带有自定义验证代码的 django 模型。当我使用 django-rest-framework 尝试创建/更新模型时，验证代码确实会运行，但不是输出一些带有错误内容的 JSON，而是失败并带有 ValidationError。为什么 django-rest-framework 没有捕捉到这个错误？

models.py

class MyModel(models.Model):
  is_default = models.BooleanField()
  type = models.CharField(max_length=64, choices=[("a","a"), ("b","b"), ("c","c")]

  def clean(self, *args, **kwargs):
    # there can be only 1 "default" model per type
    if self.is_default:
      other_models = MyModel.objects.filter(
        type=self.type, 
        default=True).exclude(pk=self.pk)
      if other_models.count() != 0:
        raise ValidationError({"default": "There can be only one default model per type.")
    super(MyModel, self).clean(*args, **kwargs)

serializers.py

class MyModelSerializer(ModelSerializer):

    class Meta:
        model = MyModel
        fields = ('id', 'default', 'type')

当我尝试发布 type="d" 的 JSON 数据时，我正确地得到以下响应：{ "type": ["this is not one of the valid choices"]}。

但是当我尝试在 default=true 的情况下 POST JSON 数据（并且数据库中已经存在相同类型的默认模型）时，我只是得到了 ValidationError 而不是格式良好的 JSON。

Answer 1

第一个验证错误有效，因为 DRFs 引发了它。

只有 DRF 的子类拥有 APIException 在框架内引发。所以你需要使用它的APIException 类来引发错误。

例如，

from rest_framework.exceptions import APIException

class CustomException(APIException):

    status_code = 400
    default_detail = 'There can be only one default model per type.'

然后你可以在你的模型中做到这一点。

 if other_models.count() != 0:
     raise CustomException()

附带说明您为什么要尝试使用代码在模型上创建unique together 约束？您应该使用unique_together 在数据库级别执行此操作。

例如，在模型上你可以做这样的事情......

 class Meta:
     unique_together = ("is_default", "type")

一旦你运行了迁移，它会认为它们在一起时是独一无二的！