from django.contrib import messages from django.contrib.admin import helpers from django.contrib.admin.utils import NestedObjects, quote, model_ngettext from django.contrib.auth import get_permission_codename from django.core.urlresolvers import reverse, NoReverseMatch from django.db import router from django.shortcuts import redirect, render from django.template.response import TemplateResponse from django.utils import timezone from django.utils.encoding import force_text from django.utils.html import format_html from django.utils.text import capfirst from django.utils.translation import ungettext, ugettext_lazy as _ from orchestra.admin.decorators import action_with_confirmation from orchestra.core import services from . import settings @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: messages.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): # TODO resources accounts = [] fields = [] registered_services = services.get() # First we get related manager names to fire a prefetch related for name, field in queryset.model._meta.fields_map.items(): model = field.related_model if model in registered_services and model != queryset.model: fields.append((model, name)) sorted(fields, key=lambda f: f[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, settings.ACCOUNTS_SERVICE_REPORT_TEMPLATE, context) def delete_related_services(modeladmin, request, queryset): opts = modeladmin.model._meta app_label = opts.app_label using = router.db_for_write(modeladmin.model) collector = NestedObjects(using=using) collector.collect(queryset) registered_services = services.get() related_services = [] to_delete = [] user = request.user admin_site = modeladmin.admin_site def format(obj): has_admin = obj.__class__ in admin_site._registry opts = obj._meta no_edit_link = '%s: %s' % (capfirst(opts.verbose_name), force_text(obj)) if has_admin: try: admin_url = reverse('admin:%s_%s_change' % (opts.app_label, opts.model_name), None, (quote(obj._get_pk_val()),) ) except NoReverseMatch: # Change url doesn't exist -- don't display link to edit return no_edit_link p = '%s.%s' % (opts.app_label, get_permission_codename('delete', opts)) if not user.has_perm(p): perms_needed.add(opts.verbose_name) # Display a link to the admin page. return format_html('{}: {}', capfirst(opts.verbose_name), admin_url, obj) else: # Don't display link to edit, because it either has no # admin or is edited inline. return no_edit_link def format_nested(objs, result): if isinstance(objs, list): current = [] for obj in objs: format_nested(obj, current) result.append(current) else: result.append(format(objs)) for account in collector.nested(): if isinstance(account, list): current = [] is_service = False for service in account: if type(service) in registered_services: if service == main_systemuser: continue current.append(format(service)) to_delete.append(service) is_service = True elif is_service and isinstance(service, list): nested = [] format_nested(service, nested) current.append(nested) is_service = False else: is_service = False related_services.append(current) elif isinstance(account, modeladmin.model): # Prevent the deletion of the main system user, which will delete the account main_systemuser = account.main_systemuser related_services.append(format(account)) # The user has already confirmed the deletion. # Do the deletion and return a None to display the change list view again. if request.POST.get('post'): n = queryset.count() if n: for obj in to_delete: obj_display = force_text(obj) modeladmin.log_deletion(request, obj, obj_display) # TODO This probably will fail in certain conditions, just capture exception obj.delete() modeladmin.message_user(request, _("Successfully deleted %(count)d %(items)s.") % { "count": n, "items": model_ngettext(modeladmin.opts, n) }, messages.SUCCESS) # Return None to display the change list page again. return None if len(queryset) == 1: objects_name = force_text(opts.verbose_name) else: objects_name = force_text(opts.verbose_name_plural) context = dict( modeladmin.admin_site.each_context(request), title=_("Are you sure?"), objects_name=objects_name, deletable_objects=[related_services], model_count=dict(collector.model_count).items(), queryset=queryset, opts=opts, action_checkbox_name=helpers.ACTION_CHECKBOX_NAME, ) request.current_app = modeladmin.admin_site.name # Display the confirmation page return TemplateResponse(request, modeladmin.delete_selected_confirmation_template or [ "admin/%s/%s/delete_selected_confirmation.html" % (app_label, opts.model_name), "admin/%s/delete_selected_confirmation.html" % app_label, "admin/delete_selected_confirmation.html" ], context) delete_related_services.short_description = _("Delete related services")