79 lines
2.9 KiB
Python
79 lines
2.9 KiB
Python
from __future__ import absolute_import
|
|
|
|
import logging
|
|
|
|
from django.core.exceptions import PermissionDenied
|
|
from django.shortcuts import get_object_or_404
|
|
|
|
from rest_framework import generics
|
|
from rest_framework.response import Response
|
|
|
|
from converter.exceptions import UnkownConvertError, UnknownFileFormat
|
|
from converter.literals import DEFAULT_PAGE_NUMBER, DEFAULT_ROTATION, DEFAULT_ZOOM_LEVEL
|
|
from documents.conf.settings import DISPLAY_SIZE, ZOOM_MAX_LEVEL, ZOOM_MIN_LEVEL
|
|
|
|
#from acls.models import AccessEntry
|
|
#from permissions.models import Permission
|
|
from .classes import StagingFile
|
|
from .resources import SerializerStagingFolder, SerializerStagingFolderFile
|
|
from .models import StagingFolder
|
|
#from .permissions import PERMISSION_DOCUMENT_VIEW
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# API Views
|
|
|
|
class APIStagingSourceFileView(generics.GenericAPIView):
|
|
def get(self, request, staging_folder_pk, filename):
|
|
staging_folder = get_object_or_404(StagingFolder, pk=staging_folder_pk)
|
|
return Response(SerializerStagingFolderFile(staging_folder.get_file(filename)).data)
|
|
|
|
|
|
class APIStagingSourceListView(generics.ListAPIView):
|
|
serializer_class = SerializerStagingFolder
|
|
queryset = StagingFolder.objects.all()
|
|
|
|
|
|
class APIStagingSourceView(generics.RetrieveAPIView):
|
|
serializer_class = SerializerStagingFolder
|
|
queryset = StagingFolder.objects.all()
|
|
|
|
|
|
class APIStagingSourceFileImageView(generics.GenericAPIView):
|
|
def get(self, request, staging_folder_pk, filename):
|
|
staging_folder = get_object_or_404(StagingFolder, pk=staging_folder_pk)
|
|
staging_file = staging_folder.get_file(encoded_filename=filename)
|
|
|
|
#try:
|
|
# Permission.objects.check_permissions(request.user, [PERMISSION_DOCUMENT_VIEW])
|
|
#except PermissionDenied:
|
|
# AccessEntry.objects.check_access(PERMISSION_DOCUMENT_VIEW, request.user, document)
|
|
# Permission.objects.check_permissions(request.user, [PERMISSION_DOCUMENT_CREATE, PERMISSION_DOCUMENT_NEW_VERSION])
|
|
|
|
size = request.GET.get('size', DISPLAY_SIZE)
|
|
|
|
page = int(request.GET.get('page', DEFAULT_PAGE_NUMBER))
|
|
|
|
zoom = int(request.GET.get('zoom', DEFAULT_ZOOM_LEVEL))
|
|
|
|
if request.GET.get('as_base64', False):
|
|
base64_version = True
|
|
|
|
if zoom < ZOOM_MIN_LEVEL:
|
|
zoom = ZOOM_MIN_LEVEL
|
|
|
|
if zoom > ZOOM_MAX_LEVEL:
|
|
zoom = ZOOM_MAX_LEVEL
|
|
|
|
rotation = int(request.GET.get('rotation', DEFAULT_ROTATION)) % 360
|
|
|
|
try:
|
|
return Response({'status': 'success',
|
|
'data': staging_file.get_image(size=size, page=page, zoom=zoom, rotation=rotation, as_base64=True)
|
|
})
|
|
except UnknownFileFormat as exception:
|
|
return Response({'status': 'error', 'detail': 'unknown_file_format', 'message': unicode(exception)})
|
|
except UnkownConvertError as exception:
|
|
return Response({'status': 'error', 'detail': 'converter_error', 'message': unicode(exception)})
|
|
|