277 lines
11 KiB
Python
277 lines
11 KiB
Python
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.views.generic.list_detail import object_list
|
|
from django.core.urlresolvers import reverse
|
|
from django.core.exceptions import PermissionDenied
|
|
|
|
from documents.literals import PERMISSION_DOCUMENT_VIEW
|
|
from documents.models import Document
|
|
from permissions.api import check_permissions
|
|
from common.utils import encapsulate
|
|
|
|
from folders.models import Folder, FolderDocument
|
|
from folders.forms import FolderForm, AddDocumentForm
|
|
|
|
|
|
def folder_list(request, queryset=None, extra_context=None):
|
|
context = {
|
|
'title': _(u'folders'),
|
|
'multi_select_as_buttons': True,
|
|
'extra_columns': [
|
|
{'name': _(u'created'), 'attribute': 'datetime_created'},
|
|
{'name': _(u'documents'), 'attribute': encapsulate(lambda x: x.folderdocument_set.count())}
|
|
]
|
|
}
|
|
if extra_context:
|
|
context.update(extra_context)
|
|
|
|
return object_list(
|
|
request,
|
|
queryset=queryset if not (queryset is None) else Folder.objects.filter(user=request.user),
|
|
template_name='generic_list.html',
|
|
extra_context=context,
|
|
)
|
|
|
|
|
|
def folder_create(request):
|
|
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:
|
|
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)
|
|
|
|
if not request.user.is_staff and not request.user.is_superuser and not request.user == folder.user:
|
|
raise PermissionDenied
|
|
|
|
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)
|
|
|
|
if not request.user.is_staff and not request.user.is_superuser and not request.user == folder.user:
|
|
raise PermissionDenied
|
|
|
|
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)
|
|
|
|
if not request.user.is_staff and not request.user.is_superuser and not request.user == folder.user:
|
|
raise PermissionDenied
|
|
|
|
return render_to_response('generic_list.html', {
|
|
'object_list': [fd.document for fd in folder.folderdocument_set.all()],
|
|
'hide_links': True,
|
|
'title': _(u'documents in folder: %s') % folder,
|
|
'multi_select_as_buttons': True,
|
|
'object': folder,
|
|
'object_name': _(u'folder'),
|
|
}, context_instance=RequestContext(request))
|
|
|
|
|
|
def folder_add_document_sidebar(request, document_id):
|
|
check_permissions(request.user, [PERMISSION_DOCUMENT_VIEW])
|
|
document = get_object_or_404(Document, pk=document_id)
|
|
|
|
previous = request.META.get('HTTP_REFERER', '/')
|
|
|
|
if request.method == 'POST':
|
|
form = AddDocumentForm(request.POST, user=request.user)
|
|
if form.is_valid():
|
|
if form.cleaned_data['existing_folder']:
|
|
folder = form.cleaned_data['existing_folder']
|
|
elif form.cleaned_data['title']:
|
|
folder, created = Folder.objects.get_or_create(user=request.user, title=form.cleaned_data['title'])
|
|
if created:
|
|
messages.success(request, _(u'Folder created successfully'))
|
|
else:
|
|
messages.error(request, _(u'A folder named: %s, already exists.') % form.cleaned_data['title'])
|
|
return HttpResponseRedirect(previous)
|
|
else:
|
|
messages.error(request, _(u'Must specify a new folder or an existing one.'))
|
|
return HttpResponseRedirect(previous)
|
|
|
|
folder_document, created = FolderDocument.objects.get_or_create(folder=folder, document=document)
|
|
if created:
|
|
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(previous)
|
|
|
|
|
|
def folder_add_document(request, document_id):
|
|
# TODO: merge with folder_add_document_sidebar
|
|
check_permissions(request.user, [PERMISSION_DOCUMENT_VIEW])
|
|
document = get_object_or_404(Document, pk=document_id)
|
|
|
|
next = request.POST.get('next', request.GET.get('next', request.META.get('HTTP_REFERER', '/')))#reverse('document_tags', args=[document.pk]))))
|
|
|
|
if request.method == 'POST':
|
|
form = AddDocumentForm(request.POST, user=request.user)
|
|
if form.is_valid():
|
|
if form.cleaned_data['existing_folder']:
|
|
folder = form.cleaned_data['existing_folder']
|
|
elif form.cleaned_data['title']:
|
|
folder, created = Folder.objects.get_or_create(user=request.user, title=form.cleaned_data['title'])
|
|
if created:
|
|
messages.success(request, _(u'Folder "%s" created successfully') % form.cleaned_data['title'])
|
|
else:
|
|
messages.error(request, _(u'A folder named: %s, already exists.') % form.cleaned_data['title'])
|
|
return HttpResponseRedirect(next)
|
|
else:
|
|
messages.error(request, _(u'Must specify a new folder or an existing one.'))
|
|
return HttpResponseRedirect(next)
|
|
|
|
folder_document, created = FolderDocument.objects.get_or_create(folder=folder, document=document)
|
|
if created:
|
|
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 = AddDocumentForm(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):
|
|
check_permissions(request.user, [PERMISSION_DOCUMENT_VIEW])
|
|
document = get_object_or_404(Document, pk=document_id)
|
|
|
|
return folder_list(
|
|
request,
|
|
queryset=Folder.objects.filter(user=request.user).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(FolderDocument, folder__pk=folder_id, document__pk=document_id)]
|
|
elif document_id_list:
|
|
folder_documents = [get_object_or_404(FolderDocument, folder__pk=folder_id, 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', '/'))
|
|
|
|
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_document.delete()
|
|
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_documents[0].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_documents[0].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', []))
|