from rest_framework.renderers import JSONRenderer
from rest_framework.response import Response
from rest_framework.views import APIView

from storage.models import File, Comment
from storage.serializers import FileSerializer, CommentSerializer


class FileView(APIView):
    renderer_classes = (JSONRenderer,)

    @staticmethod
    def get(request):
        tokens = request.GET.getlist('tokens', None)
        if not tokens:
            return Response("Attribute tokens not set", status=400)

        file_list = []

        for token in tokens:
            try:
                file_list.append(FileSerializer(File.objects.get(token=token)).data)
            except File.DoesNotExist:
                file_list.append({'upload': 'error'})

        return Response(file_list, status=200)


class CommentView(APIView):
    renderer_classes = (JSONRenderer,)

    @staticmethod
    def get(request):
        tokens = request.GET.getlist('tokens', None)

        if not tokens:
            return Response("Attribute tokens not set", status=400)

        comment_list = []

        for token in tokens:
            try:
                comment_list.append(CommentSerializer(Comment.objects.get(token=token)).data)
            except Comment.DoesNotExist:
                comment_list.append({'upload': 'error'})

        return Response(comment_list, status=200)