from functools import partial from django.contrib import admin from django.core.exceptions import PermissionDenied from django.core.mail import send_mass_mail from django.shortcuts import render from django.utils.translation import gettext_lazy as _ from django.utils.translation import ngettext from .. import settings from .decorators import action_with_confirmation from .forms import SendEmailForm class SendEmail(object): """ Form wizard for billing orders admin action """ short_description = _("Send email") form = SendEmailForm template = 'admin/orchestra/generic_confirmation.html' default_from = settings.ORCHESTRA_DEFAULT_SUPPORT_FROM_EMAIL __name__ = 'semd_email' def __call__(self, modeladmin, request, queryset): """ make this monster behave like a function """ self.modeladmin = modeladmin self.queryset = queryset self.opts = modeladmin.model._meta app_label = self.opts.app_label self.context = { 'action_name': _("Send email"), 'action_value': self.__name__, 'opts': self.opts, 'app_label': app_label, 'queryset': queryset, 'action_checkbox_name': admin.helpers.ACTION_CHECKBOX_NAME, } return self.write_email(request) def write_email(self, request): if not request.user.is_superuser: raise PermissionDenied() initial={ 'email_from': self.default_from, 'to': ' '.join(self.get_email_addresses()) } form = self.form(initial=initial) if request.POST.get('post'): form = self.form(request.POST, initial=initial) if form.is_valid(): options = { 'email_from': form.cleaned_data['email_from'], 'extra_to': form.cleaned_data['extra_to'], 'subject': form.cleaned_data['subject'], 'message': form.cleaned_data['message'], } return self.confirm_email(request, **options) self.context.update({ 'title': _("Send e-mail to %s") % self.opts.verbose_name_plural, 'content_title': "", 'form': form, 'submit_value': _("Continue"), }) # Display confirmation page return render(request, self.template, self.context) def get_email_addresses(self): return self.queryset.values_list('email', flat=True) def confirm_email(self, request, **options): email_from = options['email_from'] extra_to = options['extra_to'] subject = options['subject'] message = options['message'] # The user has already confirmed if request.POST.get('post') == 'email_confirmation': emails = [] num = 0 for email in self.get_email_addresses(): emails.append((subject, message, email_from, [email])) num += 1 if extra_to: emails.append((subject, message, email_from, extra_to)) send_mass_mail(emails, fail_silently=False) msg = ngettext( _("Message has been sent to one %s.") % self.opts.verbose_name_plural, _("Message has been sent to %i %s.") % (num, self.opts.verbose_name_plural), num ) self.modeladmin.message_user(request, msg) return None form = self.form(initial={ 'email_from': email_from, 'extra_to': ', '.join(extra_to), 'subject': subject, 'message': message }) self.context.update({ 'title': _("Are you sure?"), 'content_message': _( "Are you sure you want to send the following message to the following %s?" ) % self.opts.verbose_name_plural, 'display_objects': ["%s (%s)" % (contact, email) for contact, email in zip(self.queryset, self.get_email_addresses())], 'form': form, 'subject': subject, 'message': message, 'post_value': 'email_confirmation', }) # Display the confirmation page return render(request, self.template, self.context) def base_disable(modeladmin, request, queryset, disable=True): num = 0 action_name = _("disabled") if disable else _("enabled") for obj in queryset: obj.disable() if disable else obj.enable() modeladmin.log_change(request, obj, action_name.capitalize()) num += 1 opts = modeladmin.model._meta context = { 'action_name': action_name, 'verbose_name': opts.verbose_name, 'verbose_name_plural': opts.verbose_name_plural, 'num': num } msg = ngettext( _("Selected %(verbose_name)s and related services has been %(action_name)s.") % context, _("%(num)s selected %(verbose_name_plural)s and related services have been %(action_name)s.") % context, num) modeladmin.message_user(request, msg) @action_with_confirmation() def disable(modeladmin, request, queryset): return base_disable(modeladmin, request, queryset) disable.url_name = 'disable' disable.short_description = _("Disable") @action_with_confirmation() def enable(modeladmin, request, queryset): return base_disable(modeladmin, request, queryset, disable=False) enable.url_name = 'enable' enable.short_description = _("Enable")