django-orchestra/orchestra/admin/menu.py

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
from orchestra.utils.apps import isinstalled


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 get_services():
    result = []
    for model, options in services.get().iteritems():
        if options.get('menu', True):
            opts = model._meta
            url = reverse('admin:%s_%s_changelist' % (opts.app_label, opts.model_name))
            name = capfirst(options.get('verbose_name_plural'))
            result.append(items.MenuItem(name, url))
    return sorted(result, key=lambda i: i.title)


def get_accounts():
    accounts = [
        items.MenuItem(_("Accounts"), reverse('admin:accounts_account_changelist'))
    ]
    if isinstalled('orchestra.apps.contacts'):
        url = reverse('admin:contacts_contact_changelist')
        accounts.append(items.MenuItem(_("Contacts"), url))
    if isinstalled('orchestra.apps.users'):
        url = reverse('admin:users_user_changelist')
        users = [items.MenuItem(_("Users"), url)]
        if isinstalled('rest_framework.authtoken'):
            tokens = reverse('admin:authtoken_token_changelist')
            users.append(items.MenuItem(_("Tokens"), tokens))
        accounts.append(items.MenuItem(_("Users"), url, children=users))
    if isinstalled('orchestra.apps.prices'):
        url = reverse('admin:prices_pack_changelist')
        accounts.append(items.MenuItem(_("Packs"), url))
    if isinstalled('orchestra.apps.orders'):
        url = reverse('admin:orders_order_changelist')
        accounts.append(items.MenuItem(_("Orders"), url))
    return accounts


def get_administration():
    administration = []
    return administration


def get_administration_models():
    administration_models = []
    if isinstalled('orchestra.apps.orchestration'):
        administration_models.append('orchestra.apps.orchestration.*')
    if isinstalled('djcelery'):
        administration_models.append('djcelery.*')
    if isinstalled('orchestra.apps.issues'):
        administration_models.append('orchestra.apps.issues.*')
    if isinstalled('orchestra.apps.resources'):
        administration_models.append('orchestra.apps.resources.*')
    if isinstalled('orchestra.apps.miscellaneous'):
        administration_models.append('orchestra.apps.miscellaneous.models.MiscService')
    if isinstalled('orchestra.apps.orders'):
        administration_models.append('orchestra.apps.orders.models.Service')
    return administration_models


class OrchestraMenu(Menu):
    def init_with_context(self, context):
        self.children += [
            items.MenuItem(
                _('Dashboard'),
                reverse('admin:index')
            ),
            items.Bookmarks(),
            items.MenuItem(
                _("Services"),
                reverse('admin:index'),
                children=get_services()
            ),
            items.MenuItem(
                _("Accounts"),
                reverse('admin:accounts_account_changelist'),
                children=get_accounts()
            ),
            items.AppList(
                _("Administration"),
                models=get_administration_models(),
                children=get_administration()
            ),
            items.MenuItem("API", api_link(context))
        ]