Django 管理器覆盖

Question

我有一道选择题，其中模型 Question 是问题，Choices 是答案。我想将可以创建的 Choices 的数量限制为 Question 到 4。

一个models.Manager用来验证题目的选项数。

class Question(models.Model):
    QUESTION_TYPES = (
    ('MC', 'Multiple Choice'),
    ('SB', 'Subjective'),
    )
    question_type = models.CharField(choices=QUESTION_TYPES, max_length=2, default='MC')
    question_text = models.TextField(null=False, blank=False)

和Choice

class Choice(models.Model):
    choice_text = models.CharField(max_length=100, null=True)
    question= models.ForeignKey(Question, null=True , related_name='choices')
    is_answer = models.BooleanField(default=False)
    objects = ChoiceManager()

自定义Manager

class ChoiceManager(models.Manager):
    def create(self, **kwargs):
        question = kwargs.pop('question',None)
        if question is not None:
            if question.choices.all().count() > 4:    # see related_name 
                raise AssertionError
            else:
                return self

如果我使用 python shell 创建模型实例，一切正常。

但是： 当我使用 AdminSite 时。我可以为一个问题创建 4 个以上的选择。我如何在 AdminSite 上获得我想要的行为（在管理站点上引发错误）？在管理员级别覆盖经理？我将如何进行？

admin.site.register(Question)
admin.site.register(Choice)

Answer 1

需要覆盖模型 class 本身中的 save 方法。

class Choice(models.Model):
    choice_text = models.CharField(max_length=100, null=True)
    question= models.ForeignKey(Question, null=True , related_name='choices')
    is_answer = models.BooleanField(default=False)
    objects = ChoiceManager()

    def save(self, *args, **kwargs):
        if self.question.choices.all().count() > 4:
            print "You Shall Not Save"
            raise ValueError
        else:
            print "Super Method Called"
            super(Choice, self).save(*args, **kwargs)

Django 管理器覆盖

Django Admin Manager Override

python

django

django-admin

django-managers