Django - как заставить пользователя выбрать определенное количество опций в checbox modelform?

Я делаю простой опрос, построенный на основе modelform:

models.py

class Fruits(models.Model):
     author = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE)
     title = models.CharField(max_length=200, default='')
     text = models.TextField(default='')
     banana = models.BooleanField(default=False)
     grape = models.BooleanField(default=False)
     apple = models.BooleanField(default=False)
     mango = models.BooleanField(default=False)

forms.py

class FruitPicker(ModelForm):
    class Meta:
        model = Fruits
        fields = [
        'title',
        'text',
        'banana',
        'grape',
        'apple',
        'mango'
    ]
    
    widgets = {
        'text': forms.Textarea(attrs={"style": "height:10em;" "width:60em;"}),
        'banana': forms.CheckboxInput(attrs={"style": "margin-left:350px;"}),
        'grape': forms.CheckboxInput(attrs={"style": "margin-left:350px;"}),
        'apple': forms.CheckboxInput(attrs={"style": "margin-left:350px;"}),
        'mango': forms.CheckboxInput(attrs={"style": "margin-left:350px;"})
    }

Допустим, я хочу, чтобы пользователь выбрал минимальное количество фруктов - 2, а максимальное - 3. Как мне сделать это на задней панели?

Вы можете переопределить метод clean формы, чтобы проверить, проверяется ли от двух до трех BooleanField:

from django.core.exceptions import ValidationError

class FruitPicker(ModelForm):
     # ⋮

    def clean(self, *args, **kwargs):
        data = super().clean(*args, **kwargs)
        total = sum([data[x] for x in ['banana', 'grape', 'apple', 'mango']])
        if not (2 <= total <= 3):
            raise ValidationError('You should select between 2 and 3 fruits.')
        return data

Здесь sum(…) суммирует bool числа и использует 0 для False, и 1 для True, таким образом, он подсчитывает количество True. Если это число не находится между 2 и 3, то возникает ошибка ValidationError.

Вернуться на верх