django-orchestra/orchestra/contrib/orchestration/helpers.py

import textwrap

from django.contrib import messages
from django.core.mail import mail_admins
from django.urls import reverse, NoReverseMatch
from django.utils.html import escape
from django.utils.safestring import mark_safe
from django.utils.translation import ungettext, ugettext_lazy as _

from orchestra import settings as orchestra_settings
from orchestra.admin.utils import change_url


def get_backends_help_text(backends):
    help_texts = {}
    for backend in backends:
        help_text = backend.__doc__ or ''
        context = {
            'model': backend.model,
            'related_models': str(backend.related_models),
            'script_executable': backend.script_executable,
            'script_method': '.'.join(
                (backend.script_method.__module__, backend.script_method.__name__)),
            'function_method': '.'.join(
                (backend.function_method.__module__, backend.function_method.__name__)),
            'actions': str(backend.actions),
        }
        help_text += textwrap.dedent("""
            - Model: <tt>'%(model)s'</tt>
            - Related models: <tt>%(related_models)s</tt>
            - Script executable: <tt>%(script_executable)s</tt>
            - Script method: <tt>%(script_method)s</tt>
            - Function method: <tt>%(function_method)s</tt>
            - Actions: <tt>%(actions)s</tt>
            """
        ) % context
        help_text = help_text.lstrip().splitlines()
        help_settings = ['']
        if backend.doc_settings:
            module, names = backend.doc_settings
            for name in names:
                value = getattr(module, name)
                if isinstance(value, str):
                    help_settings.append("<tt>%s = '%s'</tt>" % (name, value))
                else:
                    help_settings.append("<tt>%s = %s</tt>" % (name, value))
        help_text += help_settings
        help_texts[backend.get_name()] = '<br>'.join(help_text)
    return help_texts


def get_instance_url(operation):
    try:
        url = change_url(operation.instance)
    except NoReverseMatch:
        alt_repr = '%s-%s' % (operation.content_type, operation.object_id)
        return _("Deleted {0}").format(operation.instance_repr or alt_repr)
    return orchestra_settings.ORCHESTRA_SITE_URL + url


def send_report(method, args, log):
    server = args[0]
    backend = method.__self__.__class__.__name__
    subject = '[Orchestra] %s execution %s on %s'  % (backend, log.state, server)
    separator = "\n%s\n\n" % ('~ '*40,)
    operations = '\n'.join(
        [' '.join((op.action, get_instance_url(op))) for op in log.operations.all()]
    )
    log_url = reverse('admin:orchestration_backendlog_change', args=(log.pk,))
    log_url = orchestra_settings.ORCHESTRA_SITE_URL + log_url
    message = separator.join([
        "[EXIT CODE] %s" % log.exit_code,
        "[STDERR]\n%s" % log.stderr,
        "[STDOUT]\n%s" % log.stdout,
        "[SCRIPT]\n%s" % log.script,
        "[TRACEBACK]\n%s" % log.traceback,
        "[OPERATIONS]\n%s" % operations,
        "[BACKEND LOG] %s" % log_url,
    ])
    html_message = '\n\n'.join([
        '<h4 style="color:#505050;">Exit code %s</h4>' % log.exit_code,
        '<h4 style="color:#505050;">Stderr</h4>'
            '<pre style="margin-left:20px;font-size:11px">%s</pre>' % escape(log.stderr),
        '<h4 style="color:#505050;">Stdout</h4>'
            '<pre style="margin-left:20px;font-size:11px">%s</pre>' % escape(log.stdout),
        '<h4 style="color:#505050;">Script</h4>'
            '<pre style="margin-left:20px;font-size:11px">%s</pre>' % escape(log.script),
        '<h4 style="color:#505050;">Traceback</h4>'
            '<pre style="margin-left:20px;font-size:11px">%s</pre>' % escape(log.traceback),
        '<h4 style="color:#505050;">Operations</h4>'
            '<pre style="margin-left:20px;font-size:11px">%s</pre>' % escape(operations),
        '<h4 style="color:#505050;">Backend log <a href="%s">%s</h4>' % (log_url, log_url),
    ])
    mail_admins(subject, message, html_message=html_message)


def get_backend_url(ids):
    if len(ids) == 1:
        return reverse('admin:orchestration_backendlog_change', args=ids)
    elif len(ids) > 1:
        url = reverse('admin:orchestration_backendlog_changelist')
        return url + '?id__in=%s' % ','.join(map(str, ids))
    return ''


def get_messages(logs):
    messages = []
    total, successes, run_async = 0, 0, 0
    ids = []
    async_ids = []
    for log in logs:
        total += 1
        try:
            # Some EXCEPTION logs are not stored on the database
            ids.append(log.pk)
        except AttributeError:
            pass
        if log.is_success:
            successes += 1
        elif not log.has_finished:
            run_async += 1
            async_ids.append(log.id)
    errors = total-successes-run_async
    url = get_backend_url(ids)
    async_url = get_backend_url(async_ids)
    async_msg = ''
    if run_async:
        async_msg = ungettext(
            _('<a href="{async_url}">{name}</a> is running on the background'),
            _('<a href="{async_url}">{run_async} backends</a> are running on the background'),
            run_async)
    if errors:
        if total == 1:
            msg = _('<a href="{url}">{name}</a> has fail to execute')
        else:
            msg = ungettext(
                _('<a href="{url}">{errors} out of {total} backends</a> has fail to execute'),
                _('<a href="{url}">{errors} out of {total} backends</a> have fail to execute'),
                errors)
        if async_msg:
            msg += ', ' + str(async_msg)
        msg = msg.format(errors=errors, run_async=run_async, async_url=async_url, total=total, url=url,
            name=log.backend)
        messages.append(('error', msg + '.'))
    elif successes:
        if async_msg:
            if total == 1:
                msg = _('<a href="{url}">{name}</a> has been executed')
            else:
                msg = ungettext(
                    _('<a href="{url}">{successes} out of {total} backends</a> has been executed'),
                    _('<a href="{url}">{successes} out of {total} backends</a> have been executed'),
                    successes)
            msg += ', ' + str(async_msg)
        else:
            msg = ungettext(
                _('<a href="{url}">{name}</a> has been executed'),
                _('<a href="{url}">{total} backends</a> have been executed'),
                total)
        msg = msg.format(
            total=total, url=url, async_url=async_url, run_async=run_async, successes=successes,
            name=log.backend
        )
        messages.append(('success', msg + '.'))
    else:
        msg = async_msg.format(url=url, async_url=async_url, run_async=run_async, name=log.backend)
        messages.append(('success', msg + '.'))
    return messages


def message_user(request, logs):
    for func, msg in get_messages(logs):
        getattr(messages, func)(request, mark_safe(msg))