我使用 django-reversion 来控制模型的变化,但我对模型清洁方法有问题。
例如,我的模型如下所示:
class Document(models.Model):
name = models.CharField(max_length=255, blank=True, null=True)
client = models.ForeignKey(Client, on_delete=models.CASCADE)
template = models.ForeignKey(Template, on_delete=models.CASCADE)
def clean(self):
if self.template.only_one and (Document.objects.filter(client=self.client, template=self.template).count() > 0):
raise ValidationError('Test')
我在 admin.py 中注册了这个模型:
@admin.register(Document)
class DocumentReversion(VersionAdmin):
pass
现在,我在管理部分中为文档创建一条记录,该记录引用了True
字段only_one
的模板。 如果我删除此记录并检索它,则会触发验证错误。 为什么?因为我删除了唯一可用的记录。很好奇...
对任何更新执行清理。如果您因此编辑Document
,那么它也将检查给定的条件。由于该条件成立:模板具有only_one=True
,并且有这样的文档(您的文档(,因此它将引发ValidationError
。
因此,您需要从查询集中排除当前项:
class Document(models.Model):
name = models.CharField(max_length=255, blank=True, null=True)
client = models.ForeignKey(Client, on_delete=models.CASCADE)
template = models.ForeignKey(Template, on_delete=models.CASCADE)
def clean(self):
qs = Document.objects.filter(client_id=self.client_id, template_id=self.template_id)
ifself.pk is not None:
qs = qs.exclude(pk=self.pk)
if self.template.only_one and qs.exists():
raise ValidationError('A document with this template already exists')
return super().clean()
如果这样设置了主键,我们可以从查询集中排除该主键。事实上,我们在这里不需要 if 条件,因为无论如何.exclude(pk=None)
总是会成功的。