diff --git a/recette/forms.py b/recette/forms.py index e1b9d32..d63d18f 100644 --- a/recette/forms.py +++ b/recette/forms.py @@ -6,10 +6,13 @@ from .models import Recette class SearchForm(forms.Form): text = forms.CharField(widget=forms.TextInput(attrs={'autofocus': True})) saison = forms.ChoiceField(choices=(('all', "Toutes"),) + Recette.SAISON_CHOICES) + veget = forms.BooleanField(label="Végétarien", required=False) def search(self): if self.is_valid(): qs = Recette.objects.all() + if self.cleaned_data['veget']: + qs = qs.filter(is_veget=True) if self.cleaned_data['text']: qs = qs.filter(nom__icontains=self.cleaned_data['text']) if self.cleaned_data['saison'] != 'all': diff --git a/recette/migrations/0003_recette_is_veget.py b/recette/migrations/0003_recette_is_veget.py new file mode 100644 index 0000000..a4c6a5f --- /dev/null +++ b/recette/migrations/0003_recette_is_veget.py @@ -0,0 +1,16 @@ +from django.db import migrations, models + + +class Migration(migrations.Migration): + + dependencies = [ + ('recette', '0002_recette_saison'), + ] + + operations = [ + migrations.AddField( + model_name='recette', + name='is_veget', + field=models.BooleanField(default=False, verbose_name='Mets végétarien'), + ), + ] diff --git a/recette/models.py b/recette/models.py index 1b06524..b6fee15 100644 --- a/recette/models.py +++ b/recette/models.py @@ -32,6 +32,7 @@ class Recette(models.Model): models.CharField(max_length=10, choices=SAISON_CHOICES, blank=True), verbose_name="Saison", ) + is_veget = models.BooleanField("Mets végétarien", default=False) prep = models.TextField("Préparation", blank=True) source = models.CharField("Source", max_length=200, blank=True) ingredients = models.ManyToManyField(Ingredient, through='Composition', blank=True)