76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from django.http import HttpResponse
|
|
from django.shortcuts import get_object_or_404
|
|
|
|
from converter.models import Transformation
|
|
from rest_framework import generics
|
|
from rest_framework.response import Response
|
|
|
|
from .models import StagingFolderSource
|
|
from .serializers import StagingFolderFileSerializer, StagingFolderSerializer
|
|
|
|
|
|
class APIStagingSourceFileView(generics.GenericAPIView):
|
|
"""
|
|
get: Details of the selected staging file.
|
|
"""
|
|
serializer_class = StagingFolderFileSerializer
|
|
|
|
def get(self, request, staging_folder_pk, encoded_filename):
|
|
staging_folder = get_object_or_404(
|
|
StagingFolderSource, pk=staging_folder_pk
|
|
)
|
|
return Response(
|
|
StagingFolderFileSerializer(
|
|
staging_folder.get_file(encoded_filename=encoded_filename),
|
|
context={'request': request}
|
|
).data
|
|
)
|
|
|
|
|
|
class APIStagingSourceListView(generics.ListAPIView):
|
|
"""
|
|
get: Returns a list of all the staging folders and the files they contain.
|
|
"""
|
|
serializer_class = StagingFolderSerializer
|
|
queryset = StagingFolderSource.objects.all()
|
|
|
|
|
|
class APIStagingSourceView(generics.RetrieveAPIView):
|
|
"""
|
|
get: Details of the selected staging folders and the files it contains.
|
|
"""
|
|
serializer_class = StagingFolderSerializer
|
|
queryset = StagingFolderSource.objects.all()
|
|
|
|
|
|
class APIStagingSourceFileImageView(generics.RetrieveAPIView):
|
|
"""
|
|
get: Returns an image representation of the selected document.
|
|
"""
|
|
def get_serializer(self, *args, **kwargs):
|
|
return None
|
|
|
|
def get_serializer_class(self):
|
|
return None
|
|
|
|
def retrieve(self, request, *args, **kwargs):
|
|
staging_folder = get_object_or_404(
|
|
StagingFolderSource, pk=self.kwargs['staging_folder_pk']
|
|
)
|
|
staging_file = staging_folder.get_file(
|
|
encoded_filename=self.kwargs['encoded_filename']
|
|
)
|
|
|
|
size = request.GET.get('size')
|
|
|
|
return HttpResponse(
|
|
staging_file.get_image(
|
|
size=size,
|
|
transformations=Transformation.objects.get_for_model(
|
|
staging_folder, as_classes=True
|
|
)
|
|
), content_type='image'
|
|
)
|