from collections import defaultdict from django.contrib import messages from django.contrib.admin import helpers from django.shortcuts import render from django.utils.safestring import mark_safe from django.utils.translation import ungettext, ugettext_lazy as _ from orchestra.admin.utils import get_object_from_url, change_url from orchestra.contrib.orchestration.helpers import message_user from orchestra.utils.python import OrderedSet from . import manager, Operation from .models import BackendOperation, Route, Server def retry_backend(modeladmin, request, queryset): related_operations = queryset.values_list('operations__id', flat=True).distinct() related_operations = BackendOperation.objects.filter(pk__in=related_operations) related_operations = related_operations.select_related('log__server').prefetch_related('instance') if request.POST.get('post') == 'generic_confirmation': operations = [] for operation in related_operations: if operation.instance: op = Operation.load(operation) operations.append(op) if not operations: messages.warning(request, _("No backend operation has been executed.")) else: logs = Operation.execute(operations) message_user(request, logs) return opts = modeladmin.model._meta display_objects = [] deleted_objects = [] for op in related_operations: if not op.instance: deleted_objects.append(op) else: context = { 'backend': op.log.backend, 'action': op.action, 'instance': op.instance, 'instance_url': change_url(op.instance), 'server': op.log.server, 'server_url': change_url(op.log.server), } display_objects.append(mark_safe( '%(backend)s.%(action)s(%(instance)s) @ %(server)s' % context )) context = { 'title': _("Are you sure to execute the following backends?"), 'action_name': _('Retry backend'), 'action_value': 'retry_backend', 'display_objects': display_objects, 'deleted_objects': deleted_objects, 'queryset': queryset, 'opts': opts, 'app_label': opts.app_label, 'action_checkbox_name': helpers.ACTION_CHECKBOX_NAME, 'obj': get_object_from_url(modeladmin, request), } return render(request, 'admin/orchestration/backends/retry.html', context) retry_backend.short_description = _("Retry") retry_backend.url_name = 'retry' def orchestrate(modeladmin, request, queryset): operations = set() action = Operation.SAVE operations = OrderedSet() if queryset.model is Route: for route in queryset: routes = [route] backend = route.backend_class if action not in backend.actions: continue for instance in backend.model_class().objects.all(): if route.matches(instance): operations.add(Operation(backend, instance, action, routes=routes)) elif queryset.model is Server: models = set() for server in queryset: routes = server.routes.all() for route in routes.filter(is_active=True): model = route.backend_class.model_class() models.add(model) querysets = [model.objects.order_by('id') for model in models] route_cache = {} for model in models: for instance in model.objects.all(): manager.collect(instance, action, operations=operations, route_cache=route_cache) routes = [] result = [] for operation in operations: routes = [route for route in operation.routes if route.host in queryset] operation.routes = routes if routes: result.append(operation) operations = result if not operations: messages.warning(request, _("No operations.")) if request.POST.get('post') == 'generic_confirmation': logs = Operation.execute(operations) message_user(request, logs) return opts = modeladmin.model._meta display_objects = {} for operation in operations: try: display_objects[operation.backend].append(operation) except KeyError: display_objects[operation.backend] = [operation] context = { 'title': _("Are you sure to execute the following operations?"), 'action_name': _('Orchestrate'), 'action_value': 'orchestrate', 'display_objects': display_objects, 'queryset': queryset, 'opts': opts, 'app_label': opts.app_label, 'action_checkbox_name': helpers.ACTION_CHECKBOX_NAME, 'obj': get_object_from_url(modeladmin, request), } return render(request, 'admin/orchestration/orchestrate.html', context)