plstackapi/core/admin.py

   1 from plstackapi.core.models import Site
   2 from plstackapi.core.models import *
   3 from django.contrib import admin
   4 from django import forms
   5
   6 from django.contrib.auth.admin import UserAdmin
   7 from django.contrib.admin.widgets import FilteredSelectMultiple
   8
   9
  10 class ReadonlyTabularInline(admin.TabularInline):
  11     can_delete = False
  12     extra = 0
  13     editable_fields = []
  14
  15     def get_readonly_fields(self, request, obj=None):
  16         fields = []
  17         for field in self.model._meta.get_all_field_names():
  18             if (not field == 'id'):
  19                 if (field not in self.editable_fields):
  20                     fields.append(field)
  21         return fields
  22
  23     def has_add_permission(self, request):
  24         return False
  25
  26 class SliverInline(admin.TabularInline):
  27     model = Sliver
  28     extra = 0
  29
  30 class SiteInline(admin.TabularInline):
  31     model = Site
  32     extra = 0
  33
  34 class NodeInline(admin.TabularInline):
  35     model = Node
  36     extra = 0
  37
  38 class PlanetStackBaseAdmin(admin.ModelAdmin):
  39     save_on_top = False
  40
  41 class DeploymentNetworkAdminForm(forms.ModelForm):
  42     sites = forms.ModelMultipleChoiceField(
  43         queryset=Site.objects.all(),
  44         required=False,
  45         widget=FilteredSelectMultiple(
  46             verbose_name=('Sites'), is_stacked=False
  47         )
  48     )
  49     class Meta:
  50         model = DeploymentNetwork
  51
  52     def __init__(self, *args, **kwargs):
  53         super(DeploymentNetworkAdminForm, self).__init__(*args, **kwargs)
  54
  55         if self.instance and self.instance.pk:
  56             self.fields['sites'].initial = self.instance.sites.all()
  57
  58     def save(self, commit=True):
  59         deploymentNetwork = super(DeploymentNetworkAdminForm, self).save(commit=False)
  60
  61         if commit:
  62             deploymentNetwork.save()
  63
  64         if deploymentNetwork.pk:
  65             deploymentNetwork.sites = self.cleaned_data['sites']
  66             self.save_m2m()
  67
  68         return deploymentNetwork
  69
  70 class DeploymentNetworkAdmin(PlanetStackBaseAdmin):
  71     form = DeploymentNetworkAdminForm
  72     inlines = [NodeInline,]
  73
  74 class SiteAdmin(admin.ModelAdmin):
  75     fieldsets = [
  76         (None, {'fields': ['name', 'site_url', 'enabled', 'is_public', 'login_base']}),
  77         ('Location', {'fields': ['latitude', 'longitude']}),
  78         ('Deployment Networks', {'fields': ['deployments']})
  79     ]
  80     list_display = ('name', 'login_base','site_url', 'enabled')
  81     filter_horizontal = ('deployments',)
  82     inlines = [NodeInline,]
  83     search_fields = ['name']
  84
  85 class SliceAdmin(PlanetStackBaseAdmin):
  86     fields = ['name', 'site', 'instantiation', 'description', 'slice_url']
  87     list_display = ('name', 'site','slice_url', 'instantiation')
  88     inlines = [SliverInline]
  89
  90 class SubnetAdmin(admin.ModelAdmin):
  91     fields = ['cidr', 'ip_version', 'start', 'end', 'slice']
  92
  93 class ImageAdmin(admin.ModelAdmin):
  94     fields = ['image_id', 'name', 'disk_format', 'container_format']
  95
  96 class NodeAdmin(admin.ModelAdmin):
  97     list_display = ('name', 'site', 'deploymentNetwork')
  98     list_filter = ('deploymentNetwork',)
  99
 100
 101
 102 class IgnoredField(forms.Field):
 103     def validate(self, value):
 104         return
 105
 106
 107
 108 class RoleForm(forms.ModelForm):
 109     role_id = IgnoredField()
 110     role_type = forms.CharField()
 111
 112
 113 class RoleAdmin(admin.ModelAdmin):
 114     form = RoleForm
 115     list_display = ('role_type',)
 116
 117 admin.site.register(Site, SiteAdmin)
 118 admin.site.register(SitePrivilege)
 119 admin.site.register(Slice, SliceAdmin)
 120 admin.site.register(SliceMembership)
 121 admin.site.register(Subnet, SubnetAdmin)
 122 admin.site.register(Image, ImageAdmin)
 123 admin.site.register(Node, NodeAdmin)
 124 admin.site.register(Sliver)
 125 admin.site.register(Flavor)
 126 admin.site.register(Key)
 127 admin.site.register(Role, RoleAdmin)
 128 admin.site.register(User)
 129 admin.site.register(DeploymentNetwork, DeploymentNetworkAdmin)
 130