django-orchestra/orchestra/admin/menu.py

111 lines
3.9 KiB
Python
Raw Normal View History

2014-05-08 16:59:35 +00:00
from admin_tools.menu import items, Menu
from django.core.urlresolvers import reverse
2014-07-08 15:19:15 +00:00
from django.utils.text import capfirst
2014-05-08 16:59:35 +00:00
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:
2014-07-08 15:19:15 +00:00
return reverse('%s-detail' % opts.model_name, args=[object_id])
2014-05-08 16:59:35 +00:00
except:
return reverse('api-root')
try:
2014-07-08 15:19:15 +00:00
return reverse('%s-list' % opts.model_name)
2014-05-08 16:59:35 +00:00
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))
2014-07-08 15:19:15 +00:00
name = capfirst(options.get('verbose_name_plural'))
result.append(items.MenuItem(name, url))
2014-05-08 16:59:35 +00:00
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))
2014-05-27 15:55:09 +00:00
if isinstalled('orchestra.apps.prices'):
2014-07-16 15:20:16 +00:00
url = reverse('admin:prices_pack_changelist')
accounts.append(items.MenuItem(_("Packs"), url))
2014-05-27 15:55:09 +00:00
if isinstalled('orchestra.apps.orders'):
url = reverse('admin:orders_order_changelist')
accounts.append(items.MenuItem(_("Orders"), url))
2014-05-08 16:59:35 +00:00
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.*')
2014-07-08 15:19:15 +00:00
if isinstalled('orchestra.apps.resources'):
administration_models.append('orchestra.apps.resources.*')
2014-07-16 15:20:16 +00:00
if isinstalled('orchestra.apps.miscellaneous'):
administration_models.append('orchestra.apps.miscellaneous.models.MiscService')
2014-07-23 16:24:56 +00:00
if isinstalled('orchestra.apps.orders'):
administration_models.append('orchestra.apps.orders.models.Service')
2014-05-08 16:59:35 +00:00
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))
]