2019-07-06 02:41:16 -04:00
parent 9e068c3e83
commit fbb0f0b9bd
15 changed files with 194 additions and 55 deletions
							
							
							
						
@@ -1,6 +1,8 @@
from __future__ import absolute_import, unicode_literals
from django.http import HttpResponse
from django.shortcuts import get_object_or_404
from django.views.decorators.cache import cache_control, patch_cache_control
from rest_framework import generics
							
							
							
								
							
						
@@ -10,6 +12,7 @@ from mayan.apps.documents.permissions import permission_document_type_view
from mayan.apps.rest_api.filters import MayanObjectPermissionsFilter
from mayan.apps.rest_api.permissions import MayanPermission
from .literals import WORKFLOW_IMAGE_TASK_TIMEOUT
from .models import Workflow
from .permissions import (
    permission_workflow_create, permission_workflow_delete,
							
							
							
								
							
						
@@ -23,6 +26,10 @@ from .serializers import (
    WritableWorkflowTransitionSerializer
)
from .settings import settings_workflow_image_cache_time
from .storages import storage_workflowimagecache
from .tasks import task_generate_workflow_image
class APIDocumentTypeWorkflowListView(generics.ListAPIView):
    """
							
								
							
							
								
							
							
						
@@ -172,6 +179,41 @@ class APIWorkflowDocumentTypeView(generics.RetrieveDestroyAPIView):
        self.get_workflow().document_types.remove(instance)
class APIWorkflowImageView(generics.RetrieveAPIView):
    """
    get: Returns an image representation of the selected workflow.
    """
    filter_backends = (MayanObjectPermissionsFilter,)
    mayan_object_permissions = {
        'GET': (permission_workflow_view,),
    }
    queryset = Workflow.objects.all()
    def get_serializer(self, *args, **kwargs):
        return None
    def get_serializer_class(self):
        return None
    @cache_control(private=True)
    def retrieve(self, request, *args, **kwargs):
        task = task_generate_workflow_image.apply_async(
            kwargs=dict(
                document_state_id=self.get_object().pk,
            )
        )
        cache_filename = task.get(timeout=WORKFLOW_IMAGE_TASK_TIMEOUT)
        with storage_workflowimagecache.open(cache_filename) as file_object:
            response = HttpResponse(file_object.read(), content_type='image')
            if '_hash' in request.GET:
                patch_cache_control(
                    response,
                    max_age=settings_workflow_image_cache_time.value
                )
            return response
class APIWorkflowListView(generics.ListCreateAPIView):
    """
    get: Returns a list of all the workflows.