mayan-edms/mayan/apps/converter/views.py

from __future__ import absolute_import, unicode_literals

import logging

from django.contrib.contenttypes.models import ContentType
from django.http import Http404
from django.shortcuts import get_object_or_404
from django.template import RequestContext
from django.urls import reverse
from django.utils.translation import ugettext_lazy as _

from mayan.apps.acls.models import AccessControlList
from mayan.apps.common.generics import (
    SingleObjectCreateView, SingleObjectDeleteView, SingleObjectEditView,
    SingleObjectListView
)

from .forms import TransformationForm
from .icons import icon_transformation_list
from .links import link_transformation_create
from .models import Transformation
from .permissions import (
    permission_transformation_create, permission_transformation_delete,
    permission_transformation_edit, permission_transformation_view
)

logger = logging.getLogger(__name__)


class TransformationCreateView(SingleObjectCreateView):
    form_class = TransformationForm

    def dispatch(self, request, *args, **kwargs):
        content_type = get_object_or_404(
            klass=ContentType, app_label=self.kwargs['app_label'],
            model=self.kwargs['model']
        )

        try:
            self.content_object = content_type.get_object_for_this_type(
                pk=self.kwargs['object_id']
            )
        except content_type.model_class().DoesNotExist:
            raise Http404

        AccessControlList.objects.check_access(
            obj=self.content_object,
            permissions=(permission_transformation_create,), user=request.user
        )

        return super(TransformationCreateView, self).dispatch(
            request, *args, **kwargs
        )

    def form_valid(self, form):
        instance = form.save(commit=False)
        instance.content_object = self.content_object
        try:
            instance.full_clean()
            instance.save()
        except Exception as exception:
            logger.debug('Invalid form, exception: %s', exception)
            return super(TransformationCreateView, self).form_invalid(form)
        else:
            return super(TransformationCreateView, self).form_valid(form)

    def get_extra_context(self):
        return {
            'content_object': self.content_object,
            'navigation_object_list': ('content_object',),
            'title': _(
                'Create new transformation for: %s'
            ) % self.content_object,
        }

    def get_post_action_redirect(self):
        return reverse(
            viewname='converter:transformation_list', kwargs={
                'app_label': self.kwargs['app_label'],
                'model': self.kwargs['model'],
                'object_id': self.kwargs['object_id']
            }
        )

    def get_queryset(self):
        return Transformation.objects.get_for_object(obj=self.content_object)


class TransformationDeleteView(SingleObjectDeleteView):
    model = Transformation

    def dispatch(self, request, *args, **kwargs):
        self.transformation = get_object_or_404(
            klass=Transformation, pk=self.kwargs['pk']
        )

        AccessControlList.objects.check_access(
            obj=self.transformation.content_object,
            permissions=(permission_transformation_delete,), user=request.user
        )

        return super(TransformationDeleteView, self).dispatch(
            request, *args, **kwargs
        )

    def get_post_action_redirect(self):
        return reverse(
            viewname='converter:transformation_list', kwargs={
                'app_label': self.transformation.content_type.app_label,
                'model': self.transformation.content_type.model,
                'object_id': self.transformation.object_id
            }
        )

    def get_extra_context(self):
        return {
            'content_object': self.transformation.content_object,
            'navigation_object_list': ('content_object', 'transformation'),
            'previous': reverse(
                viewname='converter:transformation_list', kwargs={
                    'app_label': self.transformation.content_type.app_label,
                    'model': self.transformation.content_type.model,
                    'object_id': self.transformation.object_id
                }
            ),
            'title': _(
                'Delete transformation "%(transformation)s" for: '
                '%(content_object)s?'
            ) % {
                'transformation': self.transformation,
                'content_object': self.transformation.content_object
            },
            'transformation': self.transformation,
        }


class TransformationEditView(SingleObjectEditView):
    form_class = TransformationForm
    model = Transformation

    def dispatch(self, request, *args, **kwargs):
        self.transformation = get_object_or_404(
            klass=Transformation, pk=self.kwargs['pk']
        )

        AccessControlList.objects.check_access(
            obj=self.transformation.content_object,
            permissions=(permission_transformation_edit,), user=request.user
        )

        return super(TransformationEditView, self).dispatch(
            request, *args, **kwargs
        )

    def form_valid(self, form):
        instance = form.save(commit=False)
        try:
            instance.full_clean()
            instance.save()
        except Exception as exception:
            logger.debug('Invalid form, exception: %s', exception)
            return super(TransformationEditView, self).form_invalid(form=form)
        else:
            return super(TransformationEditView, self).form_valid(form=form)

    def get_extra_context(self):
        return {
            'content_object': self.transformation.content_object,
            'navigation_object_list': ('content_object', 'transformation'),
            'title': _(
                'Edit transformation "%(transformation)s" for: %(content_object)s'
            ) % {
                'transformation': self.transformation,
                'content_object': self.transformation.content_object
            },
            'transformation': self.transformation,
        }

    def get_post_action_redirect(self):
        return reverse(
            viewname='converter:transformation_list', kwargs={
                'app_label': self.transformation.content_type.app_label,
                'model': self.transformation.content_type.model,
                'object_id': self.transformation.object_id
            }
        )


class TransformationListView(SingleObjectListView):
    def dispatch(self, request, *args, **kwargs):
        content_type = get_object_or_404(
            klass=ContentType, app_label=self.kwargs['app_label'],
            model=self.kwargs['model']
        )

        try:
            self.content_object = content_type.get_object_for_this_type(
                pk=self.kwargs['object_id']
            )
        except content_type.model_class().DoesNotExist:
            raise Http404

        AccessControlList.objects.check_access(
            obj=self.content_object,
            permissions=(permission_transformation_view,), user=request.user
        )

        return super(TransformationListView, self).dispatch(
            request, *args, **kwargs
        )

    def get_extra_context(self):
        return {
            'content_object': self.content_object,
            'hide_link': True,
            'hide_object': True,
            'navigation_object_list': ('content_object',),
            'no_results_icon': icon_transformation_list,
            'no_results_main_link': link_transformation_create.resolve(
                context=RequestContext(
                    request=self.request, dict_={
                        'content_object': self.content_object
                    }
                )
            ),
            'no_results_text': _(
                'Transformations allow changing the visual appearance '
                'of documents without making permanent changes to the '
                'document file themselves.'
            ),
            'no_results_title': _('No transformations'),
            'title': _('Transformations for: %s') % self.content_object,
        }

    def get_source_queryset(self):
        return Transformation.objects.get_for_object(obj=self.content_object)