Coverage for cookbook/helper/dal.py: 68%

1from cookbook.models import Food, Keyword, Recipe, Unit 

2 

3from dal import autocomplete 

4 

5 

6class BaseAutocomplete(autocomplete.Select2QuerySetView): 

7 model = None 

8 

9 def get_queryset(self): 

10 if not self.request.user.is_authenticated: 

11 return self.model.objects.none() 

12 

13 qs = self.model.objects.filter(space=self.request.space).all() 

14 

15 if self.q: 

16 qs = qs.filter(name__icontains=self.q) 

17 

18 return qs 

19 

20 

21class KeywordAutocomplete(BaseAutocomplete): 

22 model = Keyword 

23 

24 

25class IngredientsAutocomplete(BaseAutocomplete): 

26 model = Food 

27 

28 

29class RecipeAutocomplete(BaseAutocomplete): 

30 model = Recipe 

31 

32 

33class UnitAutocomplete(BaseAutocomplete): 

34 model = Unit