126 lines
4 KiB
Python
126 lines
4 KiB
Python
|
from django.contrib import admin
|
||
|
|
from django.core.urlresolvers import reverse
|
||
|
|
from django.utils.html import escape
|
||
|
|
from django.utils.translation import ugettext_lazy as _
|
||
|
|
|
||
|
|
from orchestra.admin.html import monospace_format
|
||
|
|
from orchestra.admin.utils import link
|
||
|
|
|
||
|
|
from .models import Server, Route, BackendLog, BackendOperation
|
||
|
|
|
||
|
|
|
||
|
|
STATE_COLORS = {
|
||
|
|
BackendLog.RECEIVED: 'darkorange',
|
||
|
|
BackendLog.TIMEOUT: 'red',
|
||
|
|
BackendLog.STARTED: 'blue',
|
||
|
|
BackendLog.SUCCESS: 'green',
|
||
|
|
BackendLog.FAILURE: 'red',
|
||
|
|
BackendLog.ERROR: 'red',
|
||
|
|
BackendLog.REVOKED: 'magenta',
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
class RouteAdmin(admin.ModelAdmin):
|
||
|
|
list_display = [
|
||
|
|
'id', 'backend', 'host', 'match', 'display_model', 'is_active'
|
||
|
|
]
|
||
|
|
list_editable = ['backend', 'host', 'match', 'is_active']
|
||
|
|
list_filter = ['backend', 'host', 'is_active']
|
||
|
|
|
||
|
|
def display_model(self, route):
|
||
|
|
try:
|
||
|
|
return route.get_backend().model
|
||
|
|
except KeyError:
|
||
|
|
return "<span style='color: red;'>NOT AVAILABLE</span>"
|
||
|
|
display_model.short_description = _("model")
|
||
|
|
display_model.allow_tags = True
|
||
|
|
|
||
|
|
|
||
|
|
class BackendOperationInline(admin.TabularInline):
|
||
|
|
model = BackendOperation
|
||
|
|
fields = ('action', 'instance_link')
|
||
|
|
readonly_fields = ('action', 'instance_link')
|
||
|
|
extra = 0
|
||
|
|
can_delete = False
|
||
|
|
|
||
|
|
class Media:
|
||
|
|
css = {
|
||
|
|
'all': ('orchestra/css/hide-inline-id.css',)
|
||
|
|
}
|
||
|
|
|
||
|
|
def instance_link(self, operation):
|
||
|
|
try:
|
||
|
|
return link('instance')(self, operation)
|
||
|
|
except:
|
||
|
|
return _("deleted %s %s") % (operation.content_type, operation.object_id)
|
||
|
|
instance_link.allow_tags = True
|
||
|
|
instance_link.short_description = _("Instance")
|
||
|
|
|
||
|
|
def has_add_permission(self, *args, **kwargs):
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
class BackendLogAdmin(admin.ModelAdmin):
|
||
|
|
list_display = (
|
||
|
|
'id', 'backend', 'server_link', 'display_state', 'exit_code', 'created',
|
||
|
|
'execution_time',
|
||
|
|
)
|
||
|
|
list_display_links = ('id', 'backend')
|
||
|
|
list_filter = ('state', 'backend')
|
||
|
|
date_hierarchy = 'last_update'
|
||
|
|
inlines = [BackendOperationInline]
|
||
|
|
fields = [
|
||
|
|
'backend', 'server', 'state', 'mono_script', 'mono_stdout', 'mono_stderr',
|
||
|
|
'mono_traceback', 'exit_code', 'task_id', 'created', 'last_update',
|
||
|
|
'execution_time'
|
||
|
|
]
|
||
|
|
readonly_fields = [
|
||
|
|
'backend', 'server', 'state', 'mono_script', 'mono_stdout', 'mono_stderr',
|
||
|
|
'mono_traceback', 'exit_code', 'task_id', 'created', 'last_update',
|
||
|
|
'execution_time'
|
||
|
|
]
|
||
|
|
|
||
|
|
def server_link(self, log):
|
||
|
|
url = reverse('admin:orchestration_server_change', args=(log.server.pk,))
|
||
|
|
return '<a href="%s">%s</a>' % (url, log.server.name)
|
||
|
|
server_link.short_description = _("server")
|
||
|
|
server_link.allow_tags = True
|
||
|
|
|
||
|
|
def display_state(self, log):
|
||
|
|
color = STATE_COLORS.get(log.state, 'grey')
|
||
|
|
return '<span style="color: %s;">%s</span>' % (color, log.state)
|
||
|
|
display_state.short_description = _("state")
|
||
|
|
display_state.allow_tags = True
|
||
|
|
display_state.admin_order_field = 'state'
|
||
|
|
|
||
|
|
def mono_script(self, log):
|
||
|
|
return monospace_format(escape(log.script))
|
||
|
|
mono_script.short_description = _("script")
|
||
|
|
|
||
|
|
def mono_stdout(self, log):
|
||
|
|
return monospace_format(escape(log.stdout))
|
||
|
|
mono_stdout.short_description = _("stdout")
|
||
|
|
|
||
|
|
def mono_stderr(self, log):
|
||
|
|
return monospace_format(escape(log.stderr))
|
||
|
|
mono_stderr.short_description = _("stderr")
|
||
|
|
|
||
|
|
def mono_traceback(self, log):
|
||
|
|
return monospace_format(escape(log.traceback))
|
||
|
|
mono_traceback.short_description = _("traceback")
|
||
|
|
|
||
|
|
def queryset(self, request):
|
||
|
|
""" Order by structured name and imporve performance """
|
||
|
|
qs = super(BackendLogAdmin, self).queryset(request)
|
||
|
|
return qs.select_related('server')
|
||
|
|
|
||
|
|
|
||
|
|
class ServerAdmin(admin.ModelAdmin):
|
||
|
|
list_display = ('name', 'address', 'os')
|
||
|
|
list_filter = ('os',)
|
||
|
|
|
||
|
|
|
||
|
|
admin.site.register(Server, ServerAdmin)
|
||
|
|
admin.site.register(BackendLog, BackendLogAdmin)
|
||
|
|
admin.site.register(Route, RouteAdmin)
|