繁体   English   中英

Django Admin Manager替代

[英]Django Admin Manager Override

我有一个多项选择题,其中模型Question是问题和Choices作为答案。 我想将一个Question可以创建的Choices数量限制为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) 

需要重写模型类本身中的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)

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM