Coverage for burials/dal_views.py: 41%

1from dal import autocomplete 

2from django.db.models import Q 

3 

4from bib.models import Book 

5from places.models import Place 

6from .models import ( 

7 BurialSite, 

8 BurialGroup, 

9 Burial, 

10 UrnCover, 

11) 

12 

13 

14class PlaceAC(autocomplete.Select2QuerySetView): 

15 def qet_queryset(self): 

16 qs = Place.objects.all() 

17 if self.q: 

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

19 

20 return qs 

21 

22 

23class BookAC(autocomplete.Select2QuerySetView): 

24 def get_queryset(self): 

25 qs = Book.objects.all() 

26 if self.q: 

27 qs = qs.filter(Q(author__icontains=self.q) | Q(title__icontains=self.q)) 

28 

29 return qs 

30 

31 

32class BurialSiteAC(autocomplete.Select2QuerySetView): 

33 def get_queryset(self): 

34 qs = BurialSite.objects.all() 

35 if self.q: 

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

37 

38 return qs 

39 

40 

41class BurialGroupAC(autocomplete.Select2QuerySetView): 

42 def get_queryset(self): 

43 qs = BurialGroup.objects.all() 

44 if self.q: 

45 qs = qs.filter(id__icontains=self.q) 

46 

47 return qs 

48 

49 

50class BurialAC(autocomplete.Select2QuerySetView): 

51 def get_queryset(self): 

52 qs = Burial.objects.all() 

53 if self.q: 

54 qs = qs.filter(burial_id__icontains=self.q) 

55 

56 return qs 

57 

58 

59class UrnCoverAC(autocomplete.Select2QuerySetView): 

60 def get_queryset(self): 

61 qs = UrnCover.objects.all() 

62 if self.q: 

63 qs = qs.filter(id__icontains=self.q) 

64 

65 return qs