Add csv renderer to some API endpoints for csv exports

References #121

Author: u8sand

FAIRshakeAPI/views.py

@@ -17,6 +17,7 @@
 from django.urls import reverse
 from django.core.exceptions import MultipleObjectsReturned
 from rest_framework import views, viewsets, schemas, response, mixins, decorators, renderers, permissions, status
+from rest_framework_csv.renderers import CSVRenderer
 from functools import reduce
 from collections import defaultdict, OrderedDict
 
@@ -59,6 +60,7 @@ def get_template_context(self, data, renderer_context):
 class CustomModelViewSet(viewsets.ModelViewSet):
   renderer_classes = [
     renderers.JSONRenderer,
+    CSVRenderer,
     CustomTemplateHTMLRenderer,
     CustomBrowsableAPIRenderer,
   ]
@@ -138,14 +140,27 @@ def get_or_create(self, request, **kwargs):
     headers = self.get_success_headers(serialized_item.data)
     return response.Response(serialized_item.data, status=status.HTTP_200_OK if found else status.HTTP_201_CREATED, headers=headers)
 
-  @swagger_auto_schema(methods=['get'], auto_schema=None)
+  @swagger_auto_schema(methods=['get'])
   @decorators.action(
     detail=True, methods=['get'], schema=None,
-    renderer_classes=[CustomTemplateHTMLRenderer],
+    renderer_classes=[
+      renderers.JSONRenderer,
+      CSVRenderer,
+      CustomTemplateHTMLRenderer
+    ],
   )
   def assessments(self, request, pk=None, **kwargs):
     self.check_permissions(request)
-    return response.Response()
+    if isinstance(request.accepted_renderer, CustomTemplateHTMLRenderer):
+      return response.Response()
+    # redirect /object/{pk}/assessments => assessments?object={pk}
+    GET = request.GET.copy()
+    GET[self.get_model_name()] = pk
+    return shortcuts.redirect(
+      reverse('assessment-list')
+      + '?'
+      + GET.urlencode()
+    )
 
   @swagger_auto_schema(methods=['get', 'post'], auto_schema=None)
   @decorators.action(
@@ -323,6 +338,7 @@ class DigitalObjectViewSet(IdentifiableModelViewSet):
     detail=True, methods=['get'],
     renderer_classes=[
       renderers.JSONRenderer,
+      CSVRenderer,
       CustomBrowsableAPIRenderer,
     ],
   )

requirements.txt

@@ -12,6 +12,7 @@ django-filter==2.3.0
 django-livereload-server==0.3.2
 django-rest-auth==0.9.5
 Django==2.2.13
+djangorestframework-csv==2.1.1
 djangorestframework==3.11.0
 drf-yasg==1.17.1
 dynamic-rest==1.9.6