Coverage for burials/api_views.py: 100%

1from rest_framework import viewsets 

2from rest_framework import pagination 

3from .models import ( 

4 BurialSite, 

5 BurialGroup, 

6 Burial, 

7 Urn, 

8 UrnCover, 

9 GraveGood, 

10 GraveGoodOther, 

11 DeadBodyRemains, 

12 AnimalRemains, 

13) 

14from .serializers import ( 

15 BurialSiteSerializer, 

16 BurialGroupSerializer, 

17 BurialSerializer, 

18 UrnSerializer, 

19 UrnCoverSerializer, 

20 GraveGoodSerializer, 

21 GraveGoodOtherSerializer, 

22 DeadBodyRemainsSerializer, 

23 AnimalRemainsSerializer, 

24) 

25 

26 

27class LargeResultsSetPagination(pagination.PageNumberPagination): 

28 page_size = 25 

29 page_size_query_param = "page_size" 

30 max_page_size = 10000 

31 

32 

33class BurialSiteViewSet(viewsets.ModelViewSet): 

34 queryset = BurialSite.objects.all() 

35 serializer_class = BurialSiteSerializer 

36 pagination_class = LargeResultsSetPagination 

37 

38 

39class BurialGroupViewSet(viewsets.ModelViewSet): 

40 queryset = BurialGroup.objects.all() 

41 serializer_class = BurialGroupSerializer 

42 pagination_class = LargeResultsSetPagination 

43 

44 

45class BurialViewSet(viewsets.ModelViewSet): 

46 queryset = Burial.objects.all() 

47 serializer_class = BurialSerializer 

48 pagination_class = LargeResultsSetPagination 

49 

50 

51class UrnViewSet(viewsets.ModelViewSet): 

52 queryset = Urn.objects.all() 

53 serializer_class = UrnSerializer 

54 pagination_class = LargeResultsSetPagination 

55 

56 

57class UrnCoverViewSet(viewsets.ModelViewSet): 

58 queryset = UrnCover.objects.all() 

59 serializer_class = UrnCoverSerializer 

60 pagination_class = LargeResultsSetPagination 

61 

62 

63class GraveGoodViewSet(viewsets.ModelViewSet): 

64 queryset = GraveGood.objects.all() 

65 serializer_class = GraveGoodSerializer 

66 pagination_class = LargeResultsSetPagination 

67 

68 

69class GraveGoodOtherViewSet(viewsets.ModelViewSet): 

70 queryset = GraveGoodOther.objects.all() 

71 serializer_class = GraveGoodOtherSerializer 

72 pagination_class = LargeResultsSetPagination 

73 

74 

75class AnthropologyViewSet(viewsets.ModelViewSet): 

76 queryset = DeadBodyRemains.objects.all() 

77 serializer_class = DeadBodyRemainsSerializer 

78 pagination_class = LargeResultsSetPagination 

79 

80 

81class AnimalRemainsViewSet(viewsets.ModelViewSet): 

82 queryset = AnimalRemains.objects.all() 

83 serializer_class = AnimalRemainsSerializer 

84 pagination_class = LargeResultsSetPagination