Update document API to use new MayanAPIviewset

Signed-off-by: Roberto Rosario <Roberto.Rosario.Gonzalez@mayan-edms.com>
This commit is contained in:
Roberto Rosario
2019-02-19 03:37:16 -04:00
parent 9370b2effb
commit 023d82c96c

View File

@@ -13,8 +13,9 @@ from rest_framework import generics, status, viewsets
from rest_framework.response import Response from rest_framework.response import Response
from mayan.apps.acls.models import AccessControlList from mayan.apps.acls.models import AccessControlList
from mayan.apps.rest_api.filters import MayanObjectPermissionsFilter from mayan.apps.rest_api.viewsets import (
from mayan.apps.rest_api.permissions import MayanPermission MayanAPIModelViewSet, MayanAPIReadOnlyModelViewSet
)
from .literals import DOCUMENT_IMAGE_TASK_TIMEOUT from .literals import DOCUMENT_IMAGE_TASK_TIMEOUT
from .models import Document, DocumentVersion, DocumentType, RecentDocument from .models import Document, DocumentVersion, DocumentType, RecentDocument
@@ -47,7 +48,7 @@ from rest_framework.decorators import action, detail_route
from rest_framework.response import Response from rest_framework.response import Response
class DocumentViewSet(viewsets.ModelViewSet): class DocumentViewSet(MayanAPIModelViewSet):
lookup_field = 'pk' lookup_field = 'pk'
lookup_url_kwarg = 'document_id' lookup_url_kwarg = 'document_id'
queryset = Document.objects.all() queryset = Document.objects.all()
@@ -66,7 +67,8 @@ class DocumentViewSet(viewsets.ModelViewSet):
return Response(serializer.data) return Response(serializer.data)
""" """
class DocumentPageViewSet(viewsets.ModelViewSet): #class DocumentPageViewSet(MayanAPIModelViewSet):
class DocumentPageViewSet(MayanAPIReadOnlyModelViewSet):
lookup_field = 'pk' lookup_field = 'pk'
lookup_url_kwarg = 'document_page_id' lookup_url_kwarg = 'document_page_id'
serializer_class = DocumentPageSerializer serializer_class = DocumentPageSerializer
@@ -83,9 +85,6 @@ class DocumentPageViewSet(viewsets.ModelViewSet):
) )
@cache_control(private=True) @cache_control(private=True)
def document_page_image(self, request, *args, **kwargs): def document_page_image(self, request, *args, **kwargs):
"""
asdasd
"""
transformation_dict = { transformation_dict = {
'kwargs': {}, 'kwargs': {},
'name': {} 'name': {}
@@ -157,7 +156,7 @@ class DocumentPageViewSet(viewsets.ModelViewSet):
return Response(serializer.data) return Response(serializer.data)
""" """
class DocumentTypeViewSet(viewsets.ModelViewSet): class DocumentTypeViewSet(MayanAPIModelViewSet):
lookup_field = 'pk' lookup_field = 'pk'
lookup_url_kwarg = 'document_type_id' lookup_url_kwarg = 'document_type_id'
queryset = DocumentType.objects.all() queryset = DocumentType.objects.all()
@@ -175,7 +174,7 @@ class DocumentTypeViewSet(viewsets.ModelViewSet):
return Response(serializer.data) return Response(serializer.data)
class DocumentVersionViewSet(viewsets.ModelViewSet): class DocumentVersionViewSet(MayanAPIModelViewSet):
lookup_field = 'pk' lookup_field = 'pk'
lookup_url_kwarg = 'document_version_id' lookup_url_kwarg = 'document_version_id'
serializer_class = DocumentVersionSerializer serializer_class = DocumentVersionSerializer