Use OccupationSerializer for the Contacts API

This commit is contained in:
Aarni Halinen
2019-10-10 18:37:15 +03:00
parent 4b63de6fc4
commit e5cc62bbbf
4 changed files with 34 additions and 21 deletions
+4 -4
View File
@@ -108,15 +108,15 @@ class FeedViewSet(viewsets.ModelViewSet):
class ContactsViewSet(viewsets.ReadOnlyModelViewSet):
queryset = Official.objects.all()
serializer_class = ContactsSerializer
queryset = Occupation.objects.all()
serializer_class = OccupationSerializer
permission_classes = [IsAuthenticatedOrReadOnly]
def get_queryset(self):
year = self.request.query_params.get('year')
if not year:
return Official.official_by_year(timezone.now().year)
return Official.official_by_year(int(year))
return Occupation.by_year(timezone.now().year)
return Occupation.by_year(int(year))
class TagsViewSet(viewsets.ReadOnlyModelViewSet):