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

from copy import deepcopy

from admin_tools.menu import items, Menu
from django.core.urlresolvers import reverse
from django.utils.text import capfirst
from django.utils.translation import ugettext_lazy as _

from orchestra.core import services, accounts, administration


def api_link(context):
    """ Dynamically generates API related URL """
    if 'opts' in context:
        opts = context['opts']
    elif 'cl' in context:
        opts = context['cl'].opts
    else:
        return reverse('api-root')
    if 'object_id' in context: 
        object_id = context['object_id']
        try:
            return reverse('%s-detail' % opts.model_name, args=[object_id])
        except:
            return reverse('api-root')
    try:
        return reverse('%s-list' % opts.model_name)
    except:
        return reverse('api-root')


def process_registry(register):
    def get_item(model, options, name=None):
        if name is None:
            name = capfirst(options.get('verbose_name_plural'))
        if isinstance(model, str):
            url = reverse('admin:'+model)
        else:
            opts = model._meta
            url = reverse('admin:{}_{}_changelist'.format(
                opts.app_label, opts.model_name)
            )
        item = items.MenuItem(name, url)
        item.options = options
        return item
    
    childrens = {}
    for model, options in register.get().items():
        if options.get('menu', True):
            parent = options.get('parent')
            if parent:
                name = capfirst(model._meta.app_label)
                parent_item = childrens.get(parent)
                if parent_item:
                    if not parent_item.children:
                        parent_item.children.append(deepcopy(parent_item))
                        parent_item.title = name
                else:
                    parent_item = get_item(parent, register[parent], name=name)
                    parent_item.children = []
                parent_item.children.append(get_item(model, options))
                childrens[parent] = parent_item
            elif model not in childrens:
                childrens[model] = get_item(model, options)
            else:
                childrens[model].children.insert(0, get_item(model, options))
    return sorted(childrens.values(), key=lambda i: i.title)


class OrchestraMenu(Menu):
    template = 'admin/orchestra/menu.html'
    
    def init_with_context(self, context):
        self.children = [
#            items.MenuItem(
#                mark_safe('{site_name} <span style="{version_style}">v{version}</span>'.format(
#                    site_name=force_text(settings.SITE_VERBOSE_NAME),
#                    version_style="text-transform:none; float:none; font-size:smaller; background:none;",
#                    version=get_version())),
#                reverse('admin:index')
#            ),
#            items.MenuItem(
#                _('Dashboard'),
#                reverse('admin:index')
#            ),
#            items.Bookmarks(),
            items.MenuItem(
                _("Services"),
                children=process_registry(services)
            ),
            items.MenuItem(
                _("Accounts"),
                reverse('admin:accounts_account_changelist'),
                children=process_registry(accounts)
            ),
            items.MenuItem(
                _("Administration"),
                children=process_registry(administration)
            ),
            items.MenuItem("API", api_link(context)),
        ]