Set main choice between plat and dessert

recette/forms.py

@@ -11,16 +11,18 @@ class SearchForm(forms.Form):
         label="Texte", required=False
     )
     saison = forms.ChoiceField(choices=(('all', "Toutes"),) + Recette.SAISON_CHOICES)
+    type_princ = forms.ChoiceField(
+        label='',
+        choices=(('plat', 'Plat principal'), ('dessert', 'Dessert')),
+        widget=forms.RadioSelect
+    )
     veget = forms.BooleanField(label="Végétarien", required=False)
-    dessert = forms.BooleanField(label="Dessert", required=False)
 
     def search(self):
         if self.is_valid():
-            qs = Recette.objects.all()
+            qs = Recette.objects.filter(is_dessert=self.cleaned_data['type_princ'] == 'dessert')
             if self.cleaned_data['veget']:
                 qs = qs.filter(is_veget=True)
-            if self.cleaned_data['dessert']:
-                qs = qs.filter(is_dessert=True)
             if self.cleaned_data['text']:
                 terms = [t for t in self.cleaned_data['text'].split() if t not in STOP_WORDS]
                 for term in terms:

recette/views.py

@@ -5,7 +5,7 @@ from .models import Recette
 
 
 def home(request):
-    form = SearchForm(request.POST or None)
+    form = SearchForm(request.POST or None, initial={'type_princ': 'plat'})
     recettes = []
     num_results = None
     if request.method == 'POST':