Add proper document creation API endpoint

This commit is contained in:
Roberto Rosario
2014-10-27 04:49:29 -04:00
parent 165d01610b
commit 9c9632f920
5 changed files with 135 additions and 11 deletions

View File

@@ -1,10 +1,13 @@
from __future__ import absolute_import
import tempfile
from django.core.exceptions import PermissionDenied
from django.shortcuts import get_object_or_404
from rest_framework import generics, status
from rest_framework.response import Response
from rest_framework.settings import api_settings
from acls.models import AccessEntry
from converter.exceptions import UnkownConvertError, UnknownFileFormat
@@ -26,14 +29,15 @@ from .permissions import (PERMISSION_DOCUMENT_CREATE,
PERMISSION_DOCUMENT_TYPE_VIEW)
from .serializers import (DocumentImageSerializer, DocumentPageSerializer,
DocumentSerializer, DocumentTypeSerializer,
DocumentVersionSerializer)
DocumentVersionSerializer, NewDocumentSerializer)
from .settings import DISPLAY_SIZE, ZOOM_MAX_LEVEL, ZOOM_MIN_LEVEL
from .tasks import task_get_document_image
from .tasks import task_get_document_image, task_new_document
DOCUMENT_IMAGE_TASK_TIMEOUT = 10
class APIDocumentListView(generics.ListCreateAPIView):
class APIDocumentListView(generics.ListAPIView):
"""
Returns a list of all the documents.
"""
@@ -44,8 +48,51 @@ class APIDocumentListView(generics.ListCreateAPIView):
permission_classes = (MayanPermission,)
filter_backends = (MayanObjectPermissionsFilter,)
mayan_object_permissions = {'GET': [PERMISSION_DOCUMENT_VIEW]}
class APINewDocumentView(generics.GenericAPIView):
serializer_class = NewDocumentSerializer
permission_classes = (MayanPermission,)
mayan_view_permissions = {'POST': [PERMISSION_DOCUMENT_CREATE]}
def post(self, request, *args, **kwargs):
"""Create a new document."""
serializer = self.get_serializer(data=request.DATA, files=request.FILES)
if serializer.is_valid():
print serializer.data
temporary_file = tempfile.NamedTemporaryFile(delete=False)
source_file = request.FILES['file']
for chunk in source_file.chunks():
temporary_file.write(chunk)
temporary_file.close()
source_file.close()
task_new_document.apply_async(kwargs=dict(
file_path=temporary_file.name,
document_type_id=serializer.data['document_type'],
description=serializer.data['description'],
expand=serializer.data['expand'],
label=serializer.data['label'],
language=serializer.data['language'],
user_id=serializer.data['user']
), queue='uploads')
headers = self.get_success_headers(serializer.data)
return Response(serializer.data, status=status.HTTP_202_ACCEPTED,
headers=headers)
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
def get_success_headers(self, data):
try:
return {'Location': data[api_settings.URL_FIELD_NAME]}
except (TypeError, KeyError):
return {}
class APIDocumentView(generics.RetrieveUpdateDestroyAPIView):
"""