1

如果我有一个带有 a 的模型，ManyToManyField并且我想将其限制为具有特定属性的实例，那么最好的方法是什么？我可以在表单验证或视图中做到这一点，但我想更接近模型。

例如，如何只允许将 is_cool 设置为 True 的 B 类实例与 A 类实例关联？

from django.db import models

class A(models.Model):
    cool_bees = models.models.ManyToManyField('B') 

class B(models.Model):
    is_cool = models.BooleanField(default=False)
4

1 回答 1

3

为了更接近模型，您可以使用 m2m_changed 信号来检查模型是否符合您的要求，因此代码可能如下所示：

import django.db.models.signals

def validate(sender, instance, action, reverse, model, pk_set, **kwargs):
    if action == "pre_add":
        # if we're adding some A's that're not cool - a.cool_bees.add(b)
        if not reverse and model.objects.filter(pk__in=pk_set, is_cool=False):
              raise ValidationError("You're adding an B that is not cool")
        # or if we add using the reverse - b.A_set.add(a)
        if reverse and not instance.is_cool:
              raise ValidationError("You cannot insert this non-cool B to A")


signals.m2m_changed.connect(validate, sender=A.cool_bees.through)
于 2012-07-01T09:45:06.093 回答