How can I create sophisticated Django Model Validation for Django Admin?

佐手、 提交于 2019-12-10 17:09:11

问题


I have the following model in Django:

class Bout (models.Model):
    fighter_1 = models.ForeignKey(Fighter, related_name="bout_fighter_1")
    fighter_2 = models.ForeignKey(Fighter, related_name="bout_fighter_2")
    winner = models.ForeignKey(Fighter, related_name="bout_winner", 
        blank=True, null=True, help_text='Leave blank for draw.') 
    date = models.DateField()
    cancelled = models.BooleanField()

I would like to "idiot-proof" the administration for its records. Incidently, I want to create three rules:

  1. Fighter 1 is not the same as fighter 2 (which is only good for a monty python skit).

  2. Winner should be in the bout (i.e., either Fighter 1 or Fighter 2)

  3. The winner can't be set before the match takes place. (After all, this isn't WWE.)

All three of these rules require checking one field against another field in the same record. Is it possible to do this in django, either using native django methods or resorting to python?


回答1:


Short answer: you can achieve this in Django using "native django methods". I am not sure what exactly you mean by "native Django methods"; I am assuming that you mean making calls to the Django API.

There are a couple of ways to go about this. If your users can only create Bout instances using a form that you provide then the form's validation methods can test for the conditions you mentioned. For e.g.:

class BoutForm(forms.ModelForm):
    class Meta:
        model = Bout

    def clean(self):
        fighter_1 = self.cleaned_data.get('fighter_1')
        fighter_2 = self.cleaned_data.get('fighter_2')
        winner = self.cleaned_data.get('winner')  
        date = self.cleaned_data.get('date')

        if not (fighter_1 and fighter_2 and (fighter_1.id != fighter_2)):
            raise forms.ValidationError("Both fighters cannot be the same")

        if not (winner and (winner.id == fighter_1.id or winner.id == fighter_2.id)):
            raise forms.ValidationError("Winner is not in the fight")

        if not (date and date < datetime.today()):
            raise forms.ValidationError("Winner is not in the fight")

        return self.cleaned_data

The above snippet is incomplete. You can tweak it to meet your needs. Also take a look at Django's new fangled form validators.

If on the other hand your users can create instances using the API (say, by instantiating the Bout class in their programs) then you'll have to do the validation by overriding the save() method of the Bout class.




回答2:


While Manoj Govindan's answer looks very good, I also came up with my own solution... I'll include it here in case anyone finds my slightly shorter solution preferable:

def clean(self):
    if self.fighter_1 == self.fighter_2:
        raise ValidationError('Fighter 1 can not be Fighter 2.')
    if (self.winner != self.fighter_1) and (self.winner != self.fighter_2):
        raise ValidationError('Winner must be in the bout.')
    if (self.date >= datetime.date.today()) and (self.winner):
        raise ValidationError('Winner can not be set before match.')


来源:https://stackoverflow.com/questions/3541201/how-can-i-create-sophisticated-django-model-validation-for-django-admin

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!