56 lines
1.3 KiB
Python
56 lines
1.3 KiB
Python
from dal import autocomplete
|
|
|
|
from cookbook.models import Keyword, RecipeIngredient, Recipe, Unit, Ingredient
|
|
|
|
|
|
class KeywordAutocomplete(autocomplete.Select2QuerySetView):
|
|
def get_queryset(self):
|
|
if not self.request.user.is_authenticated:
|
|
return Keyword.objects.none()
|
|
|
|
qs = Keyword.objects.all()
|
|
|
|
if self.q:
|
|
qs = qs.filter(name__istartswith=self.q)
|
|
|
|
return qs
|
|
|
|
|
|
class IngredientsAutocomplete(autocomplete.Select2QuerySetView):
|
|
def get_queryset(self):
|
|
if not self.request.user.is_authenticated:
|
|
return Ingredient.objects.none()
|
|
|
|
qs = Ingredient.objects.all()
|
|
|
|
if self.q:
|
|
qs = qs.filter(name__icontains=self.q)
|
|
|
|
return qs
|
|
|
|
|
|
class RecipeAutocomplete(autocomplete.Select2QuerySetView):
|
|
def get_queryset(self):
|
|
if not self.request.user.is_authenticated:
|
|
return Recipe.objects.none()
|
|
|
|
qs = Recipe.objects.all()
|
|
|
|
if self.q:
|
|
qs = qs.filter(name__icontains=self.q)
|
|
|
|
return qs
|
|
|
|
|
|
class UnitAutocomplete(autocomplete.Select2QuerySetView):
|
|
def get_queryset(self):
|
|
if not self.request.user.is_authenticated:
|
|
return Unit.objects.none()
|
|
|
|
qs = Unit.objects.all()
|
|
|
|
if self.q:
|
|
qs = qs.filter(name__icontains=self.q)
|
|
|
|
return qs
|