from django.contrib import messages from django.core.urlresolvers import reverse from django.db import transaction from django.shortcuts import redirect, render from django.utils import timezone from django.utils.translation import ungettext, ugettext_lazy as _ from orchestra.admin.decorators import action_with_confirmation from orchestra.core import services @transaction.atomic @action_with_confirmation() def disable(modeladmin, request, queryset): num = 0 for account in queryset: account.disable() modeladmin.log_change(request, account, _("Disabled")) num += 1 msg = ungettext( _("Selected account and related services has been disabled."), _("%s selected accounts and related services have been disabled.") % num, num) modeladmin.message_user(request, msg) disable.url_name = 'disable' disable.verbose_name = _("Disable") def list_contacts(modeladmin, request, queryset): ids = queryset.values_list('id', flat=True) if not ids: message.warning(request, "Select at least one account.") return url = reverse('admin:contacts_contact_changelist') url += '?account__in=%s' % ','.join(map(str, ids)) return redirect(url) list_contacts.verbose_name = _("List contacts") def service_report(modeladmin, request, queryset): accounts = [] fields = [] # First we get related manager names to fire a prefetch related for name, field in queryset.model._meta._name_map.iteritems(): model = field[0].model if model in services.get() and model != queryset.model: fields.append((model, name)) sorted(fields, key=lambda i: i[0]._meta.verbose_name_plural.lower()) fields = [field for model, field in fields] for account in queryset.prefetch_related(*fields): items = [] for field in fields: related_manager = getattr(account, field) items.append((related_manager.model._meta, related_manager.all())) accounts.append((account, items)) context = { 'accounts': accounts, 'date': timezone.now().today() } return render(request, 'admin/accounts/account/service_report.html', context)