from __future__ import absolute_import import logging from django.utils.translation import ugettext_lazy as _ from django.http import HttpResponseRedirect from django.shortcuts import render_to_response, get_object_or_404 from django.template import RequestContext from django.contrib import messages from django.core.urlresolvers import reverse from django.core.exceptions import PermissionDenied from documents.permissions import PERMISSION_DOCUMENT_VIEW from documents.models import Document from documents.views import document_list from permissions.models import Permission from common.utils import encapsulate from acls.models import AccessEntry from acls.views import acl_list_for from acls.utils import apply_default_acls from .models import Folder from .forms import FolderForm, FolderListForm from .permissions import (PERMISSION_FOLDER_CREATE, PERMISSION_FOLDER_EDIT, PERMISSION_FOLDER_DELETE, PERMISSION_FOLDER_REMOVE_DOCUMENT, PERMISSION_FOLDER_VIEW, PERMISSION_FOLDER_ADD_DOCUMENT) logger = logging.getLogger(__name__) def folder_list(request, queryset=None, extra_context=None): context = { 'title': _(u'folders'), 'extra_columns': [ {'name': _(u'created'), 'attribute': 'datetime_created'}, {'name': _(u'documents'), 'attribute': encapsulate(lambda x: x.folderdocument_set.count())} ], 'hide_link': True, } if extra_context: context.update(extra_context) queryset = queryset if not (queryset is None) else Folder.objects.all() try: Permission.objects.check_permissions(request.user, [PERMISSION_FOLDER_VIEW]) except PermissionDenied: queryset = AccessEntry.objects.filter_objects_by_access(PERMISSION_FOLDER_VIEW, request.user, queryset) context['object_list'] = queryset return render_to_response('generic_list.html', context, context_instance=RequestContext(request) ) def folder_create(request): Permission.objects.check_permissions(request.user, [PERMISSION_FOLDER_CREATE]) if request.method == 'POST': form = FolderForm(request.POST) if form.is_valid(): folder, created = Folder.objects.get_or_create(user=request.user, title=form.cleaned_data['title']) if created: apply_default_acls(folder, request.user) messages.success(request, _(u'Folder created successfully')) return HttpResponseRedirect(reverse('folder_list')) else: messages.error(request, _(u'A folder named: %s, already exists.') % form.cleaned_data['title']) else: form = FolderForm() return render_to_response('generic_form.html', { 'title': _(u'create folder'), 'form': form, }, context_instance=RequestContext(request)) def folder_edit(request, folder_id): folder = get_object_or_404(Folder, pk=folder_id) try: Permission.objects.check_permissions(request.user, [PERMISSION_FOLDER_EDIT]) except PermissionDenied: AccessEntry.objects.check_access(PERMISSION_FOLDER_EDIT, request.user, folder) if request.method == 'POST': form = FolderForm(request.POST) if form.is_valid(): folder.title = form.cleaned_data['title'] try: folder.save() messages.success(request, _(u'Folder edited successfully')) return HttpResponseRedirect(reverse('folder_list')) except Exception, e: messages.error(request, _(u'Error editing folder; %s') % e) else: form = FolderForm(instance=folder) return render_to_response('generic_form.html', { 'title': _(u'edit folder: %s') % folder, 'form': form, 'object': folder, 'object_name': _(u'folder'), }, context_instance=RequestContext(request)) def folder_delete(request, folder_id): folder = get_object_or_404(Folder, pk=folder_id) try: Permission.objects.check_permissions(request.user, [PERMISSION_FOLDER_DELETE]) except PermissionDenied: AccessEntry.objects.check_access(PERMISSION_FOLDER_DELETE, request.user, folder) post_action_redirect = reverse('folder_list') previous = request.POST.get('previous', request.GET.get('previous', request.META.get('HTTP_REFERER', '/'))) next = request.POST.get('next', request.GET.get('next', post_action_redirect if post_action_redirect else request.META.get('HTTP_REFERER', '/'))) if request.method == 'POST': try: folder.delete() messages.success(request, _(u'Folder: %s deleted successfully.') % folder) except Exception, e: messages.error(request, _(u'Folder: %(folder)s delete error: %(error)s') % { 'folder': folder, 'error': e}) return HttpResponseRedirect(next) context = { 'object_name': _(u'folder'), 'delete_view': True, 'previous': previous, 'next': next, 'object': folder, 'title': _(u'Are you sure you with to delete the folder: %s?') % folder, 'form_icon': u'folder_delete.png', } return render_to_response('generic_confirm.html', context, context_instance=RequestContext(request)) def folder_view(request, folder_id): folder = get_object_or_404(Folder, pk=folder_id) try: Permission.objects.check_permissions(request.user, [PERMISSION_FOLDER_VIEW]) except PermissionDenied: AccessEntry.objects.check_access(PERMISSION_FOLDER_VIEW, request.user, folder) context = { 'hide_links': True, 'multi_select_as_buttons': True, 'object': folder, 'object_name': _(u'folder'), } return document_list( request, object_list=folder.documents, title=_(u'documents in folder: %s') % folder, extra_context=context ) def folder_add_document(request, document_id): document = get_object_or_404(Document, pk=document_id) try: Permission.objects.check_permissions(request.user, [PERMISSION_FOLDER_ADD_DOCUMENT]) except PermissionDenied: AccessEntry.objects.check_access(PERMISSION_FOLDER_ADD_DOCUMENT, request.user, document) next = request.POST.get('next', request.GET.get('next', request.META.get('HTTP_REFERER', '/'))) if request.method == 'POST': form = FolderListForm(request.POST, user=request.user) if form.is_valid(): folder = form.cleaned_data['folder'] if folder.add_document(document): messages.success(request, _(u'Document: %(document)s added to folder: %(folder)s successfully.') % { 'document': document, 'folder': folder}) else: messages.warning(request, _(u'Document: %(document)s is already in folder: %(folder)s.') % { 'document': document, 'folder': folder}) return HttpResponseRedirect(next) else: form = FolderListForm(user=request.user) return render_to_response('generic_form.html', { 'title': _(u'add document "%s" to a folder') % document, 'form': form, 'object': document, 'next': next, }, context_instance=RequestContext(request)) def document_folder_list(request, document_id): document = get_object_or_404(Document, pk=document_id) try: Permission.objects.check_permissions(request.user, [PERMISSION_DOCUMENT_VIEW]) except PermissionDenied: AccessEntry.objects.check_access(PERMISSION_DOCUMENT_VIEW, request.user, document) return folder_list( request, queryset=Folder.objects.filter(folderdocument__document=document), extra_context={ 'title': _(u'folders containing: %s') % document, 'object': document, } ) def folder_document_remove(request, folder_id, document_id=None, document_id_list=None): post_action_redirect = None folder = get_object_or_404(Folder, pk=folder_id) if document_id: folder_documents = [get_object_or_404(Document, pk=document_id)] elif document_id_list: folder_documents = [get_object_or_404(Document, pk=document_id) for document_id in document_id_list.split(',')] else: messages.error(request, _(u'Must provide at least one folder document.')) return HttpResponseRedirect(request.META.get('HTTP_REFERER', '/')) logger.debug('folder_documents (pre permission check): %s' % folder_documents) try: Permission.objects.check_permissions(request.user, [PERMISSION_FOLDER_REMOVE_DOCUMENT]) except PermissionDenied: folder_documents = AccessEntry.objects.filter_objects_by_access(PERMISSION_FOLDER_REMOVE_DOCUMENT, request.user, folder_documents, exception_on_empty=True) logger.debug('folder_documents (post permission check): %s' % folder_documents) previous = request.POST.get('previous', request.GET.get('previous', request.META.get('HTTP_REFERER', '/'))) next = request.POST.get('next', request.GET.get('next', post_action_redirect if post_action_redirect else request.META.get('HTTP_REFERER', '/'))) if request.method == 'POST': for folder_document in folder_documents: try: folder.remove_document(folder_document) messages.success(request, _(u'Document: %s removed successfully.') % folder_document) except Exception, e: messages.error(request, _(u'Document: %(document)s delete error: %(error)s') % { 'document': folder_document, 'error': e}) return HttpResponseRedirect(next) context = { 'object_name': _(u'folder document'), 'previous': previous, 'next': next, 'form_icon': u'delete.png', 'object': folder } if len(folder_documents) == 1: context['object'] = folder_documents[0] context['title'] = _(u'Are you sure you wish to remove the document: %(document)s from the folder "%(folder)s"?') % { 'document': ', '.join([unicode(d) for d in folder_documents]), 'folder': folder} elif len(folder_documents) > 1: context['title'] = _(u'Are you sure you wish to remove the documents: %(documents)s from the folder "%(folder)s"?') % { 'documents': ', '.join([unicode(d) for d in folder_documents]), 'folder': folder} return render_to_response('generic_confirm.html', context, context_instance=RequestContext(request)) def folder_document_multiple_remove(request, folder_id): return folder_document_remove(request, folder_id, document_id_list=request.GET.get('id_list', [])) def folder_acl_list(request, folder_pk): folder = get_object_or_404(Folder, pk=folder_pk) logger.debug('folder: %s' % folder) return acl_list_for( request, folder, extra_context={ 'object': folder, } )