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.core.exceptions import PermissionDenied
from django.core.urlresolvers import reverse
from django.http import Http404
from django.shortcuts import get_object_or_404
from django.utils.translation import ugettext_lazy as _

from acls.models import AccessControlList
from common.utils import encapsulate
from common.views import (
    SingleObjectCreateView, SingleObjectDeleteView, SingleObjectEditView,
    SingleObjectListView
)
from permissions import Permission

from .models import Transformation
from .permissions import (
    permission_transformation_create, permission_transformation_delete,
    permission_transformation_edit, permission_transformation_view
)

logger = logging.getLogger(__name__)


class TransformationDeleteView(SingleObjectDeleteView):
    model = Transformation

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

        try:
            Permission.check_permissions(request.user, [permission_transformation_delete])
        except PermissionDenied:
            AccessControlList.objects.check_access(permission_transformation_delete, request.user, self.transformation.content_object)

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

    def get_post_action_redirect(self):
        return reverse('converter:transformation_list', args=[self.transformation.content_type.app_label, self.transformation.content_type.model, self.transformation.object_id])

    def get_extra_context(self):
        return {
            'content_object': self.transformation.content_object,
            'navigation_object_list': ['content_object', 'transformation'],
            'previous': reverse('converter:transformation_list', args=[self.transformation.content_type.app_label, self.transformation.content_type.model, 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 TransformationCreateView(SingleObjectCreateView):
    fields = ('name', 'arguments')

    def dispatch(self, request, *args, **kwargs):
        content_type = get_object_or_404(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

        try:
            Permission.check_permissions(request.user, [permission_transformation_create])
        except PermissionDenied:
            AccessControlList.objects.check_access(permission_transformation_create, request.user, self.content_object)

        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:
            return super(TransformationCreateView, self).form_invalid(form)
        else:
            return super(TransformationCreateView, self).form_valid(form)

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

    def get_queryset(self):
        return Transformation.objects.get_for_model(self.content_object)

    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,
        }


class TransformationEditView(SingleObjectEditView):
    fields = ('name', 'arguments', 'order')
    model = Transformation

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

        try:
            Permission.check_permissions(request.user, [permission_transformation_edit])
        except PermissionDenied:
            AccessControlList.objects.check_access(permission_transformation_edit, request.user, self.transformation.content_object)

        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:
            return super(TransformationEditView, self).form_invalid(form)
        else:
            return super(TransformationEditView, self).form_valid(form)

    def get_post_action_redirect(self):
        return reverse('converter:transformation_list', args=[self.transformation.content_type.app_label, self.transformation.content_type.model, self.transformation.object_id])

    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,
        }


class TransformationListView(SingleObjectListView):
    def dispatch(self, request, *args, **kwargs):
        content_type = get_object_or_404(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

        try:
            Permission.check_permissions(request.user, [permission_transformation_view])
        except PermissionDenied:
            AccessControlList.objects.check_access(permission_transformation_view, request.user, self.content_object)

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

    def get_queryset(self):
        return Transformation.objects.get_for_model(self.content_object)

    def get_extra_context(self):
        return {
            'content_object': self.content_object,
            'extra_columns': [
                {'name': _('Order'), 'attribute': 'order'},
                {'name': _('Transformation'), 'attribute': encapsulate(lambda transformation: unicode(transformation))},
                {'name': _('Arguments'), 'attribute': 'arguments'}
            ],
            'hide_link': True,
            'hide_object': True,
            'navigation_object_list': ['content_object'],
            'title': _('Transformations for: %s') % self.content_object,
        }