77from rest_framework .response import Response
88from rest_framework .views import APIView
99from base .models import Auth , Key
10- from census .models import Census
11- from rest_framework .renderers import JSONRenderer
1210from authentication .serializers import UserSerializer
13- from administration .serializers import UserAdminSerializer , UserSerializer
1411from administration .serializers import *
1512from base .serializers import AuthSerializer , KeySerializer
1613from .serializers import CensusSerializer
1714from base .perms import IsAdminAPI
18- from utils .utils import get_ids , is_valid
19-
2015
2116
2217def index (request ):
2318 return render (request , "build/index.html" )
2419
20+
2521class CensussAPI (APIView ):
2622 permission_classes = (IsAdminAPI ,)
2723
@@ -38,11 +34,11 @@ def post(self, request):
3834 return Response ({}, status = HTTP_200_OK )
3935
4036 def delete (self , request ):
41- if request .data [ "idList" ] is None :
37+ if request .get ( "idList" ) is None :
4238 Census .objects .all ().delete ()
4339 return Response ({}, status = HTTP_200_OK )
4440 else :
45- ids = get_ids ( request .data [ "idList" ] )
41+ ids = request .get ( "idList" )
4642 Census .objects .filter (id__in = ids ).delete ()
4743 return Response ({}, status = HTTP_200_OK )
4844
@@ -62,7 +58,7 @@ def put(self, request, census_id):
6258 return Response ({"result" : "Census object is not valid" }, status = HTTP_400_BAD_REQUEST )
6359 else :
6460 try :
65- census = Census .objects .all ().filter (id = census_id ).get ()
61+ census = Census .objects .all ().filter (id = census_id ).get ()
6662 except ObjectDoesNotExist :
6763 return Response ({}, status = HTTP_404_NOT_FOUND )
6864 for key , value in request .data .items ():
@@ -74,6 +70,7 @@ def delete(self, request, census_id):
7470 Census .objects .all ().filter (id = census_id ).delete ()
7571 return Response ({}, status = HTTP_200_OK )
7672
73+
7774class CensussAPI (APIView ):
7875 permission_classes = (IsAdminAPI ,)
7976
0 commit comments