EGCETSII#13 - feat: Added new endpoints for Voting API

+Bulk delete +Get by id +Delete all elements Co-Authored-By: Beatriz María Beltrán Álvarez <[email protected]>
Full-Tortuga · Jan 3, 2022 · 7f89b13 · 7f89b13
1 parent 16531fb
commit 7f89b13
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/decide/administration/urls.py b/decide/administration/urls.py
@@ -20,6 +20,7 @@
     path('api/votings/question', views.QuestionsAPI.as_view()),
     path('api/votings/question/<int:question_id>/', views.QuestionAPI.as_view()),
     path('api/votings', views.VotingAPI.as_view()),
+    path('api/votings/<int:voting_id>/', views.VotingsAPI.as_view()),
     # react-app
     url('', views.index)
 ]
diff --git a/decide/administration/views.py b/decide/administration/views.py
@@ -107,6 +107,32 @@ def put(self, request):
             st = status.HTTP_400_BAD_REQUEST
         return Response(msg, status=st)
 
+    def delete(self, request):
+        if request.data.get("idList") is None:
+            Voting.objects.all().delete()
+            return Response({}, status=HTTP_200_OK)
+        else:
+            ids = request.data.get("idList")
+            is_valid(len(ids) > 0, 'The format of the ids list is not correct')
+            Voting.objects.filter(id__in=ids).delete()
+            return Response({}, status=HTTP_200_OK)
+
+class VotingsAPI(APIView):
+    permission_classes = (IsAdminAPI,)
+
+    def get(self, request, voting_id):
+        try:
+            query = Voting.objects.filter(id=voting_id).get()
+        except ObjectDoesNotExist:
+            return Response({}, status=HTTP_404_NOT_FOUND)
+        rest = VotingSerializer(query).data
+        return Response(rest, status=HTTP_200_OK)
+
+
+    def delete(self, request, voting_id):
+        Voting.objects.all().filter(id=voting_id).delete()
+        return Response({}, status=HTTP_200_OK)
+
 
 class QuestionsAPI(APIView):
     permission_classes = (IsAdminAPI,)