Refactor the document mailing views.
Add support for sending multiple documents via email at the same time.
This commit is contained in:
Roberto Rosario
@@ -1,17 +1,11 @@
|
||||
from __future__ import absolute_import, unicode_literals
|
||||
|
||||
from django.conf import settings
|
||||
from django.contrib import messages
|
||||
from django.contrib.sites.models import Site
|
||||
from django.core.urlresolvers import reverse
|
||||
from django.http import HttpResponseRedirect
|
||||
from django.shortcuts import render_to_response
|
||||
from django.template import Context, RequestContext, Template
|
||||
from django.template import Context, Template
|
||||
from django.utils.html import strip_tags
|
||||
from django.utils.translation import ugettext_lazy as _
|
||||
from django.utils.translation import ungettext, ugettext_lazy as _
|
||||
|
||||
from acls.models import AccessControlList
|
||||
from common.generics import SingleObjectListView
|
||||
from common.generics import MultipleObjectFormActionView, SingleObjectListView
|
||||
from documents.models import Document
|
||||
|
||||
from .forms import DocumentMailForm
|
||||
@@ -32,96 +26,81 @@ class LogEntryListView(SingleObjectListView):
|
||||
view_permission = permission_view_error_log
|
||||
|
||||
|
||||
def send_document_link(request, document_id=None, document_id_list=None, as_attachment=False):
|
||||
if document_id:
|
||||
documents = Document.objects.filter(pk=document_id)
|
||||
elif document_id_list:
|
||||
documents = Document.objects.filter(pk__in=document_id_list)
|
||||
class MailDocumentView(MultipleObjectFormActionView):
|
||||
as_attachment = True
|
||||
form_class = DocumentMailForm
|
||||
model = Document
|
||||
object_permission = permission_mailing_send_document
|
||||
|
||||
if as_attachment:
|
||||
permission = permission_mailing_send_document
|
||||
else:
|
||||
permission = permission_mailing_link
|
||||
|
||||
documents = AccessControlList.objects.filter_by_access(
|
||||
permission, request.user, queryset=documents
|
||||
success_message = _('%(count)d document queued for email delivery')
|
||||
success_message_plural = _(
|
||||
'%(count)d documents queued for email delivery'
|
||||
)
|
||||
title = 'Email document'
|
||||
title_plural = 'Email documents'
|
||||
title_document = 'Email document: %s'
|
||||
|
||||
if not documents:
|
||||
messages.error(request, _('Must provide at least one document.'))
|
||||
return HttpResponseRedirect(
|
||||
request.META.get(
|
||||
'HTTP_REFERER', reverse(settings.LOGIN_REDIRECT_URL)
|
||||
def get_extra_context(self):
|
||||
queryset = self.get_queryset()
|
||||
|
||||
result = {
|
||||
'submit_icon': 'fa fa-envelope',
|
||||
'submit_label': _('Send'),
|
||||
'title': ungettext(
|
||||
self.title,
|
||||
self.title_plural,
|
||||
queryset.count()
|
||||
)
|
||||
}
|
||||
|
||||
if queryset.count() == 1:
|
||||
result.update(
|
||||
{
|
||||
'object': queryset.first(),
|
||||
'title': _(self.title_document) % queryset.first()
|
||||
}
|
||||
)
|
||||
|
||||
return result
|
||||
|
||||
def get_form_extra_kwargs(self):
|
||||
return {
|
||||
'as_attachment': self.as_attachment
|
||||
}
|
||||
|
||||
def object_action(self, form, instance):
|
||||
context = Context({
|
||||
'link': 'http://%s%s' % (
|
||||
Site.objects.get_current().domain,
|
||||
instance.get_absolute_url()
|
||||
),
|
||||
'document': instance
|
||||
})
|
||||
body_template = Template(form.cleaned_data['body'])
|
||||
body_html_content = body_template.render(context)
|
||||
body_text_content = strip_tags(body_html_content)
|
||||
|
||||
subject_template = Template(form.cleaned_data['subject'])
|
||||
subject_text = strip_tags(subject_template.render(context))
|
||||
|
||||
task_send_document.apply_async(
|
||||
args=(
|
||||
subject_text, body_text_content, self.request.user.email,
|
||||
form.cleaned_data['email']
|
||||
), kwargs={
|
||||
'document_id': instance.pk,
|
||||
'as_attachment': self.as_attachment
|
||||
}
|
||||
)
|
||||
|
||||
post_action_redirect = reverse('documents:document_list_recent')
|
||||
|
||||
next = request.POST.get(
|
||||
'next', request.GET.get(
|
||||
'next', request.META.get('HTTP_REFERER', post_action_redirect)
|
||||
)
|
||||
)
|
||||
|
||||
for document in documents:
|
||||
document.add_as_recent_document_for_user(request.user)
|
||||
|
||||
if request.method == 'POST':
|
||||
form = DocumentMailForm(request.POST, as_attachment=as_attachment)
|
||||
if form.is_valid():
|
||||
|
||||
for document in documents:
|
||||
context = Context({
|
||||
'link': 'http://%s%s' % (
|
||||
Site.objects.get_current().domain,
|
||||
document.get_absolute_url()
|
||||
),
|
||||
'document': document
|
||||
})
|
||||
body_template = Template(form.cleaned_data['body'])
|
||||
body_html_content = body_template.render(context)
|
||||
body_text_content = strip_tags(body_html_content)
|
||||
|
||||
subject_template = Template(form.cleaned_data['subject'])
|
||||
subject_text = strip_tags(subject_template.render(context))
|
||||
|
||||
task_send_document.apply_async(
|
||||
args=(
|
||||
subject_text, body_text_content, request.user.email,
|
||||
form.cleaned_data['email']
|
||||
), kwargs={
|
||||
'document_id': document.pk,
|
||||
'as_attachment': as_attachment
|
||||
}
|
||||
)
|
||||
|
||||
# TODO: Pluralize
|
||||
messages.success(
|
||||
request, _('Successfully queued for delivery via email.')
|
||||
)
|
||||
return HttpResponseRedirect(next)
|
||||
else:
|
||||
form = DocumentMailForm(as_attachment=as_attachment)
|
||||
|
||||
context = {
|
||||
'form': form,
|
||||
'next': next,
|
||||
'submit_label': _('Send'),
|
||||
'submit_icon': 'fa fa-envelope'
|
||||
}
|
||||
if documents.count() == 1:
|
||||
context['object'] = documents.first()
|
||||
if as_attachment:
|
||||
context['title'] = _('Email document: %s') % ', '.join([unicode(d) for d in documents])
|
||||
else:
|
||||
context['title'] = _('Email link for document: %s') % ', '.join([unicode(d) for d in documents])
|
||||
elif documents.count() > 1:
|
||||
if as_attachment:
|
||||
context['title'] = _('Email documents: %s') % ', '.join([unicode(d) for d in documents])
|
||||
else:
|
||||
context['title'] = _('Email links for documents: %s') % ', '.join([unicode(d) for d in documents])
|
||||
|
||||
return render_to_response(
|
||||
'appearance/generic_form.html', context,
|
||||
context_instance=RequestContext(request)
|
||||
class MailDocumentLinkView(MailDocumentView):
|
||||
as_attachment = False
|
||||
object_permission = permission_mailing_link
|
||||
success_message = _('%(count)d document link queued for email delivery')
|
||||
success_message_plural = _(
|
||||
'%(count)d document links queued for email delivery'
|
||||
)
|
||||
title = 'Email document link'
|
||||
title_plural = 'Email document links'
|
||||
title_document = 'Email link for document: %s'
|
||||
|
||||
Reference in New Issue
Block a user