Add proper document creation API endpoint
This commit is contained in:
@@ -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):
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user