You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
3.7 KiB
88 lines
3.7 KiB
from django.http import HttpResponse, HttpResponseGone
|
|
from rest_framework import permissions, status
|
|
from rest_framework.generics import CreateAPIView, ListAPIView, RetrieveUpdateDestroyAPIView
|
|
from rest_framework.views import APIView
|
|
from rest_framework.response import Response
|
|
|
|
from ..models import Document, Question
|
|
from ..serializer import CreateDocumentSerializer, DocumentMetadataSerializer, UserResponseSerializer
|
|
from ..utils.question_step_mapping import question_step_mapping
|
|
from ..utils.user_response import save_to_session, save_to_db
|
|
|
|
|
|
class UserResponseHandler(APIView):
|
|
def post(self, request):
|
|
if request.data == {}:
|
|
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
|
|
serializer = UserResponseSerializer(data=request.data)
|
|
question_key = request.data['question']
|
|
|
|
value = request.data['value'].replace('<', '<').replace('>', '>')
|
|
user_attribute_updated = False
|
|
try:
|
|
try:
|
|
question = Question.objects.get(pk=question_key)
|
|
except Question.DoesNotExist:
|
|
question = None
|
|
|
|
if question is not None:
|
|
# As a result of discussion, decide to escape < and > only
|
|
if request.user.is_authenticated:
|
|
save_to_db(serializer, question, value, request.user)
|
|
else:
|
|
# only prequalification questions can be answered when you
|
|
# aren't logged into BCeID
|
|
if question_key not in question_step_mapping['prequalification']:
|
|
return Response(data="Not logged in",
|
|
status=status.HTTP_511_NETWORK_AUTHENTICATION_REQUIRED)
|
|
save_to_session(request, question, value)
|
|
else:
|
|
if request.user.is_authenticated and hasattr(request.user, question_key):
|
|
setattr(request.user, question_key, value == 'true')
|
|
request.user.save()
|
|
user_attribute_updated = True
|
|
except Exception as e:
|
|
if question is None and not user_attribute_updated:
|
|
return Response(data="Question: '%s' does not exist" % question_key,
|
|
status=status.HTTP_400_BAD_REQUEST)
|
|
return Response(status=status.HTTP_500_INTERNAL_ERROR)
|
|
|
|
return Response(status=status.HTTP_200_OK)
|
|
|
|
|
|
class DocumentCreateView(CreateAPIView):
|
|
serializer_class = CreateDocumentSerializer
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
queryset = Document.objects.all()
|
|
|
|
|
|
class DocumentMetaDataView(ListAPIView):
|
|
serializer_class = DocumentMetadataSerializer
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
|
|
def get_queryset(self):
|
|
doc_type = self.kwargs['doc_type']
|
|
party_code = self.kwargs['party_code']
|
|
return Document.objects.filter(doc_type=doc_type, party_code=party_code, bceid_user=self.request.user).order_by('sort_order')
|
|
|
|
|
|
class DocumentView(RetrieveUpdateDestroyAPIView):
|
|
serializer_class = DocumentMetadataSerializer
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
|
|
def get_object(self):
|
|
return Document.objects.get(bceid_user=self.request.user, **self.kwargs)
|
|
|
|
def retrieve(self, request, *args, **kwargs):
|
|
""" Return the file instead of meta data """
|
|
doc = self.get_object()
|
|
content_type = 'application/pdf' if 'pdf' in doc.filename else 'image/jpeg'
|
|
|
|
# If file doesn't exist anymore, delete it
|
|
try:
|
|
file_contents = doc.file.read()
|
|
except TypeError:
|
|
doc.delete()
|
|
return HttpResponseGone('File no longer exists')
|
|
return HttpResponse(file_contents, content_type=content_type)
|