124 lines
4.2 KiB
Python
124 lines
4.2 KiB
Python
from __future__ import absolute_import
|
|
|
|
from django.core.exceptions import PermissionDenied
|
|
|
|
from converter.exceptions import UnkownConvertError, UnknownFileFormat
|
|
from converter.literals import (DEFAULT_FILE_FORMAT_MIMETYPE, DEFAULT_PAGE_NUMBER,
|
|
DEFAULT_ROTATION, DEFAULT_ZOOM_LEVEL)
|
|
from permissions.models import Permission
|
|
from rest_framework import generics
|
|
from rest_framework.response import Response
|
|
|
|
from .conf.settings import (DISPLAY_SIZE, PREVIEW_SIZE, RECENT_COUNT,
|
|
ROTATION_STEP, ZOOM_PERCENT_STEP, ZOOM_MAX_LEVEL,
|
|
ZOOM_MIN_LEVEL)
|
|
from rest_api.filters import MayanObjectPermissionsFilter
|
|
from rest_api.permissions import MayanPermission
|
|
|
|
from .permissions import PERMISSION_DOCUMENT_VIEW
|
|
from .models import Document, DocumentPage, DocumentVersion
|
|
from .serializers import (DocumentImageSerializer, DocumentPageSerializer,
|
|
DocumentSerializer, DocumentVersionSerializer)
|
|
|
|
|
|
class APIDocumentListView(generics.ListAPIView):
|
|
"""
|
|
Returns a list of all the documents.
|
|
"""
|
|
|
|
serializer_class = DocumentSerializer
|
|
queryset = Document.objects.all()
|
|
|
|
filter_backends = (MayanObjectPermissionsFilter,)
|
|
mayan_object_permissions = [PERMISSION_DOCUMENT_VIEW]
|
|
|
|
|
|
class APIDocumentPageView(generics.RetrieveAPIView):
|
|
"""
|
|
Returns the selected document page details.
|
|
"""
|
|
|
|
allowed_methods = ['GET']
|
|
serializer_class = DocumentPageSerializer
|
|
queryset = DocumentPage.objects.all()
|
|
|
|
permission_classes = (MayanPermission,)
|
|
mayan_object_permissions = [PERMISSION_DOCUMENT_VIEW]
|
|
mayan_permission_attribute_check = 'document'
|
|
|
|
|
|
class APIDocumentView(generics.RetrieveAPIView):
|
|
"""
|
|
Returns the selected document details.
|
|
"""
|
|
|
|
allowed_methods = ['GET']
|
|
serializer_class = DocumentSerializer
|
|
queryset = Document.objects.all()
|
|
|
|
permission_classes = (MayanPermission,)
|
|
mayan_object_permissions = [PERMISSION_DOCUMENT_VIEW]
|
|
|
|
|
|
class APIDocumentVersionView(generics.RetrieveAPIView):
|
|
"""
|
|
Returns the selected document version details.
|
|
"""
|
|
|
|
allowed_methods = ['GET']
|
|
serializer_class = DocumentVersionSerializer
|
|
queryset = DocumentVersion.objects.all()
|
|
|
|
permission_classes = (MayanPermission,)
|
|
mayan_object_permissions = [PERMISSION_DOCUMENT_VIEW]
|
|
mayan_permission_attribute_check = 'document'
|
|
|
|
|
|
class APIDocumentImageView(generics.GenericAPIView):
|
|
"""
|
|
Returns an image representation of the selected document.
|
|
size -- 'x' seprated width and height of the desired image representation.
|
|
page -- Page number of the document to be imaged.
|
|
zoom -- Zoom level of the image to be generated, numeric value only.
|
|
version -- Version number of the document to be imaged.
|
|
"""
|
|
serializer_class = DocumentImageSerializer
|
|
|
|
def get(self, request, pk):
|
|
document = get_object_or_404(Document, pk=pk)
|
|
|
|
logger.debug('document: %s' % document)
|
|
|
|
try:
|
|
Permission.objects.check_permissions(request.user, [PERMISSION_DOCUMENT_VIEW])
|
|
except PermissionDenied:
|
|
AccessEntry.objects.check_access(PERMISSION_DOCUMENT_VIEW, request.user, document)
|
|
|
|
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))
|
|
|
|
version = int(request.GET.get('version', document.latest_version.pk))
|
|
|
|
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': document.get_image(size=size, page=page, zoom=zoom, rotation=rotation, as_base64=True, version=version)
|
|
})
|
|
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)})
|