django-orchestra-test/orchestra/admin/__init__.py

import itertools
from collections import OrderedDict
from functools import update_wrapper

from django.contrib import admin
from django.urls import reverse
from django.shortcuts import render, redirect
from django.utils.safestring import mark_safe
from django.utils.translation import gettext_lazy as _

from .dashboard import *
from .options import *
from ..core import accounts, services


# monkey-patch admin.site in order to porvide some extra admin urls

urls = []
def register_url(pattern, view, name=""):
    global urls
    urls.append((pattern, view, name))
admin.site.register_url = register_url


site_get_urls = admin.site.get_urls
def get_urls():
    def wrap(view, cacheable=False):
        def wrapper(*args, **kwargs):
            return admin.site.admin_view(view, cacheable)(*args, **kwargs)
        wrapper.admin_site = admin.site
        return update_wrapper(wrapper, view)
    global urls
    extra_patterns = []
    for pattern, view, name in urls:
        extra_patterns.append(
            url(pattern, wrap(view), name=name)
        )
    return site_get_urls() + extra_patterns
admin.site.get_urls = get_urls


def get_model(model_name, model_name_map):
    try:
        return model_name_map[model_name.lower()]
    except KeyError:
        return


def search(request):
    query = request.GET.get('q', '')
    search_term = query
    models = set()
    selected_models = set()
    model_name_map = {}
    for service in itertools.chain(services, accounts):
        if service.search:
            models.add(service.model)
            model_name_map[service.model._meta.model_name] = service.model

    # Account direct access
    if search_term.endswith('!'):
        from ..contrib.accounts.models import Account
        search_term = search_term.replace('!', '')
        try:
            account = Account.objects.get(username=search_term)
        except Account.DoesNotExist:
            pass
        else:
            account_url = reverse('admin:accounts_account_change', args=(account.pk,))
            return redirect(account_url)
    # Search for specific model
    elif ':' in search_term:
        new_search_term = []
        for part in search_term.split():
            if ':' in part:
                model_name, term = part.split(':')
                model = get_model(model_name, model_name_map)
                # Retry with singular version
                if model is None and model_name.endswith('s'):
                    model = get_model(model_name[:-1], model_name_map)
                if model is None:
                    new_search_term.append(':'.join((model_name, term)))
                else:
                    selected_models.add(model)
                    new_search_term.append(term)
            else:
                new_search_term.append(part)
        search_term = ' '.join(new_search_term)
    if selected_models:
        models = selected_models
    results = OrderedDict()
    models = sorted(models, key=lambda m: m._meta.verbose_name_plural.lower())
    total = 0
    for model in models:
        try:
            modeladmin = admin.site._registry[model]
        except KeyError:
            pass
        else:
            qs = modeladmin.get_queryset(request)
            qs, search_use_distinct = modeladmin.get_search_results(request, qs, search_term)
            if search_use_distinct:
                qs = qs.distinct()
            num = len(qs)
            if num:
                total += num
                results[model._meta] = qs
    title = _("{total} search results for '<tt>{query}</tt>'").format(total=total, query=query)
    context = {
        'title': mark_safe(title),
        'total': total,
        'columns': min(int(total/17), 3),
        'query': query,
        'search_term': search_term,
        'results': results,
        'search_autofocus': True,
    }
    return render(request, 'admin/orchestra/search.html', context)


admin.site.register_url(r'^search/$', search, 'orchestra_search_view')