отмените выбор, если имя начинается с A Django - PullRequest
Новая
       6

отмените выбор, если имя начинается с A Django

0 голосов
/ 11 марта 2020

спасибо за ваше время: я бы хотел снять один вариант выбора, если имя People начинается с буквы 'a':

в основном, если People.nome или (Animal.pessoa) начинаются с буква «А» я хотел бы снять выбор (2, «GATO»). Есть ли способ сделать это или я должен попробовать другой подход, кроме ChoiceField 

from django.db import models
from django.core.validators import RegexValidator



class People(models.Model):
    nome = models.CharField(max_length=200)
    birthday_date = models.DateField()
    cpf = models.CharField(max_length=11, validators=[RegexValidator(r'^\d{1,10}$')])

    def __str__(self):
        return '%s' % (self.nome)

def cant_have_cat(self):
    field_choices = [
        (1, 'CACHORRO'),
        (2, 'GATO'),
        (3, 'OUTRO'),
    ]
    self.maiusculo = self.pessoa.upper
    if self.maiusculo[0] == 'A':
        self.tipo != 2
    return field_choices

class Animal(models.Model):
    pessoa = models.ForeignKey(People, on_delete=models.CASCADE)
    name = models.CharField(max_length=100)
    tipo = models.IntegerField(choices=cant_have_cat)
    custo = models.DecimalField(max_digits=7, decimal_places=2)

    def __str__(self):
        return '%s %s' % (self.pessoa, self.name)

forms.py 



class AnimalForm(forms.ModelForm):
    field_choices = [
        (1, 'CACHORRO'),
        (2, 'GATO'),
        (3, 'OUTRO'),
    ]
    name = forms.CharField(max_length=100)
    tipo = forms.ChoiceField(choices=field_choices)
    custo = forms.DecimalField(max_digits=7, decimal_places=2)

    class Meta:
        prefix = 'animal'
        model = Animal
        fields = ('name', 'tipo', 'custo')

    def clean(self):
        people_name = self.People.nome
        upper = people_name.upper()
        if upper[0] == 'A':
            Animal.tipo != 2

1 Ответ

0 голосов
/ 11 марта 2020

Вы можете переопределить __init__ вашего AnimalForm для обработки логики c. Как то так: 

class AnimalForm(forms.ModelForm):
    field_choices = [
        (1, 'CACHORRO'),
        (2, 'GATO'),
        (3, 'OUTRO'),
    ]
    field_choices_subset = [
        (1, 'CACHORRO'),
        (3, 'OUTRO'),
    ]

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        if kwargs.get("instance") and kwargs.get("instance").pessoa.nome.startswith("a"):
            self.fields["tipo"].choices = field_choices_subset

    name = forms.CharField(max_length=100)
    tipo = forms.ChoiceField(choices=field_choices)
    custo = forms.DecimalField(max_digits=7, decimal_places=2)

    class Meta:
        prefix = 'animal'
        model = Animal
        fields = ('name', 'tipo', 'custo')
...