refactor: Use custom CSV export view

Remove dependency on djangorestframework-csv

people/views/__init__.py

@@ -3,6 +3,7 @@ Views for displaying or manipulating models within the `people` app.
 """
 
 from . import (
+    export,
     network,
     person,
     relationship

people/views/export.py

@@ -0,0 +1,33 @@
+import csv
+import typing
+
+from django.http import HttpResponse
+from django.views.generic.list import BaseListView
+
+from .. import models, serializers
+
+
+class CsvExportView(BaseListView):
+    model = None
+    serializer_class = None
+
+    def render_to_response(self, context: typing.Dict) -> HttpResponse:
+        response = HttpResponse(content_type='text/csv')
+        response['Content-Disposition'] = f'attachment; filename="{self.get_context_object_name(self.object_list)}.csv"'
+        
+        serializer = self.serializer_class(self.get_queryset(), many=True)
+        writer = csv.DictWriter(response, fieldnames=self.serializer_class.Meta.fields)
+        writer.writeheader()
+        writer.writerows(serializer.data)
+            
+        return response
+
+
+class PersonExportView(CsvExportView):
+    model = models.Person
+    serializer_class = serializers.PersonExportSerializer
+    
+
+class RelationshipExportView(CsvExportView):
+    model = models.Relationship
+    serializer_class = serializers.RelationshipSerializer

people/views/person.py

@@ -4,9 +4,7 @@ Views for displaying or manipulating instances of :class:`Person`.
 
 from django.views.generic import CreateView, DetailView, ListView, UpdateView
 
-from rest_framework.views import APIView, Response
-
-from people import forms, models, permissions, serializers
+from people import forms, models, permissions
 
 
 class PersonCreateView(CreateView):
@@ -62,17 +60,3 @@ class PersonUpdateView(permissions.UserIsLinkedPersonMixin, UpdateView):
     model = models.Person
     template_name = 'people/person/update.html'
     form_class = forms.PersonForm
-
-
-class PersonApiView(APIView):
-    """
-    List all :class:`Person` instances.
-    """
-
-    def get(self, request, format=None):
-        """
-        List all :class:`Person` instances.
-        """
-        serializer = serializers.PersonSerializer(models.Person.objects.all(),
-                                                  many=True)
-        return Response(serializer.data)

people/views/relationship.py

@@ -6,9 +6,7 @@ from django.urls import reverse
 from django.utils import timezone
 from django.views.generic import CreateView, DetailView
 
-from rest_framework.views import APIView, Response
-
-from people import forms, models, permissions, serializers
+from people import forms, models, permissions
 
 
 class RelationshipDetailView(permissions.UserIsLinkedPersonMixin, DetailView):
@@ -130,17 +128,3 @@ class RelationshipUpdateView(permissions.UserIsLinkedPersonMixin, CreateView):
             answer_set.save()
 
         return response
-
-
-class RelationshipApiView(APIView):
-    """
-    List all :class:`Relationship` instances.
-    """
-
-    def get(self, request, format=None):
-        """
-        List all :class:`Relationship` instances.
-        """
-        serializer = serializers.RelationshipSerializer(models.Relationship.objects.all(),
-                                                        many=True)
-        return Response(serializer.data)