1from rest_framework.viewsets import ReadOnlyModelViewSet 

2from rest_framework.response import Response 

3from rest_framework.decorators import action 

4 

5from .serializers import ( 

6 SimpleSummaryGroupSerializer, 

7 SummaryGroupSerializer, 

8 SummaryGroupTrendSerializer, 

9) 

10from .models import SummaryGroup 

11 

12 

13class SummaryGroupViewSet(ReadOnlyModelViewSet): 

14 queryset = SummaryGroup.objects.all() 

15 serializer_class = SummaryGroupSerializer 

16 

17 def get_serializer_class(self): 

18 if self.action == 'list': 

19 return SimpleSummaryGroupSerializer 

20 return super().get_serializer_class() 

21 

22 @action( 

23 detail=True, 

24 methods=['get'], 

25 url_path='trend', 

26 ) 

27 def trend(self, request, pk, version): 

28 instance = self.get_object() 

29 serializer = SummaryGroupTrendSerializer(instance) 

30 return Response(serializer.data)