Move API code to the views modules

mayan/apps/sources/views.py

@@ -12,8 +12,15 @@ from django.utils.safestring import mark_safe
 from django.utils.translation import ugettext
 from django.utils.translation import ugettext_lazy as _
 
+from rest_framework import generics
+from rest_framework.response import Response
+
 from acls.models import AccessEntry
 from common.utils import encapsulate
+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 documents.exceptions import NewDocumentVersionNotAllowed
 from documents.models import DocumentType, Document
 from documents.permissions import (PERMISSION_DOCUMENT_CREATE,
@@ -31,6 +38,8 @@ from .models import (WebForm, StagingFolder, SourceTransformation,
 from .permissions import (PERMISSION_SOURCES_SETUP_VIEW,
     PERMISSION_SOURCES_SETUP_EDIT, PERMISSION_SOURCES_SETUP_DELETE,
     PERMISSION_SOURCES_SETUP_CREATE)
+from .serializers import (StagingFolderSerializer, StagingFolderFileSerializer,
+    StagingSourceFileImageSerializer)
 
 
 def document_create_siblings(request, document_id):
@@ -674,3 +683,75 @@ def setup_source_transformation_create(request, source_type, source_id):
         'navigation_object_name': 'source',
         'title': _(u'Create new transformation for source: %s') % source,
     }, context_instance=RequestContext(request))
+
+
+# API Views
+
+
+class APIStagingSourceFileView(generics.GenericAPIView):
+    """
+    Details of the selected staging file.
+    """
+    serializer_class = StagingFolderFileSerializer
+
+    def get(self, request, staging_folder_pk, encoded_filename):
+        staging_folder = get_object_or_404(StagingFolder, pk=staging_folder_pk)
+        return Response(StagingFolderFileSerializer(staging_folder.get_file(encoded_filename=encoded_filename), context={'request': request}).data)
+
+
+class APIStagingSourceListView(generics.ListAPIView):
+    """
+    Returns a list of all the staging folders and the files they contain.
+    """
+
+    serializer_class = StagingFolderSerializer
+    queryset = StagingFolder.objects.all()
+
+
+class APIStagingSourceView(generics.RetrieveAPIView):
+    """
+    Details of the selected staging folders and the files it contains.
+    """
+    serializer_class = StagingFolderSerializer
+    queryset = StagingFolder.objects.all()
+
+
+class APIStagingSourceFileImageView(generics.GenericAPIView):
+    """
+    Image of the selected staging file.
+    size -- 'x' seprated width and height of the desired image representation.
+    page -- Page number of the staging file to be imaged.
+    zoom -- Zoom level of the image to be generated, numeric value only.
+    """
+
+    serializer_class = StagingSourceFileImageSerializer
+
+    def get(self, request, staging_folder_pk, encoded_filename):
+        staging_folder = get_object_or_404(StagingFolder, pk=staging_folder_pk)
+        staging_file = staging_folder.get_file(encoded_filename=encoded_filename)
+
+        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)})