138 lines
5.4 KiB
Python
138 lines
5.4 KiB
Python
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
|
|
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 process_registered_models(register):
|
|
childrens = []
|
|
for model, options in register.get().items():
|
|
if options.get('menu', True):
|
|
opts = model._meta
|
|
url = reverse('admin:{}_{}_changelist'.format(
|
|
opts.app_label, opts.model_name))
|
|
name = capfirst(options.get('verbose_name_plural'))
|
|
childrens.append(items.MenuItem(name, url))
|
|
return childrens
|
|
|
|
|
|
def get_services():
|
|
childrens = process_registered_models(services)
|
|
return sorted(childrens, key=lambda i: i.title)
|
|
|
|
|
|
def get_accounts():
|
|
childrens=[]
|
|
if isinstalled('orchestra.contrib.payments'):
|
|
url = reverse('admin:payments_transactionprocess_changelist')
|
|
childrens.append(items.MenuItem(_("Transaction processes"), url))
|
|
if isinstalled('orchestra.contrib.issues'):
|
|
url = reverse('admin:issues_ticket_changelist')
|
|
childrens.append(items.MenuItem(_("Tickets"), url))
|
|
childrens.extend(process_registered_models(accounts))
|
|
return sorted(childrens, key=lambda i: i.title)
|
|
|
|
|
|
def get_administration_items():
|
|
childrens = []
|
|
if isinstalled('orchestra.contrib.settings'):
|
|
url = reverse('admin:settings_setting_change')
|
|
childrens.append(items.MenuItem(_("Settings"), url))
|
|
if isinstalled('orchestra.contrib.services'):
|
|
url = reverse('admin:services_service_changelist')
|
|
childrens.append(items.MenuItem(_("Services"), url))
|
|
url = reverse('admin:plans_plan_changelist')
|
|
childrens.append(items.MenuItem(_("Plans"), url))
|
|
if isinstalled('orchestra.contrib.orchestration'):
|
|
route = reverse('admin:orchestration_route_changelist')
|
|
backendlog = reverse('admin:orchestration_backendlog_changelist')
|
|
server = reverse('admin:orchestration_server_changelist')
|
|
childrens.append(items.MenuItem(_("Orchestration"), route, children=[
|
|
items.MenuItem(_("Routes"), route),
|
|
items.MenuItem(_("Backend logs"), backendlog),
|
|
items.MenuItem(_("Servers"), server),
|
|
]))
|
|
if isinstalled('orchestra.contrib.resources'):
|
|
resource = reverse('admin:resources_resource_changelist')
|
|
data = reverse('admin:resources_resourcedata_changelist')
|
|
monitor = reverse('admin:resources_monitordata_changelist')
|
|
childrens.append(items.MenuItem(_("Resources"), resource, children=[
|
|
items.MenuItem(_("Resources"), resource),
|
|
items.MenuItem(_("Data"), data),
|
|
items.MenuItem(_("Monitoring"), monitor),
|
|
]))
|
|
if isinstalled('orchestra.contrib.miscellaneous'):
|
|
url = reverse('admin:miscellaneous_miscservice_changelist')
|
|
childrens.append(items.MenuItem(_("Miscellaneous"), url))
|
|
if isinstalled('orchestra.contrib.issues'):
|
|
url = reverse('admin:issues_queue_changelist')
|
|
childrens.append(items.MenuItem(_("Ticket queues"), url))
|
|
if isinstalled('djcelery'):
|
|
task = reverse('admin:djcelery_taskstate_changelist')
|
|
periodic = reverse('admin:djcelery_periodictask_changelist')
|
|
worker = reverse('admin:djcelery_workerstate_changelist')
|
|
childrens.append(items.MenuItem(_("Tasks"), task, children=[
|
|
items.MenuItem(_("Logs"), task),
|
|
items.MenuItem(_("Periodic tasks"), periodic),
|
|
items.MenuItem(_("Workers"), worker),
|
|
]))
|
|
childrens.extend(process_registered_models(administration))
|
|
return childrens
|
|
|
|
|
|
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=get_services()
|
|
),
|
|
items.MenuItem(
|
|
_("Accounts"),
|
|
reverse('admin:accounts_account_changelist'),
|
|
children=get_accounts()
|
|
),
|
|
items.MenuItem(
|
|
_("Administration"),
|
|
children=get_administration_items()
|
|
),
|
|
items.MenuItem("API", api_link(context)),
|
|
]
|