93 lines
3.0 KiB
Python
93 lines
3.0 KiB
Python
|
import sys
|
||
|
|
|
||
|
|
from django.contrib import admin
|
||
|
|
from django.contrib.contenttypes import generic
|
||
|
|
from django.utils.functional import cached_property
|
||
|
|
from django.utils.translation import ugettext_lazy as _
|
||
|
|
|
||
|
|
from orchestra.admin.filters import UsedContentTypeFilter
|
||
|
|
from orchestra.admin.utils import insertattr, get_modeladmin
|
||
|
|
|
||
|
|
from .forms import ResourceForm
|
||
|
|
from .models import Resource, ResourceAllocation, Monitor, MonitorData
|
||
|
|
|
||
|
|
|
||
|
|
class ResourceAdmin(admin.ModelAdmin):
|
||
|
|
list_display = (
|
||
|
|
'name', 'verbose_name', 'content_type', 'period', 'ondemand',
|
||
|
|
'default_allocation', 'disable_trigger'
|
||
|
|
)
|
||
|
|
list_filter = (UsedContentTypeFilter, 'period', 'ondemand', 'disable_trigger')
|
||
|
|
|
||
|
|
def save_model(self, request, obj, form, change):
|
||
|
|
super(ResourceAdmin, self).save_model(request, obj, form, change)
|
||
|
|
model = obj.content_type.model_class()
|
||
|
|
modeladmin = get_modeladmin(model)
|
||
|
|
resources = obj.content_type.resource_set.filter(is_active=True)
|
||
|
|
inlines = []
|
||
|
|
for inline in modeladmin.inlines:
|
||
|
|
if inline.model is ResourceAllocation:
|
||
|
|
inline = resource_inline_factory(resources)
|
||
|
|
inlines.append(inline)
|
||
|
|
modeladmin.inlines = inlines
|
||
|
|
|
||
|
|
|
||
|
|
class ResourceAllocationAdmin(admin.ModelAdmin):
|
||
|
|
list_display = ('id', 'resource', 'content_object', 'value')
|
||
|
|
list_filter = ('resource',)
|
||
|
|
|
||
|
|
|
||
|
|
class MonitorAdmin(admin.ModelAdmin):
|
||
|
|
list_display = ('backend', 'resource', 'crontab')
|
||
|
|
list_filter = ('backend', 'resource')
|
||
|
|
|
||
|
|
|
||
|
|
class MonitorDataAdmin(admin.ModelAdmin):
|
||
|
|
list_display = ('id', 'monitor', 'content_object', 'date', 'value')
|
||
|
|
list_filter = ('monitor',)
|
||
|
|
|
||
|
|
|
||
|
|
admin.site.register(Resource, ResourceAdmin)
|
||
|
|
admin.site.register(ResourceAllocation, ResourceAllocationAdmin)
|
||
|
|
admin.site.register(Monitor, MonitorAdmin)
|
||
|
|
admin.site.register(MonitorData, MonitorDataAdmin)
|
||
|
|
|
||
|
|
|
||
|
|
# Mokey-patching
|
||
|
|
|
||
|
|
def resource_inline_factory(resources):
|
||
|
|
class ResourceInlineFormSet(generic.BaseGenericInlineFormSet):
|
||
|
|
def total_form_count(self):
|
||
|
|
return len(resources)
|
||
|
|
|
||
|
|
@cached_property
|
||
|
|
def forms(self):
|
||
|
|
forms = []
|
||
|
|
for i, resource in enumerate(resources):
|
||
|
|
forms.append(self._construct_form(i, resource=resource))
|
||
|
|
return forms
|
||
|
|
|
||
|
|
class ResourceInline(generic.GenericTabularInline):
|
||
|
|
model = ResourceAllocation
|
||
|
|
verbose_name_plural = _("resources")
|
||
|
|
form = ResourceForm
|
||
|
|
formset = ResourceInlineFormSet
|
||
|
|
|
||
|
|
class Media:
|
||
|
|
css = {
|
||
|
|
'all': ('orchestra/css/hide-inline-id.css',)
|
||
|
|
}
|
||
|
|
|
||
|
|
def has_add_permission(self, *args, **kwargs):
|
||
|
|
""" Hidde add another """
|
||
|
|
return False
|
||
|
|
|
||
|
|
return ResourceInline
|
||
|
|
|
||
|
|
if not 'migrate' in sys.argv and not 'syncdb' in sys.argv:
|
||
|
|
# not run during syncdb
|
||
|
|
for resources in Resource.group_by_content_type():
|
||
|
|
inline = resource_inline_factory(resources)
|
||
|
|
model = resources[0].content_type.model_class()
|
||
|
|
insertattr(model, 'inlines', inline)
|