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.
61 lines
2.6 KiB
61 lines
2.6 KiB
from rest_framework import permissions, status
|
|
from rest_framework.views import APIView
|
|
from rest_framework.response import Response
|
|
from rest_framework.viewsets import ModelViewSet
|
|
|
|
from ..models import Document, Question
|
|
from ..serializer import DocumentSerializer, 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 DocumentViewSet(ModelViewSet):
|
|
permission_classes = [permissions.IsAuthenticatedOrReadOnly]
|
|
queryset = Document.objects.all()
|
|
serializer_class = DocumentSerializer
|
|
|
|
def get_queryset(self):
|
|
if self.request.user.is_anonymous:
|
|
return Document.objects.none()
|
|
return Document.objects.filter(bceid_user=self.request.user)
|