deb363ee16
- Move all generic API classes definitions to the rest_api.generics module. - Update API status code on insufficient access for the apps: indexes, parsing, documents, metadata, ocr, permission, user management. - Update API tests. Signed-off-by: Roberto Rosario <roberto.rosario@mayan-edms.com>
78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from django.http import HttpResponse
|
|
from django.shortcuts import get_object_or_404
|
|
|
|
from rest_framework.response import Response
|
|
|
|
from mayan.apps.rest_api import generics
|
|
|
|
from .literals import STAGING_FILE_IMAGE_TASK_TIMEOUT
|
|
from .models import StagingFolderSource
|
|
from .serializers import StagingFolderFileSerializer, StagingFolderSerializer
|
|
from .storages import storage_staging_file_image_cache
|
|
from .tasks import task_generate_staging_file_image
|
|
|
|
|
|
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(
|
|
klass=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):
|
|
width = request.GET.get('width')
|
|
height = request.GET.get('height')
|
|
|
|
task = task_generate_staging_file_image.apply_async(
|
|
kwargs=dict(
|
|
staging_folder_pk=self.kwargs['staging_folder_pk'],
|
|
encoded_filename=self.kwargs['encoded_filename'],
|
|
width=width, height=height
|
|
)
|
|
)
|
|
|
|
cache_filename = task.get(timeout=STAGING_FILE_IMAGE_TASK_TIMEOUT)
|
|
|
|
with storage_staging_file_image_cache.open(cache_filename) as file_object:
|
|
response = HttpResponse(file_object.read(), content_type='image')
|
|
return response
|