blob: bc845540f1116480e29963555e21a59df4437092 [file] [log] [blame]
Siobhan Tully30fd4292013-05-10 08:59:56 -04001from core.models import Site
2from core.models import *
3from openstack.manager import OpenStackManager
Tony Macke59a7c82013-04-27 11:08:10 -04004
Tony Mack7130ac32013-03-22 21:58:00 -04005from django.contrib import admin
Siobhan Tully53437282013-04-26 19:30:27 -04006from django.contrib.auth.models import Group
Siobhan Tully4bc09f22013-04-10 21:15:21 -04007from django import forms
Tony Mackd90cdbf2013-04-16 22:48:40 -04008from django.utils.safestring import mark_safe
Tony Mack7130ac32013-03-22 21:58:00 -04009from django.contrib.auth.admin import UserAdmin
Scott Baker9f6b8ed2014-11-17 23:44:03 -080010from django.contrib.admin.widgets import FilteredSelectMultiple, AdminTextareaWidget
Scott Bakercbfb6002014-10-03 00:32:37 -070011from django.contrib.auth.forms import ReadOnlyPasswordHashField, AdminPasswordChangeForm
Scott Bakeracd45142013-05-19 16:19:16 -070012from django.contrib.auth.signals import user_logged_in
13from django.utils import timezone
Siobhan Tullyde5450d2013-06-21 11:35:33 -040014from django.contrib.contenttypes import generic
Siobhan Tullybfd11dc2013-09-03 12:59:24 -040015from suit.widgets import LinkedSelect
Siobhan Tullycf04fb62014-01-11 11:25:57 -050016from django.core.exceptions import PermissionDenied
Scott Bakere2bbf7e2014-01-13 12:09:31 -080017from django.core.urlresolvers import reverse, NoReverseMatch
Scott Baker9f6b8ed2014-11-17 23:44:03 -080018from django.utils.encoding import force_text, python_2_unicode_compatible
19from django.utils.html import conditional_escape, format_html
20from django.forms.utils import flatatt, to_current_timezone
Scott Baker3cde7372014-10-21 21:03:08 -070021from cgi import escape as html_escape
Tony Mack7130ac32013-03-22 21:58:00 -040022
Scott Baker36f50872014-08-21 13:01:25 -070023import django_evolution
Scott Baker0a5633b2014-10-06 17:51:20 -070024import threading
25
26# thread locals necessary to work around a django-suit issue
27_thread_locals = threading.local()
Scott Baker36f50872014-08-21 13:01:25 -070028
Scott Baker40c00762014-08-21 16:55:59 -070029def backend_icon(obj): # backend_status, enacted, updated):
30 #return "%s %s %s" % (str(obj.updated), str(obj.enacted), str(obj.backend_status))
31 if (obj.enacted is not None) and obj.enacted >= obj.updated:
Scott Bakerfbf06642014-09-09 10:38:15 -070032 return '<span style="min-width:16px;"><img src="/static/admin/img/icon_success.gif"></span>'
Scott Baker40c00762014-08-21 16:55:59 -070033 else:
34 if obj.backend_status == "Provisioning in progress" or obj.backend_status=="":
Scott Bakerfbf06642014-09-09 10:38:15 -070035 return '<span style="min-width:16px;" title="%s"><img src="/static/admin/img/icon_clock.gif"></span>' % obj.backend_status
Scott Baker63d1a552014-08-21 15:19:07 -070036 else:
Scott Baker3cde7372014-10-21 21:03:08 -070037 return '<span style="min-width:16px;" title="%s"><img src="/static/admin/img/icon_error.gif"></span>' % html_escape(obj.backend_status, quote=True)
Scott Baker40c00762014-08-21 16:55:59 -070038
39def backend_text(obj):
40 icon = backend_icon(obj)
41 if (obj.enacted is not None) and obj.enacted >= obj.updated:
Scott Baker3cde7372014-10-21 21:03:08 -070042 return "%s %s" % (icon, "successfully enacted")
Scott Baker40c00762014-08-21 16:55:59 -070043 else:
Scott Baker3cde7372014-10-21 21:03:08 -070044 return "%s %s" % (icon, html_escape(obj.backend_status, quote=True))
Scott Baker63d1a552014-08-21 15:19:07 -070045
Scott Baker9f6b8ed2014-11-17 23:44:03 -080046class UploadTextareaWidget(AdminTextareaWidget):
47 def render(self, name, value, attrs=None):
48 if value is None:
49 value = ''
50 final_attrs = self.build_attrs(attrs, name=name)
51 return format_html('<input type="file" style="width: 0; height: 0" id="btn_upload_%s" onChange="uploadTextarea(event,\'%s\');">' \
52 '<button onClick="$(\'#btn_upload_%s\').click(); return false;">Upload</button>' \
53 '<br><textarea{0}>\r\n{1}</textarea>' % (attrs["id"], attrs["id"], attrs["id"]),
54 flatatt(final_attrs),
55 force_text(value))
56
Scott Baker36f50872014-08-21 13:01:25 -070057class PlainTextWidget(forms.HiddenInput):
58 input_type = 'hidden'
59
60 def render(self, name, value, attrs=None):
61 if value is None:
62 value = ''
63 return mark_safe(str(value) + super(PlainTextWidget, self).render(name, value, attrs))
64
Scott Baker86c83ab2014-10-03 13:10:47 -070065class PermissionCheckingAdminMixin(object):
Scott Bakercbfb6002014-10-03 00:32:37 -070066 # call save_by_user and delete_by_user instead of save and delete
Siobhan Tullycf04fb62014-01-11 11:25:57 -050067
68 def has_add_permission(self, request, obj=None):
69 return (not self.__user_is_readonly(request))
Scott Baker36f50872014-08-21 13:01:25 -070070
Siobhan Tullycf04fb62014-01-11 11:25:57 -050071 def has_delete_permission(self, request, obj=None):
72 return (not self.__user_is_readonly(request))
73
74 def save_model(self, request, obj, form, change):
75 if self.__user_is_readonly(request):
Scott Bakercbfb6002014-10-03 00:32:37 -070076 # this 'if' might be redundant if save_by_user is implemented right
Siobhan Tullycf04fb62014-01-11 11:25:57 -050077 raise PermissionDenied
Scott Bakercbfb6002014-10-03 00:32:37 -070078
79 obj.caller = request.user
80 # update openstack connection to use this site/tenant
81 obj.save_by_user(request.user)
82
83 def delete_model(self, request, obj):
84 obj.delete_by_user(request.user)
85
86 def save_formset(self, request, form, formset, change):
87 instances = formset.save(commit=False)
88 for instance in instances:
89 instance.save_by_user(request.user)
90
91 # BUG in django 1.7? Objects are not deleted by formset.save if
92 # commit is False. So let's delete them ourselves.
93 #
94 # code from forms/models.py save_existing_objects()
95 try:
96 forms_to_delete = formset.deleted_forms
97 except AttributeError:
98 forms_to_delete = []
99 if formset.initial_forms:
100 for form in formset.initial_forms:
101 obj = form.instance
102 if form in forms_to_delete:
103 if obj.pk is None:
104 continue
105 formset.deleted_objects.append(obj)
106 obj.delete()
107
108 formset.save_m2m()
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500109
110 def get_actions(self,request):
Scott Baker86c83ab2014-10-03 13:10:47 -0700111 actions = super(PermissionCheckingAdminMixin,self).get_actions(request)
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500112
113 if self.__user_is_readonly(request):
114 if 'delete_selected' in actions:
115 del actions['delete_selected']
116
117 return actions
118
119 def change_view(self,request,object_id, extra_context=None):
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500120 if self.__user_is_readonly(request):
Scott Bakeraf73e102014-04-22 22:40:07 -0700121 if not hasattr(self, "readonly_save"):
122 # save the original readonly fields
123 self.readonly_save = self.readonly_fields
124 self.inlines_save = self.inlines
Scott Bakere8859f92014-05-23 12:42:40 -0700125 if hasattr(self, "user_readonly_fields"):
126 self.readonly_fields=self.user_readonly_fields
127 if hasattr(self, "user_readonly_inlines"):
128 self.inlines = self.user_readonly_inlines
Scott Bakeraf73e102014-04-22 22:40:07 -0700129 else:
130 if hasattr(self, "readonly_save"):
131 # restore the original readonly fields
132 self.readonly_fields = self.readonly_save
Scott Bakere8859f92014-05-23 12:42:40 -0700133 if hasattr(self, "inlines_save"):
Scott Bakeraf73e102014-04-22 22:40:07 -0700134 self.inlines = self.inlines_save
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500135
136 try:
Scott Baker86c83ab2014-10-03 13:10:47 -0700137 return super(PermissionCheckingAdminMixin, self).change_view(request, object_id, extra_context=extra_context)
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500138 except PermissionDenied:
139 pass
140 if request.method == 'POST':
141 raise PermissionDenied
142 request.readonly = True
Scott Baker86c83ab2014-10-03 13:10:47 -0700143 return super(PermissionCheckingAdminMixin, self).change_view(request, object_id, extra_context=extra_context)
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500144
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500145 def __user_is_readonly(self, request):
146 return request.user.isReadOnlyUser()
147
Scott Baker40c00762014-08-21 16:55:59 -0700148 def backend_status_text(self, obj):
149 return mark_safe(backend_text(obj))
Scott Baker36f50872014-08-21 13:01:25 -0700150
Scott Baker63d1a552014-08-21 15:19:07 -0700151 def backend_status_icon(self, obj):
Scott Baker40c00762014-08-21 16:55:59 -0700152 return mark_safe(backend_icon(obj))
Scott Baker63d1a552014-08-21 15:19:07 -0700153 backend_status_icon.short_description = ""
154
Scott Baker24ded6a2014-11-05 09:05:38 -0800155 def get_form(self, request, obj=None, **kwargs):
Scott Baker5c432692014-10-16 00:57:55 -0700156 # Save obj and request in thread-local storage, so suit_form_tabs can
157 # use it to determine whether we're in edit or add mode, and can
158 # determine whether the user is an admin.
159 _thread_locals.request = request
160 _thread_locals.obj = obj
Scott Baker24ded6a2014-11-05 09:05:38 -0800161 return super(PermissionCheckingAdminMixin, self).get_form(request, obj, **kwargs)
Scott Baker5c432692014-10-16 00:57:55 -0700162
163 def get_inline_instances(self, request, obj=None):
164 inlines = super(PermissionCheckingAdminMixin, self).get_inline_instances(request, obj)
165
166 # inlines that should only be shown to an admin user
167 if request.user.is_admin:
168 for inline_class in getattr(self, "admin_inlines", []):
169 inlines.append(inline_class(self.model, self.admin_site))
170
171 return inlines
172
Scott Baker86c83ab2014-10-03 13:10:47 -0700173class ReadOnlyAwareAdmin(PermissionCheckingAdminMixin, admin.ModelAdmin):
174 # Note: Make sure PermissionCheckingAdminMixin is listed before
175 # admin.ModelAdmin in the class declaration.
176
Scott Bakercbfb6002014-10-03 00:32:37 -0700177 pass
178
179class PlanetStackBaseAdmin(ReadOnlyAwareAdmin):
180 save_on_top = False
Scott Baker36f50872014-08-21 13:01:25 -0700181
Scott Bakere8859f92014-05-23 12:42:40 -0700182class SingletonAdmin (ReadOnlyAwareAdmin):
Siobhan Tullyce652d02013-10-08 21:52:35 -0400183 def has_add_permission(self, request):
Scott Bakere8859f92014-05-23 12:42:40 -0700184 if not super(SingletonAdmin, self).has_add_permission(request):
185 return False
186
Siobhan Tullyce652d02013-10-08 21:52:35 -0400187 num_objects = self.model.objects.count()
188 if num_objects >= 1:
189 return False
190 else:
191 return True
192
Siobhan Tullyd3515752013-06-21 16:34:53 -0400193class PlStackTabularInline(admin.TabularInline):
Scott Baker86568322014-01-12 16:53:31 -0800194 def __init__(self, *args, **kwargs):
195 super(PlStackTabularInline, self).__init__(*args, **kwargs)
196
197 # InlineModelAdmin as no get_fields() method, so in order to add
198 # the selflink field, we override __init__ to modify self.fields and
199 # self.readonly_fields.
200
Scott Bakere2bbf7e2014-01-13 12:09:31 -0800201 self.setup_selflink()
202
Scott Baker874936e2014-01-13 18:15:34 -0800203 def get_change_url(self, model, id):
204 """ Get the URL to a change form in the admin for this model """
205 reverse_path = "admin:%s_change" % (model._meta.db_table)
Scott Bakere2bbf7e2014-01-13 12:09:31 -0800206 try:
Scott Baker874936e2014-01-13 18:15:34 -0800207 url = reverse(reverse_path, args=(id,))
Scott Bakere2bbf7e2014-01-13 12:09:31 -0800208 except NoReverseMatch:
Scott Baker874936e2014-01-13 18:15:34 -0800209 return None
210
211 return url
212
213 def setup_selflink(self):
214 if hasattr(self, "selflink_fieldname"):
215 """ self.selflink_model can be defined to punch through a relation
216 to its target object. For example, in SliceNetworkInline, set
217 selflink_model = "network", and the URL will lead to the Network
218 object instead of trying to bring up a change view of the
219 SliceNetwork object.
220 """
221 self.selflink_model = getattr(self.model,self.selflink_fieldname).field.rel.to
222 else:
223 self.selflink_model = self.model
224
225 url = self.get_change_url(self.selflink_model, 0)
226
227 # We don't have an admin for this object, so don't create the
228 # selflink.
229 if (url == None):
Scott Bakere2bbf7e2014-01-13 12:09:31 -0800230 return
231
Scott Baker874936e2014-01-13 18:15:34 -0800232 # Since we need to add "selflink" to the field list, we need to create
233 # self.fields if it is None.
Scott Baker0165fac2014-01-13 11:49:26 -0800234 if (self.fields is None):
235 self.fields = []
236 for f in self.model._meta.fields:
237 if f.editable and f.name != "id":
238 self.fields.append(f.name)
Scott Baker86568322014-01-12 16:53:31 -0800239
Scott Baker874936e2014-01-13 18:15:34 -0800240 self.fields = tuple(self.fields) + ("selflink", )
Scott Baker86568322014-01-12 16:53:31 -0800241
Scott Baker874936e2014-01-13 18:15:34 -0800242 if self.readonly_fields is None:
243 self.readonly_fields = ()
Scott Baker86568322014-01-12 16:53:31 -0800244
Scott Baker874936e2014-01-13 18:15:34 -0800245 self.readonly_fields = tuple(self.readonly_fields) + ("selflink", )
Scott Baker86568322014-01-12 16:53:31 -0800246
247 def selflink(self, obj):
Scott Baker874936e2014-01-13 18:15:34 -0800248 if hasattr(self, "selflink_fieldname"):
249 obj = getattr(obj, self.selflink_fieldname)
250
Scott Baker86568322014-01-12 16:53:31 -0800251 if obj.id:
Scott Baker874936e2014-01-13 18:15:34 -0800252 url = self.get_change_url(self.selflink_model, obj.id)
253 return "<a href='%s'>Details</a>" % str(url)
Scott Baker86568322014-01-12 16:53:31 -0800254 else:
Scott Bakere2bbf7e2014-01-13 12:09:31 -0800255 return "Not present"
Scott Baker86568322014-01-12 16:53:31 -0800256
257 selflink.allow_tags = True
258 selflink.short_description = "Details"
Siobhan Tullyd3515752013-06-21 16:34:53 -0400259
Scott Bakerb27b62c2014-08-15 16:29:16 -0700260 def has_add_permission(self, request):
261 return not request.user.isReadOnlyUser()
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500262
263 def get_readonly_fields(self, request, obj=None):
Scott Bakerb27b62c2014-08-15 16:29:16 -0700264 readonly_fields = list(self.readonly_fields)[:]
265 if request.user.isReadOnlyUser():
266 for field in self.fields:
267 if not field in readonly_fields:
268 readonly_fields.append(field)
269 return readonly_fields
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500270
Scott Baker40c00762014-08-21 16:55:59 -0700271 def backend_status_icon(self, obj):
272 return mark_safe(backend_icon(obj))
273 backend_status_icon.short_description = ""
Scott Baker36f50872014-08-21 13:01:25 -0700274
Scott Bakerb27b62c2014-08-15 16:29:16 -0700275class PlStackGenericTabularInline(generic.GenericTabularInline):
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500276 def has_add_permission(self, request):
Scott Bakerb27b62c2014-08-15 16:29:16 -0700277 return not request.user.isReadOnlyUser()
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500278
Scott Bakerb27b62c2014-08-15 16:29:16 -0700279 def get_readonly_fields(self, request, obj=None):
280 readonly_fields = list(self.readonly_fields)[:]
281 if request.user.isReadOnlyUser():
282 for field in self.fields:
283 if not field in readonly_fields:
284 readonly_fields.append(field)
285 return readonly_fields
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500286
Scott Baker40c00762014-08-21 16:55:59 -0700287 def backend_status_icon(self, obj):
288 return mark_safe(backend_icon(obj))
289 backend_status_icon.short_description = ""
290
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400291class ReservationInline(PlStackTabularInline):
292 model = Reservation
293 extra = 0
294 suit_classes = 'suit-tab suit-tab-reservations'
Scott Baker36f50872014-08-21 13:01:25 -0700295
Tony Mack5b061472014-02-04 07:57:10 -0500296 def queryset(self, request):
297 return Reservation.select_by_user(request.user)
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400298
Scott Bakerb27b62c2014-08-15 16:29:16 -0700299class TagInline(PlStackGenericTabularInline):
Siobhan Tullyde5450d2013-06-21 11:35:33 -0400300 model = Tag
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400301 extra = 0
302 suit_classes = 'suit-tab suit-tab-tags'
Tony Mack5b061472014-02-04 07:57:10 -0500303 fields = ['service', 'name', 'value']
304
305 def queryset(self, request):
306 return Tag.select_by_user(request.user)
Siobhan Tullyde5450d2013-06-21 11:35:33 -0400307
Scott Baker74d8e622013-07-29 16:04:22 -0700308class NetworkLookerUpper:
Siobhan Tully2c780ad2013-09-06 11:22:40 -0400309 """ This is a callable that looks up a network name in a sliver and returns
310 the ip address for that network.
311 """
312
Scott Baker434ca7e2014-08-15 12:29:20 -0700313 byNetworkName = {} # class variable
314
Siobhan Tully2c780ad2013-09-06 11:22:40 -0400315 def __init__(self, name):
316 self.short_description = name
317 self.__name__ = name
318 self.network_name = name
319
320 def __call__(self, obj):
321 if obj is not None:
322 for nbs in obj.networksliver_set.all():
323 if (nbs.network.name == self.network_name):
324 return nbs.ip
Scott Baker74d8e622013-07-29 16:04:22 -0700325 return ""
326
327 def __str__(self):
328 return self.network_name
329
Scott Baker434ca7e2014-08-15 12:29:20 -0700330 @staticmethod
331 def get(network_name):
332 """ We want to make sure we alwars return the same NetworkLookerUpper
333 because sometimes django will cause them to be instantiated multiple
334 times (and we don't want different ones in form.fields vs
335 SliverInline.readonly_fields).
336 """
337 if network_name not in NetworkLookerUpper.byNetworkName:
338 NetworkLookerUpper.byNetworkName[network_name] = NetworkLookerUpper(network_name)
339 return NetworkLookerUpper.byNetworkName[network_name]
340
Siobhan Tullyd3515752013-06-21 16:34:53 -0400341class SliverInline(PlStackTabularInline):
Siobhan Tully4bc09f22013-04-10 21:15:21 -0400342 model = Sliver
Tony Mack93d1b032014-12-08 16:43:02 -0500343 fields = ['backend_status_icon', 'all_ips_string', 'instance_name', 'slice', 'controllerNetwork', 'flavor', 'image', 'node']
Siobhan Tully4bc09f22013-04-10 21:15:21 -0400344 extra = 0
Scott Baker40c00762014-08-21 16:55:59 -0700345 readonly_fields = ['backend_status_icon', 'all_ips_string', 'instance_name']
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400346 suit_classes = 'suit-tab suit-tab-slivers'
Scott Baker74d8e622013-07-29 16:04:22 -0700347
Tony Mack5b061472014-02-04 07:57:10 -0500348 def queryset(self, request):
349 return Sliver.select_by_user(request.user)
350
Scott Bakerb24cc932014-06-09 10:51:16 -0700351 def formfield_for_foreignkey(self, db_field, request=None, **kwargs):
Tony Mack93d1b032014-12-08 16:43:02 -0500352 if db_field.name == 'controllerNetwork':
Scott Baker3b678742014-06-09 13:11:54 -0700353 kwargs['queryset'] = Deployment.select_by_acl(request.user)
Scott Baker7a61dc42014-09-02 17:08:20 -0700354 kwargs['widget'] = forms.Select(attrs={'onChange': "sliver_deployment_changed(this);"})
Scott Baker32481312014-09-08 12:14:14 -0700355 elif db_field.name == 'flavor':
356 kwargs['widget'] = forms.Select(attrs={'onChange': "sliver_flavor_changed(this);"})
Scott Baker3b678742014-06-09 13:11:54 -0700357
358 field = super(SliverInline, self).formfield_for_foreignkey(db_field, request, **kwargs)
Scott Bakerb24cc932014-06-09 10:51:16 -0700359
360 return field
361
Siobhan Tullyd3515752013-06-21 16:34:53 -0400362class SiteInline(PlStackTabularInline):
Siobhan Tully4bc09f22013-04-10 21:15:21 -0400363 model = Site
364 extra = 0
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400365 suit_classes = 'suit-tab suit-tab-sites'
Siobhan Tully4bc09f22013-04-10 21:15:21 -0400366
Tony Mack5b061472014-02-04 07:57:10 -0500367 def queryset(self, request):
368 return Site.select_by_user(request.user)
369
Siobhan Tullyd3515752013-06-21 16:34:53 -0400370class UserInline(PlStackTabularInline):
Siobhan Tully30fd4292013-05-10 08:59:56 -0400371 model = User
Scott Baker40c00762014-08-21 16:55:59 -0700372 fields = ['backend_status_icon', 'email', 'firstname', 'lastname']
373 readonly_fields = ('backend_status_icon', )
Siobhan Tully30fd4292013-05-10 08:59:56 -0400374 extra = 0
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400375 suit_classes = 'suit-tab suit-tab-users'
Siobhan Tully30fd4292013-05-10 08:59:56 -0400376
Tony Mack5b061472014-02-04 07:57:10 -0500377 def queryset(self, request):
378 return User.select_by_user(request.user)
379
Siobhan Tullyd3515752013-06-21 16:34:53 -0400380class SliceInline(PlStackTabularInline):
Tony Mack00d361f2013-04-28 10:28:42 -0400381 model = Slice
Scott Baker40c00762014-08-21 16:55:59 -0700382 fields = ['backend_status_icon', 'name', 'site', 'serviceClass', 'service']
383 readonly_fields = ('backend_status_icon', )
Tony Mack00d361f2013-04-28 10:28:42 -0400384 extra = 0
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400385 suit_classes = 'suit-tab suit-tab-slices'
386
Tony Mack5b061472014-02-04 07:57:10 -0500387 def queryset(self, request):
388 return Slice.select_by_user(request.user)
389
Siobhan Tullyd3515752013-06-21 16:34:53 -0400390class NodeInline(PlStackTabularInline):
Siobhan Tully4bc09f22013-04-10 21:15:21 -0400391 model = Node
392 extra = 0
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400393 suit_classes = 'suit-tab suit-tab-nodes'
Tony Mack93d1b032014-12-08 16:43:02 -0500394 fields = ['backend_status_icon', 'name', 'site_deployment']
Scott Baker40c00762014-08-21 16:55:59 -0700395 readonly_fields = ('backend_status_icon', )
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400396
Tony Mack93d1b032014-12-08 16:43:02 -0500397class DeploymentPrivilegeInline(PlStackTabularInline):
398 model = DeploymentPrivilege
399 extra = 0
400 suit_classes = 'suit-tab suit-tab-admin-only'
401 fields = ['backend_status_icon', 'user','role','deployment']
402 readonly_fields = ('backend_status_icon', )
403
404 def queryset(self, request):
405 return DeploymentPrivilege.select_by_user(request.user)
406
Tony Mack336e0f92014-11-30 15:53:08 -0500407class ControllerPrivilegeInline(PlStackTabularInline):
408 model = ControllerPrivilege
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400409 extra = 0
Tony Mack336e0f92014-11-30 15:53:08 -0500410 suit_classes = 'suit-tab suit-tab-admin-only'
411 fields = ['backend_status_icon', 'user','role','controller']
Scott Baker40c00762014-08-21 16:55:59 -0700412 readonly_fields = ('backend_status_icon', )
Tony Mack5b061472014-02-04 07:57:10 -0500413
414 def queryset(self, request):
Tony Mack336e0f92014-11-30 15:53:08 -0500415 return ControllerPrivilege.select_by_user(request.user)
Siobhan Tully4bc09f22013-04-10 21:15:21 -0400416
Tony Mack93d1b032014-12-08 16:43:02 -0500417class ControllerSiteDeploymentsInline(PlStackTabularInline):
418 model = ControllerSiteDeployments
419 extra = 0
420 suit_classes = 'suit-tab suit-tab-admin-only'
421 fields = ['controller', 'site_deployment', 'tenant_id']
422
Siobhan Tullyd3515752013-06-21 16:34:53 -0400423class SitePrivilegeInline(PlStackTabularInline):
Siobhan Tullyaa1bcd52013-05-10 12:43:09 -0400424 model = SitePrivilege
425 extra = 0
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400426 suit_classes = 'suit-tab suit-tab-siteprivileges'
Scott Baker40c00762014-08-21 16:55:59 -0700427 fields = ['backend_status_icon', 'user','site', 'role']
428 readonly_fields = ('backend_status_icon', )
Siobhan Tullyaa1bcd52013-05-10 12:43:09 -0400429
Tony Mackc2835a92013-05-28 09:18:49 -0400430 def formfield_for_foreignkey(self, db_field, request, **kwargs):
431 if db_field.name == 'site':
Tony Mack5b061472014-02-04 07:57:10 -0500432 kwargs['queryset'] = Site.select_by_user(request.user)
Tony Mackc2835a92013-05-28 09:18:49 -0400433
434 if db_field.name == 'user':
Tony Mack5b061472014-02-04 07:57:10 -0500435 kwargs['queryset'] = User.select_by_user(request.user)
Tony Mackc2835a92013-05-28 09:18:49 -0400436 return super(SitePrivilegeInline, self).formfield_for_foreignkey(db_field, request, **kwargs)
437
Tony Mack5b061472014-02-04 07:57:10 -0500438 def queryset(self, request):
439 return SitePrivilege.select_by_user(request.user)
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400440
Sapan Bhatia39e85582014-11-19 15:07:36 -0500441class SiteDeploymentsInline(PlStackTabularInline):
442 model = SiteDeployments
Tony Macke4be32f2014-03-11 20:45:25 -0400443 extra = 0
444 suit_classes = 'suit-tab suit-tab-deployments'
Tony Mack528d4222014-12-05 17:13:08 -0500445 fields = ['backend_status_icon', 'deployment','site', 'controller']
Scott Baker40c00762014-08-21 16:55:59 -0700446 readonly_fields = ('backend_status_icon', )
Tony Macke4be32f2014-03-11 20:45:25 -0400447
448 def formfield_for_foreignkey(self, db_field, request, **kwargs):
449 if db_field.name == 'site':
450 kwargs['queryset'] = Site.select_by_user(request.user)
451
452 if db_field.name == 'deployment':
453 kwargs['queryset'] = Deployment.select_by_user(request.user)
Tony Mack528d4222014-12-05 17:13:08 -0500454
455 if db_field.name == 'controller':
456 kwargs['queryset'] = Controller.select_by_user(request.user)
457
Sapan Bhatia39e85582014-11-19 15:07:36 -0500458 return super(SiteDeploymentsInline, self).formfield_for_foreignkey(db_field, request, **kwargs)
Tony Macke4be32f2014-03-11 20:45:25 -0400459
460 def queryset(self, request):
Sapan Bhatia39e85582014-11-19 15:07:36 -0500461 return SiteDeployments.select_by_user(request.user)
Tony Macke4be32f2014-03-11 20:45:25 -0400462
463
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400464class SlicePrivilegeInline(PlStackTabularInline):
465 model = SlicePrivilege
466 suit_classes = 'suit-tab suit-tab-sliceprivileges'
467 extra = 0
Scott Baker40c00762014-08-21 16:55:59 -0700468 fields = ('backend_status_icon', 'user', 'slice', 'role')
469 readonly_fields = ('backend_status_icon', )
Siobhan Tullyaa1bcd52013-05-10 12:43:09 -0400470
Tony Mackc2835a92013-05-28 09:18:49 -0400471 def formfield_for_foreignkey(self, db_field, request, **kwargs):
472 if db_field.name == 'slice':
Scott Baker36f50872014-08-21 13:01:25 -0700473 kwargs['queryset'] = Slice.select_by_user(request.user)
Tony Mackc2835a92013-05-28 09:18:49 -0400474 if db_field.name == 'user':
Scott Baker36f50872014-08-21 13:01:25 -0700475 kwargs['queryset'] = User.select_by_user(request.user)
Tony Mackc2835a92013-05-28 09:18:49 -0400476
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400477 return super(SlicePrivilegeInline, self).formfield_for_foreignkey(db_field, request, **kwargs)
Tony Mackc2835a92013-05-28 09:18:49 -0400478
Tony Mack5b061472014-02-04 07:57:10 -0500479 def queryset(self, request):
480 return SlicePrivilege.select_by_user(request.user)
481
Scott Bakera0015eb2013-08-14 17:28:14 -0700482class SliceNetworkInline(PlStackTabularInline):
Scott Baker74d8e622013-07-29 16:04:22 -0700483 model = Network.slices.through
Scott Baker874936e2014-01-13 18:15:34 -0800484 selflink_fieldname = "network"
Scott Baker74d8e622013-07-29 16:04:22 -0700485 extra = 0
486 verbose_name = "Network Connection"
487 verbose_name_plural = "Network Connections"
Siobhan Tully2d95e482013-09-06 10:56:06 -0400488 suit_classes = 'suit-tab suit-tab-slicenetworks'
Scott Baker40c00762014-08-21 16:55:59 -0700489 fields = ['backend_status_icon', 'network']
490 readonly_fields = ('backend_status_icon', )
Scott Baker2170b972014-06-03 12:14:07 -0700491
Sapan Bhatiae9f96f62014-11-19 15:10:16 -0500492class ImageDeploymentsInline(PlStackTabularInline):
493 model = ImageDeployments
Scott Baker2170b972014-06-03 12:14:07 -0700494 extra = 0
495 verbose_name = "Image Deployments"
496 verbose_name_plural = "Image Deployments"
497 suit_classes = 'suit-tab suit-tab-imagedeployments'
Tony Mack336e0f92014-11-30 15:53:08 -0500498 fields = ['backend_status_icon', 'image', 'deployment']
499 readonly_fields = ['backend_status_icon']
500
501class ControllerImagesInline(PlStackTabularInline):
502 model = ControllerImages
503 extra = 0
504 verbose_name = "Controller Images"
505 verbose_name_plural = "Controller Images"
506 suit_classes = 'suit-tab suit-tab-admin-only'
507 fields = ['backend_status_icon', 'image', 'controller', 'glance_image_id']
Scott Baker40c00762014-08-21 16:55:59 -0700508 readonly_fields = ['backend_status_icon', 'glance_image_id']
Scott Baker74d8e622013-07-29 16:04:22 -0700509
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400510class SliceRoleAdmin(PlanetStackBaseAdmin):
511 model = SliceRole
512 pass
513
514class SiteRoleAdmin(PlanetStackBaseAdmin):
515 model = SiteRole
516 pass
517
Siobhan Tullybf1153a2013-05-27 20:53:48 -0400518class DeploymentAdminForm(forms.ModelForm):
Siobhan Tully4bc09f22013-04-10 21:15:21 -0400519 sites = forms.ModelMultipleChoiceField(
520 queryset=Site.objects.all(),
521 required=False,
Scott Baker70983182014-06-09 22:10:00 -0700522 help_text="Select which sites are allowed to host nodes in this deployment",
Siobhan Tully4bc09f22013-04-10 21:15:21 -0400523 widget=FilteredSelectMultiple(
524 verbose_name=('Sites'), is_stacked=False
525 )
526 )
Scott Bakerde0f4412014-06-11 15:40:26 -0700527 images = forms.ModelMultipleChoiceField(
528 queryset=Image.objects.all(),
529 required=False,
530 help_text="Select which images should be deployed on this deployment",
531 widget=FilteredSelectMultiple(
532 verbose_name=('Images'), is_stacked=False
533 )
534 )
Scott Baker37b47902014-09-02 14:37:41 -0700535 flavors = forms.ModelMultipleChoiceField(
536 queryset=Flavor.objects.all(),
537 required=False,
538 help_text="Select which flavors should be usable on this deployment",
539 widget=FilteredSelectMultiple(
540 verbose_name=('Flavors'), is_stacked=False
541 )
542 )
Siobhan Tully4bc09f22013-04-10 21:15:21 -0400543 class Meta:
Siobhan Tullybf1153a2013-05-27 20:53:48 -0400544 model = Deployment
Scott Baker37b47902014-09-02 14:37:41 -0700545 many_to_many = ["flavors",]
Siobhan Tully4bc09f22013-04-10 21:15:21 -0400546
Siobhan Tully320b4622014-01-17 15:11:14 -0500547 def __init__(self, *args, **kwargs):
Scott Baker5380c522014-06-06 14:49:43 -0700548 request = kwargs.pop('request', None)
Siobhan Tully320b4622014-01-17 15:11:14 -0500549 super(DeploymentAdminForm, self).__init__(*args, **kwargs)
550
Scott Baker5380c522014-06-06 14:49:43 -0700551 self.fields['accessControl'].initial = "allow site " + request.user.site.name
552
Siobhan Tully320b4622014-01-17 15:11:14 -0500553 if self.instance and self.instance.pk:
Scott Baker9f6b8ed2014-11-17 23:44:03 -0800554 self.fields['sites'].initial = [x.site for x in self.instance.sitedeployments.all()]
555 self.fields['images'].initial = [x.image for x in self.instance.imagedeployments.all()]
Scott Baker37b47902014-09-02 14:37:41 -0700556 self.fields['flavors'].initial = self.instance.flavors.all()
Scott Bakerde0f4412014-06-11 15:40:26 -0700557
558 def manipulate_m2m_objs(self, this_obj, selected_objs, all_relations, relation_class, local_attrname, foreign_attrname):
559 """ helper function for handling m2m relations from the MultipleChoiceField
560
561 this_obj: the source object we want to link from
562
563 selected_objs: a list of destination objects we want to link to
564
565 all_relations: the full set of relations involving this_obj, including ones we don't want
566
567 relation_class: the class that implements the relation from source to dest
568
569 local_attrname: field name representing this_obj in relation_class
570
571 foreign_attrname: field name representing selected_objs in relation_class
572
573 This function will remove all newobjclass relations from this_obj
574 that are not contained in selected_objs, and add any relations that
575 are in selected_objs but don't exist in the data model yet.
576 """
577
578 existing_dest_objs = []
579 for relation in list(all_relations):
580 if getattr(relation, foreign_attrname) not in selected_objs:
581 #print "deleting site", sdp.site
582 relation.delete()
583 else:
584 existing_dest_objs.append(getattr(relation, foreign_attrname))
585
586 for dest_obj in selected_objs:
587 if dest_obj not in existing_dest_objs:
588 #print "adding site", site
589 kwargs = {foreign_attrname: dest_obj, local_attrname: this_obj}
590 relation = relation_class(**kwargs)
591 relation.save()
Siobhan Tully320b4622014-01-17 15:11:14 -0500592
593 def save(self, commit=True):
594 deployment = super(DeploymentAdminForm, self).save(commit=False)
595
596 if commit:
597 deployment.save()
Scott Baker61b6aec2014-10-06 17:17:40 -0700598 # this has to be done after save() if/when a deployment is first created
599 deployment.flavors = self.cleaned_data['flavors']
Siobhan Tully320b4622014-01-17 15:11:14 -0500600
601 if deployment.pk:
Scott Bakerc9b14f72014-05-22 13:44:20 -0700602 # save_m2m() doesn't seem to work with 'through' relations. So we
603 # create/destroy the through models ourselves. There has to be
604 # a better way...
605
Sapan Bhatia39e85582014-11-19 15:07:36 -0500606 self.manipulate_m2m_objs(deployment, self.cleaned_data['sites'], deployment.sitedeployments.all(), SiteDeployments, "deployment", "site")
Tony Mack93d1b032014-12-08 16:43:02 -0500607 self.manipulate_m2m_objs(deployment, self.cleaned_data['images'], deployment.imagedeployments.all(), DeploymentImages, "deployment", "image")
Scott Bakerc9b14f72014-05-22 13:44:20 -0700608
Scott Baker37b47902014-09-02 14:37:41 -0700609 self.save_m2m()
Siobhan Tully320b4622014-01-17 15:11:14 -0500610
611 return deployment
612
Scott Bakerff5e0f32014-05-22 14:40:27 -0700613class DeploymentAdminROForm(DeploymentAdminForm):
614 def save(self, commit=True):
615 raise PermissionDenied
616
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500617class SiteAssocInline(PlStackTabularInline):
618 model = Site.deployments.through
619 extra = 0
620 suit_classes = 'suit-tab suit-tab-sites'
Siobhan Tully4bc09f22013-04-10 21:15:21 -0400621
Siobhan Tullybf1153a2013-05-27 20:53:48 -0400622class DeploymentAdmin(PlanetStackBaseAdmin):
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500623 model = Deployment
Tony Mack93d1b032014-12-08 16:43:02 -0500624 fieldList = ['backend_status_text', 'name', 'sites', 'images', 'flavors', 'accessControl']
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500625 fieldsets = [(None, {'fields': fieldList, 'classes':['suit-tab suit-tab-sites']})]
Tony Mack93d1b032014-12-08 16:43:02 -0500626 # node no longer directly connected to deployment
627 #inlines = [DeploymentPrivilegeInline,NodeInline,TagInline,ImageDeploymentsInline]
628 inlines = [DeploymentPrivilegeInline,TagInline,ImageDeploymentsInline]
Scott Baker63d1a552014-08-21 15:19:07 -0700629 list_display = ['backend_status_icon', 'name']
630 list_display_links = ('backend_status_icon', 'name', )
Scott Baker40c00762014-08-21 16:55:59 -0700631 readonly_fields = ('backend_status_text', )
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500632
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500633 user_readonly_fields = ['name']
634
Tony Mack93d1b032014-12-08 16:43:02 -0500635 # nodes no longer direclty connected to deployments
636 #suit_form_tabs =(('sites','Deployment Details'),('nodes','Nodes'),('deploymentprivileges','Privileges'),('tags','Tags'),('imagedeployments','Images'))
637 suit_form_tabs =(('sites','Deployment Details'),('deploymentprivileges','Privileges'),('tags','Tags'),('imagedeployments','Images'))
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500638
Scott Bakerff5e0f32014-05-22 14:40:27 -0700639 def get_form(self, request, obj=None, **kwargs):
640 if request.user.isReadOnlyUser():
641 kwargs["form"] = DeploymentAdminROForm
642 else:
643 kwargs["form"] = DeploymentAdminForm
Scott Baker5380c522014-06-06 14:49:43 -0700644 adminForm = super(DeploymentAdmin,self).get_form(request, obj, **kwargs)
645
646 # from stackexchange: pass the request object into the form
647
648 class AdminFormMetaClass(adminForm):
649 def __new__(cls, *args, **kwargs):
650 kwargs['request'] = request
651 return adminForm(*args, **kwargs)
652
653 return AdminFormMetaClass
654
Tony Mack528d4222014-12-05 17:13:08 -0500655class ControllerAdminForm(forms.ModelForm):
656 site_deployments = forms.ModelMultipleChoiceField(
Tony Mack93d1b032014-12-08 16:43:02 -0500657 queryset=SiteDeployments.objects.all(),
Tony Mack528d4222014-12-05 17:13:08 -0500658 required=False,
659 help_text="Select which sites deployments are managed by this controller",
660 widget=FilteredSelectMultiple(
661 verbose_name=('Site Deployments'), is_stacked=False
662 )
663 )
664
665 class Meta:
666 model = Controller
667
Tony Mack93d1b032014-12-08 16:43:02 -0500668 def __init__(self, *args, **kwargs):
Tony Mack528d4222014-12-05 17:13:08 -0500669 request = kwargs.pop('request', None)
670 super(ControllerAdminForm, self).__init__(*args, **kwargs)
671
672 if self.instance and self.instance.pk:
673 self.fields['site_deployments'].initial = [x.site_deployment for x in self.instance.controllersitedeployments.all()]
674
Tony Mack93d1b032014-12-08 16:43:02 -0500675 def manipulate_m2m_objs(self, this_obj, selected_objs, all_relations, relation_class, local_attrname, foreign_attrname):
Tony Mack528d4222014-12-05 17:13:08 -0500676 """ helper function for handling m2m relations from the MultipleChoiceField
677 this_obj: the source object we want to link from
678 selected_objs: a list of destination objects we want to link to
679 all_relations: the full set of relations involving this_obj, including ones we don't want
680 relation_class: the class that implements the relation from source to dest
681 local_attrname: field name representing this_obj in relation_class
682 foreign_attrname: field name representing selected_objs in relation_class
683 This function will remove all newobjclass relations from this_obj
684 that are not contained in selected_objs, and add any relations that
685 are in selected_objs but don't exist in the data model yet.
686 """
687 existing_dest_objs = []
688 for relation in list(all_relations):
689 if getattr(relation, foreign_attrname) not in selected_objs:
690 #print "deleting site", sdp.site
691 relation.delete()
692 else:
693 existing_dest_objs.append(getattr(relation, foreign_attrname))
694
695 for dest_obj in selected_objs:
696 if dest_obj not in existing_dest_objs:
697 #print "adding site", site
698 kwargs = {foreign_attrname: dest_obj, local_attrname: this_obj}
699 relation = relation_class(**kwargs)
700 relation.save()
701
Tony Mack93d1b032014-12-08 16:43:02 -0500702 def save(self, commit=True):
703 controller = super(ControllerAdminForm, self).save(commit=False)
704 if commit:
705 controller.save()
Tony Mack528d4222014-12-05 17:13:08 -0500706
Tony Mack93d1b032014-12-08 16:43:02 -0500707 if controller.pk:
708 # save_m2m() doesn't seem to work with 'through' relations. So we
709 # create/destroy the through models ourselves. There has to be
710 # a better way...
Tony Mack528d4222014-12-05 17:13:08 -0500711 self.manipulate_m2m_objs(controller, self.cleaned_data['site_deployments'], controller.controllersitedeployments.all(), ControllerSiteDeployments, "controller", "site_deployment")
712
Tony Mack93d1b032014-12-08 16:43:02 -0500713 self.save_m2m()
Tony Mack528d4222014-12-05 17:13:08 -0500714
Tony Mack93d1b032014-12-08 16:43:02 -0500715 return controller
Tony Mack528d4222014-12-05 17:13:08 -0500716
717class ControllerAdmin(PlanetStackBaseAdmin):
718 model = Controller
Tony Mack93d1b032014-12-08 16:43:02 -0500719 fieldList = ['name', 'version', 'backend_type', 'auth_url', 'admin_user', 'admin_tenant','admin_password']
720 #fieldsets = [(None, {'fields': fieldList, 'classes':['suit-tab suit-tab-general']})]
721 inlines = [ControllerSiteDeploymentsInline] # ,ControllerImagesInline]
Tony Mack528d4222014-12-05 17:13:08 -0500722 list_display = ['backend_status_icon', 'name', 'version', 'backend_type']
723 list_display_links = ('backend_status_icon', 'name', )
724 readonly_fields = ('backend_status_text',)
725
726 user_readonly_fields = []
727
728 def get_form(self, request, obj=None, **kwargs):
Tony Mack93d1b032014-12-08 16:43:02 -0500729 print self.fieldsets
Tony Mack528d4222014-12-05 17:13:08 -0500730 if request.user.isReadOnlyUser():
731 kwargs["form"] = ControllerAdminROForm
732 else:
733 kwargs["form"] = ControllerAdminForm
Tony Mack93d1b032014-12-08 16:43:02 -0500734 adminForm = super(ControllerAdmin,self).get_form(request, obj, **kwargs)
Tony Mack528d4222014-12-05 17:13:08 -0500735
736 # from stackexchange: pass the request object into the form
737
738 class AdminFormMetaClass(adminForm):
739 def __new__(cls, *args, **kwargs):
740 kwargs['request'] = request
741 return adminForm(*args, **kwargs)
742
743 return AdminFormMetaClass
744
Siobhan Tullyce652d02013-10-08 21:52:35 -0400745class ServiceAttrAsTabInline(PlStackTabularInline):
746 model = ServiceAttribute
747 fields = ['name','value']
748 extra = 0
749 suit_classes = 'suit-tab suit-tab-serviceattrs'
750
Siobhan Tullyce652d02013-10-08 21:52:35 -0400751class ServiceAdmin(PlanetStackBaseAdmin):
Scott Baker63d1a552014-08-21 15:19:07 -0700752 list_display = ("backend_status_icon","name","description","versionNumber","enabled","published")
753 list_display_links = ('backend_status_icon', 'name', )
Scott Baker40c00762014-08-21 16:55:59 -0700754 fieldList = ["backend_status_text","name","description","versionNumber","enabled","published"]
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500755 fieldsets = [(None, {'fields': fieldList, 'classes':['suit-tab suit-tab-general']})]
756 inlines = [ServiceAttrAsTabInline,SliceInline]
Scott Baker40c00762014-08-21 16:55:59 -0700757 readonly_fields = ('backend_status_text', )
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500758
759 user_readonly_fields = fieldList
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500760
761 suit_form_tabs =(('general', 'Service Details'),
762 ('slices','Slices'),
763 ('serviceattrs','Additional Attributes'),
764 )
Siobhan Tullyce652d02013-10-08 21:52:35 -0400765
Tony Mack0553f282013-06-10 22:54:50 -0400766class SiteAdmin(PlanetStackBaseAdmin):
Scott Baker40c00762014-08-21 16:55:59 -0700767 fieldList = ['backend_status_text', 'name', 'site_url', 'enabled', 'is_public', 'login_base', 'accountLink','location']
Siobhan Tully4bc09f22013-04-10 21:15:21 -0400768 fieldsets = [
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500769 (None, {'fields': fieldList, 'classes':['suit-tab suit-tab-general']}),
Tony Macke4be32f2014-03-11 20:45:25 -0400770 #('Deployment Networks', {'fields': ['deployments'], 'classes':['suit-tab suit-tab-deployments']}),
Siobhan Tully4bc09f22013-04-10 21:15:21 -0400771 ]
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400772 suit_form_tabs =(('general', 'Site Details'),
773 ('users','Users'),
Siobhan Tully2d95e482013-09-06 10:56:06 -0400774 ('siteprivileges','Privileges'),
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400775 ('deployments','Deployments'),
776 ('slices','Slices'),
Tony Mack93d1b032014-12-08 16:43:02 -0500777 #('nodes','Nodes'),
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400778 ('tags','Tags'),
779 )
Scott Baker40c00762014-08-21 16:55:59 -0700780 readonly_fields = ['backend_status_text', 'accountLink']
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500781
782 user_readonly_fields = ['name', 'deployments','site_url', 'enabled', 'is_public', 'login_base', 'accountLink']
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500783
Scott Baker63d1a552014-08-21 15:19:07 -0700784 list_display = ('backend_status_icon', 'name', 'login_base','site_url', 'enabled')
785 list_display_links = ('backend_status_icon', 'name', )
Siobhan Tully4bc09f22013-04-10 21:15:21 -0400786 filter_horizontal = ('deployments',)
Tony Mack93d1b032014-12-08 16:43:02 -0500787 inlines = [SliceInline,UserInline,TagInline, SitePrivilegeInline, SiteDeploymentsInline]
Siobhan Tully4bc09f22013-04-10 21:15:21 -0400788 search_fields = ['name']
789
Tony Mack04062832013-05-10 08:22:44 -0400790 def queryset(self, request):
Tony Mack5b061472014-02-04 07:57:10 -0500791 return Site.select_by_user(request.user)
Tony Mack04062832013-05-10 08:22:44 -0400792
Tony Mack5cd13202013-05-01 21:48:38 -0400793 def get_formsets(self, request, obj=None):
794 for inline in self.get_inline_instances(request, obj):
795 # hide MyInline in the add view
796 if obj is None:
797 continue
Tony Mack2bd5b412013-06-11 21:05:06 -0400798 if isinstance(inline, SliverInline):
799 inline.model.caller = request.user
Tony Mack5cd13202013-05-01 21:48:38 -0400800 yield inline.get_formset(request, obj)
801
Scott Baker545db2a2013-12-09 18:44:43 -0800802 def accountLink(self, obj):
803 link_obj = obj.accounts.all()
804 if link_obj:
805 reverse_path = "admin:core_account_change"
806 url = reverse(reverse_path, args =(link_obj[0].id,))
807 return "<a href='%s'>%s</a>" % (url, "view billing details")
808 else:
809 return "no billing data for this site"
810 accountLink.allow_tags = True
811 accountLink.short_description = "Billing"
812
Tony Mack332ee1d2014-02-04 15:33:45 -0500813 def save_model(self, request, obj, form, change):
814 # update openstack connection to use this site/tenant
815 obj.save_by_user(request.user)
816
817 def delete_model(self, request, obj):
818 obj.delete_by_user(request.user)
819
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500820
Tony Mack9bcbe4f2013-04-29 08:13:27 -0400821class SitePrivilegeAdmin(PlanetStackBaseAdmin):
Scott Baker40c00762014-08-21 16:55:59 -0700822 fieldList = ['backend_status_text', 'user', 'site', 'role']
Tony Mack00d361f2013-04-28 10:28:42 -0400823 fieldsets = [
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500824 (None, {'fields': fieldList, 'classes':['collapse']})
Tony Mack00d361f2013-04-28 10:28:42 -0400825 ]
Scott Baker40c00762014-08-21 16:55:59 -0700826 readonly_fields = ('backend_status_text', )
Scott Baker63d1a552014-08-21 15:19:07 -0700827 list_display = ('backend_status_icon', 'user', 'site', 'role')
828 list_display_links = list_display
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500829 user_readonly_fields = fieldList
830 user_readonly_inlines = []
Tony Mack00d361f2013-04-28 10:28:42 -0400831
Tony Mackc2835a92013-05-28 09:18:49 -0400832 def formfield_for_foreignkey(self, db_field, request, **kwargs):
833 if db_field.name == 'site':
834 if not request.user.is_admin:
835 # only show sites where user is an admin or pi
836 sites = set()
837 for site_privilege in SitePrivilege.objects.filer(user=request.user):
838 if site_privilege.role.role_type in ['admin', 'pi']:
839 sites.add(site_privilege.site)
840 kwargs['queryset'] = Site.objects.filter(site__in=list(sites))
841
842 if db_field.name == 'user':
843 if not request.user.is_admin:
844 # only show users from sites where caller has admin or pi role
845 roles = Role.objects.filter(role_type__in=['admin', 'pi'])
846 site_privileges = SitePrivilege.objects.filter(user=request.user).filter(role__in=roles)
847 sites = [site_privilege.site for site_privilege in site_privileges]
848 site_privileges = SitePrivilege.objects.filter(site__in=sites)
849 emails = [site_privilege.user.email for site_privilege in site_privileges]
850 users = User.objects.filter(email__in=emails)
851 kwargs['queryset'] = users
852
853 return super(SitePrivilegeAdmin, self).formfield_for_foreignkey(db_field, request, **kwargs)
854
Tony Mack04062832013-05-10 08:22:44 -0400855 def queryset(self, request):
856 # admins can see all privileges. Users can only see privileges at sites
Tony Mackc2835a92013-05-28 09:18:49 -0400857 # where they have the admin role or pi role.
Tony Mack04062832013-05-10 08:22:44 -0400858 qs = super(SitePrivilegeAdmin, self).queryset(request)
Tony Mack5b061472014-02-04 07:57:10 -0500859 #if not request.user.is_admin:
860 # roles = Role.objects.filter(role_type__in=['admin', 'pi'])
861 # site_privileges = SitePrivilege.objects.filter(user=request.user).filter(role__in=roles)
862 # login_bases = [site_privilege.site.login_base for site_privilege in site_privileges]
863 # sites = Site.objects.filter(login_base__in=login_bases)
864 # qs = qs.filter(site__in=sites)
Tony Mack04062832013-05-10 08:22:44 -0400865 return qs
866
Siobhan Tullyce652d02013-10-08 21:52:35 -0400867class SliceForm(forms.ModelForm):
868 class Meta:
869 model = Slice
870 widgets = {
Scott Baker36f50872014-08-21 13:01:25 -0700871 'service': LinkedSelect
Siobhan Tullyce652d02013-10-08 21:52:35 -0400872 }
873
Tony Mack2cbd3802014-09-29 16:10:52 -0400874 def clean(self):
875 cleaned_data = super(SliceForm, self).clean()
876 name = cleaned_data.get('name')
Scott Baker6efad462014-10-06 23:09:59 -0700877 site = cleaned_data.get('site')
Tony Mack585cb192014-10-22 12:54:19 -0400878 slice_id = self.instance.id
879 if not site and slice_id:
880 site = Slice.objects.get(id=slice_id).site
Scott Baker6efad462014-10-06 23:09:59 -0700881 if (not isinstance(site,Site)):
882 # previous code indicates 'site' could be a site_id and not a site?
883 site = Slice.objects.get(id=site.id)
Tony Mack2cbd3802014-09-29 16:10:52 -0400884 if not name.startswith(site.login_base):
885 raise forms.ValidationError('slice name must begin with %s' % site.login_base)
886 return cleaned_data
887
Tony Mack336e0f92014-11-30 15:53:08 -0500888class ControllerSlicesInline(PlStackTabularInline):
889 model = ControllerSlices
Scott Bakerc4efdc72014-10-15 16:54:04 -0700890 extra = 0
Tony Mack336e0f92014-11-30 15:53:08 -0500891 verbose_name = "Controller Slices"
892 verbose_name_plural = "Controller Slices"
Scott Bakerc4efdc72014-10-15 16:54:04 -0700893 suit_classes = 'suit-tab suit-tab-admin-only'
Tony Mack336e0f92014-11-30 15:53:08 -0500894 fields = ['backend_status_icon', 'controller', 'tenant_id']
Scott Bakerc4efdc72014-10-15 16:54:04 -0700895 readonly_fields = ('backend_status_icon', )
896
Tony Mack2bd5b412013-06-11 21:05:06 -0400897class SliceAdmin(PlanetStackBaseAdmin):
Siobhan Tullyce652d02013-10-08 21:52:35 -0400898 form = SliceForm
Tony Mackfbb26fc2014-09-02 07:03:27 -0400899 fieldList = ['backend_status_text', 'site', 'name', 'serviceClass', 'enabled','description', 'service', 'slice_url', 'max_slivers']
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500900 fieldsets = [('Slice Details', {'fields': fieldList, 'classes':['suit-tab suit-tab-general']}),]
Scott Baker40c00762014-08-21 16:55:59 -0700901 readonly_fields = ('backend_status_text', )
Tony Mack7d459902014-09-03 13:18:57 -0400902 list_display = ('backend_status_icon', 'name', 'site','serviceClass', 'slice_url', 'max_slivers')
903 list_display_links = ('backend_status_icon', 'name', )
Siobhan Tully2d95e482013-09-06 10:56:06 -0400904 inlines = [SlicePrivilegeInline,SliverInline, TagInline, ReservationInline,SliceNetworkInline]
Tony Mack336e0f92014-11-30 15:53:08 -0500905 admin_inlines = [ControllerSlicesInline]
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400906
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500907 user_readonly_fields = fieldList
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400908
Scott Bakerc4efdc72014-10-15 16:54:04 -0700909 @property
910 def suit_form_tabs(self):
911 tabs =[('general', 'Slice Details'),
912 ('slicenetworks','Networks'),
913 ('sliceprivileges','Privileges'),
914 ('slivers','Slivers'),
915 ('tags','Tags'),
916 ('reservations','Reservations'),
917 ]
918
919 request=getattr(_thread_locals, "request", None)
920 if request and request.user.is_admin:
921 tabs.append( ('admin-only', 'Admin-Only') )
922
923 return tabs
Tony Mack7b8505a2014-10-22 11:54:29 -0400924
925 def add_view(self, request, form_url='', extra_context=None):
926 # revert to default read-only fields
927 self.readonly_fields = ('backend_status_text',)
928 return super(SliceAdmin, self).add_view(request, form_url, extra_context=extra_context)
929
930 def change_view(self, request, object_id, form_url='', extra_context=None):
Tony Mack7b8505a2014-10-22 11:54:29 -0400931 # cannot change the site of an existing slice so make the site field read only
932 if object_id:
933 self.readonly_fields = ('backend_status_text','site')
934 return super(SliceAdmin, self).change_view(request, object_id, form_url)
Siobhan Tully4bc09f22013-04-10 21:15:21 -0400935
Scott Baker510fdbb2014-08-05 17:19:24 -0700936 def render_change_form(self, request, context, add=False, change=False, form_url='', obj=None):
Scott Baker510fdbb2014-08-05 17:19:24 -0700937 deployment_nodes = []
938 for node in Node.objects.all():
939 deployment_nodes.append( (node.deployment.id, node.id, node.name) )
940
Scott Baker7a61dc42014-09-02 17:08:20 -0700941 deployment_flavors = []
942 for flavor in Flavor.objects.all():
943 for deployment in flavor.deployments.all():
944 deployment_flavors.append( (deployment.id, flavor.id, flavor.name) )
945
Tony Mack93d1b032014-12-08 16:43:02 -0500946 deployment_images = []
Scott Baker93e80cd2014-09-09 09:58:49 -0700947 for image in Image.objects.all():
Tony Mack93d1b032014-12-08 16:43:02 -0500948 for deployment_image in image.imagedeployments.all():
949 deployment_images.append( (deployment_image.controller.id, image.id, image.name) )
Scott Baker93e80cd2014-09-09 09:58:49 -0700950
Tony Mackec23b992014-09-02 21:18:45 -0400951 site_login_bases = []
952 for site in Site.objects.all():
Scott Baker93e80cd2014-09-09 09:58:49 -0700953 site_login_bases.append((site.id, site.login_base))
954
Scott Baker510fdbb2014-08-05 17:19:24 -0700955 context["deployment_nodes"] = deployment_nodes
Scott Baker7a61dc42014-09-02 17:08:20 -0700956 context["deployment_flavors"] = deployment_flavors
Scott Baker93e80cd2014-09-09 09:58:49 -0700957 context["deployment_images"] = deployment_images
Tony Mackec23b992014-09-02 21:18:45 -0400958 context["site_login_bases"] = site_login_bases
Scott Baker510fdbb2014-08-05 17:19:24 -0700959 return super(SliceAdmin, self).render_change_form(request, context, add, change, form_url, obj)
960
Tony Mackc2835a92013-05-28 09:18:49 -0400961 def formfield_for_foreignkey(self, db_field, request, **kwargs):
962 if db_field.name == 'site':
Tony Mack5b061472014-02-04 07:57:10 -0500963 kwargs['queryset'] = Site.select_by_user(request.user)
Tony Mackec23b992014-09-02 21:18:45 -0400964 kwargs['widget'] = forms.Select(attrs={'onChange': "update_slice_prefix(this, $($(this).closest('fieldset')[0]).find('.field-name input')[0].id)"})
Scott Baker40c00762014-08-21 16:55:59 -0700965
Tony Mackc2835a92013-05-28 09:18:49 -0400966 return super(SliceAdmin, self).formfield_for_foreignkey(db_field, request, **kwargs)
967
Tony Mack04062832013-05-10 08:22:44 -0400968 def queryset(self, request):
969 # admins can see all keys. Users can only see slices they belong to.
Tony Mack5b061472014-02-04 07:57:10 -0500970 return Slice.select_by_user(request.user)
Tony Mack04062832013-05-10 08:22:44 -0400971
Tony Mack79748612013-05-01 14:52:03 -0400972 def get_formsets(self, request, obj=None):
973 for inline in self.get_inline_instances(request, obj):
974 # hide MyInline in the add view
975 if obj is None:
976 continue
Tony Mack2bd5b412013-06-11 21:05:06 -0400977 if isinstance(inline, SliverInline):
978 inline.model.caller = request.user
Tony Mack79748612013-05-01 14:52:03 -0400979 yield inline.get_formset(request, obj)
980
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400981class SlicePrivilegeAdmin(PlanetStackBaseAdmin):
Tony Mack00d361f2013-04-28 10:28:42 -0400982 fieldsets = [
Scott Baker40c00762014-08-21 16:55:59 -0700983 (None, {'fields': ['backend_status_text', 'user', 'slice', 'role']})
Tony Mack00d361f2013-04-28 10:28:42 -0400984 ]
Scott Baker40c00762014-08-21 16:55:59 -0700985 readonly_fields = ('backend_status_text', )
Scott Baker63d1a552014-08-21 15:19:07 -0700986 list_display = ('backend_status_icon', 'user', 'slice', 'role')
987 list_display_links = list_display
Tony Mack00d361f2013-04-28 10:28:42 -0400988
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500989 user_readonly_fields = ['user', 'slice', 'role']
990 user_readonly_inlines = []
991
Tony Mackc2835a92013-05-28 09:18:49 -0400992 def formfield_for_foreignkey(self, db_field, request, **kwargs):
993 if db_field.name == 'slice':
Tony Mack5b061472014-02-04 07:57:10 -0500994 kwargs['queryset'] = Slice.select_by_user(request.user)
Tony Mackc2835a92013-05-28 09:18:49 -0400995
996 if db_field.name == 'user':
Tony Mack5b061472014-02-04 07:57:10 -0500997 kwargs['queryset'] = User.select_by_user(request.user)
Tony Mackc2835a92013-05-28 09:18:49 -0400998
Siobhan Tullybfd11dc2013-09-03 12:59:24 -0400999 return super(SlicePrivilegeAdmin, self).formfield_for_foreignkey(db_field, request, **kwargs)
Tony Mackc2835a92013-05-28 09:18:49 -04001000
Tony Mack04062832013-05-10 08:22:44 -04001001 def queryset(self, request):
1002 # admins can see all memberships. Users can only see memberships of
1003 # slices where they have the admin role.
Tony Mack5b061472014-02-04 07:57:10 -05001004 return SlicePrivilege.select_by_user(request.user)
Siobhan Tully4bc09f22013-04-10 21:15:21 -04001005
Tony Mack9bcbe4f2013-04-29 08:13:27 -04001006 def save_model(self, request, obj, form, change):
Tony Mack951dab42013-05-02 19:51:45 -04001007 # update openstack connection to use this site/tenant
1008 auth = request.session.get('auth', {})
Tony Mackf7f79a12014-08-11 11:21:42 -04001009 auth['tenant'] = obj.slice.slicename
Tony Mack951dab42013-05-02 19:51:45 -04001010 obj.os_manager = OpenStackManager(auth=auth, caller=request.user)
Tony Mack9bcbe4f2013-04-29 08:13:27 -04001011 obj.save()
1012
1013 def delete_model(self, request, obj):
Tony Mack951dab42013-05-02 19:51:45 -04001014 # update openstack connection to use this site/tenant
1015 auth = request.session.get('auth', {})
Tony Mackf7f79a12014-08-11 11:21:42 -04001016 auth['tenant'] = obj.slice.slicename
Tony Mack951dab42013-05-02 19:51:45 -04001017 obj.os_manager = OpenStackManager(auth=auth, caller=request.user)
Tony Mack9bcbe4f2013-04-29 08:13:27 -04001018 obj.delete()
1019
Siobhan Tully567e3e62013-06-21 18:03:16 -04001020
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001021class ImageAdmin(PlanetStackBaseAdmin):
1022
Scott Baker36f50872014-08-21 13:01:25 -07001023 fieldsets = [('Image Details',
Scott Baker40c00762014-08-21 16:55:59 -07001024 {'fields': ['backend_status_text', 'name', 'disk_format', 'container_format'],
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001025 'classes': ['suit-tab suit-tab-general']})
1026 ]
Scott Baker40c00762014-08-21 16:55:59 -07001027 readonly_fields = ('backend_status_text', )
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001028
Tony Mack336e0f92014-11-30 15:53:08 -05001029 suit_form_tabs =(('general','Image Details'),('slivers','Slivers'),('imagedeployments','Deployments'), ('controllerimages', 'Controllers'))
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001030
Tony Mack336e0f92014-11-30 15:53:08 -05001031 inlines = [SliverInline, ControllerImagesInline]
Scott Bakerb6f99242014-06-11 11:34:44 -07001032
Tony Mack32e1ce32014-05-07 13:29:41 -04001033 user_readonly_fields = ['name', 'disk_format', 'container_format']
Scott Bakerb27b62c2014-08-15 16:29:16 -07001034
Scott Baker63d1a552014-08-21 15:19:07 -07001035 list_display = ['backend_status_icon', 'name']
1036 list_display_links = ('backend_status_icon', 'name', )
1037
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001038class NodeForm(forms.ModelForm):
1039 class Meta:
1040 widgets = {
1041 'site': LinkedSelect,
1042 'deployment': LinkedSelect
1043 }
Siobhan Tully4bc09f22013-04-10 21:15:21 -04001044
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001045class NodeAdmin(PlanetStackBaseAdmin):
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001046 form = NodeForm
Tony Mack93d1b032014-12-08 16:43:02 -05001047 list_display = ('backend_status_icon', 'name', 'site_deployment')
Scott Baker63d1a552014-08-21 15:19:07 -07001048 list_display_links = ('backend_status_icon', 'name', )
Tony Mack93d1b032014-12-08 16:43:02 -05001049 list_filter = ('site_deployment',)
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001050
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001051 inlines = [TagInline,SliverInline]
Tony Mack93d1b032014-12-08 16:43:02 -05001052 fieldsets = [('Node Details', {'fields': ['backend_status_text', 'name','site_deployment'], 'classes':['suit-tab suit-tab-details']})]
Scott Baker40c00762014-08-21 16:55:59 -07001053 readonly_fields = ('backend_status_text', )
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001054
Tony Mack93d1b032014-12-08 16:43:02 -05001055 user_readonly_fields = ['name','site_deployment']
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001056 user_readonly_inlines = [TagInline,SliverInline]
1057
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001058 suit_form_tabs =(('details','Node Details'),('slivers','Slivers'),('tags','Tags'))
Siobhan Tully4bc09f22013-04-10 21:15:21 -04001059
Siobhan Tully567e3e62013-06-21 18:03:16 -04001060
Tony Mackd90cdbf2013-04-16 22:48:40 -04001061class SliverForm(forms.ModelForm):
1062 class Meta:
Tony Mack1d6b85f2013-05-07 18:49:14 -04001063 model = Sliver
Tony Mackd90cdbf2013-04-16 22:48:40 -04001064 ip = forms.CharField(widget=PlainTextWidget)
Tony Mack18261812013-05-02 16:39:20 -04001065 instance_name = forms.CharField(widget=PlainTextWidget)
Tony Mackd90cdbf2013-04-16 22:48:40 -04001066 widgets = {
1067 'ip': PlainTextWidget(),
Tony Mack18261812013-05-02 16:39:20 -04001068 'instance_name': PlainTextWidget(),
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001069 'slice': LinkedSelect,
Tony Mack93d1b032014-12-08 16:43:02 -05001070 'controllerNetwork': LinkedSelect,
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001071 'node': LinkedSelect,
1072 'image': LinkedSelect
Siobhan Tully53437282013-04-26 19:30:27 -04001073 }
Tony Mackd90cdbf2013-04-16 22:48:40 -04001074
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001075class TagAdmin(PlanetStackBaseAdmin):
Scott Baker63d1a552014-08-21 15:19:07 -07001076 list_display = ['backend_status_icon', 'service', 'name', 'value', 'content_type', 'content_object',]
1077 list_display_links = list_display
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001078 user_readonly_fields = ['service', 'name', 'value', 'content_type', 'content_object',]
1079 user_readonly_inlines = []
Siobhan Tullyd3515752013-06-21 16:34:53 -04001080
Tony Mack9bcbe4f2013-04-29 08:13:27 -04001081class SliverAdmin(PlanetStackBaseAdmin):
Tony Mackd90cdbf2013-04-16 22:48:40 -04001082 form = SliverForm
Tony Mackcdec0902013-04-15 00:38:49 -04001083 fieldsets = [
Tony Mack93d1b032014-12-08 16:43:02 -05001084 ('Sliver Details', {'fields': ['backend_status_text', 'slice', 'controllerNetwork', 'node', 'ip', 'instance_name', 'flavor', 'image', ], 'classes': ['suit-tab suit-tab-general'], })
Tony Mackcdec0902013-04-15 00:38:49 -04001085 ]
Scott Baker40c00762014-08-21 16:55:59 -07001086 readonly_fields = ('backend_status_text', )
Tony Mack93d1b032014-12-08 16:43:02 -05001087 list_display = ['backend_status_icon', 'ip', 'instance_name', 'slice', 'flavor', 'image', 'node', 'controllerNetwork']
Scott Baker63d1a552014-08-21 15:19:07 -07001088 list_display_links = ('backend_status_icon', 'ip',)
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001089
1090 suit_form_tabs =(('general', 'Sliver Details'),
1091 ('tags','Tags'),
1092 )
1093
Siobhan Tullyde5450d2013-06-21 11:35:33 -04001094 inlines = [TagInline]
Tony Mack53106f32013-04-27 16:43:01 -04001095
Tony Mack93d1b032014-12-08 16:43:02 -05001096 user_readonly_fields = ['slice', 'controllerNetwork', 'node', 'ip', 'instance_name', 'flavor', 'image']
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001097
Tony Mackc2835a92013-05-28 09:18:49 -04001098 def formfield_for_foreignkey(self, db_field, request, **kwargs):
1099 if db_field.name == 'slice':
Tony Mack5b061472014-02-04 07:57:10 -05001100 kwargs['queryset'] = Slice.select_by_user(request.user)
Tony Mackc2835a92013-05-28 09:18:49 -04001101
1102 return super(SliverAdmin, self).formfield_for_foreignkey(db_field, request, **kwargs)
1103
Tony Mack04062832013-05-10 08:22:44 -04001104 def queryset(self, request):
Scott Baker36f50872014-08-21 13:01:25 -07001105 # admins can see all slivers. Users can only see slivers of
Tony Mack04062832013-05-10 08:22:44 -04001106 # the slices they belong to.
Tony Mack5b061472014-02-04 07:57:10 -05001107 return Sliver.select_by_user(request.user)
1108
Tony Mack04062832013-05-10 08:22:44 -04001109
Tony Mack1d6b85f2013-05-07 18:49:14 -04001110 def get_formsets(self, request, obj=None):
1111 # make some fields read only if we are updating an existing record
1112 if obj == None:
Scott Baker36f50872014-08-21 13:01:25 -07001113 #self.readonly_fields = ('ip', 'instance_name')
Scott Bakerfbb45862014-10-17 16:27:23 -07001114 self.readonly_fields = ('backend_status_text',)
Tony Mack1d6b85f2013-05-07 18:49:14 -04001115 else:
Scott Bakerfbb45862014-10-17 16:27:23 -07001116 self.readonly_fields = ('backend_status_text',)
Scott Baker36f50872014-08-21 13:01:25 -07001117 #self.readonly_fields = ('ip', 'instance_name', 'slice', 'image', 'key')
Tony Mack1d6b85f2013-05-07 18:49:14 -04001118
1119 for inline in self.get_inline_instances(request, obj):
1120 # hide MyInline in the add view
1121 if obj is None:
1122 continue
Scott Baker526b71e2014-05-13 13:18:01 -07001123 if isinstance(inline, SliverInline):
1124 inline.model.caller = request.user
1125 yield inline.get_formset(request, obj)
Tony Mack53106f32013-04-27 16:43:01 -04001126
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001127 #def save_model(self, request, obj, form, change):
1128 # # update openstack connection to use this site/tenant
1129 # auth = request.session.get('auth', {})
1130 # auth['tenant'] = obj.slice.name
1131 # obj.os_manager = OpenStackManager(auth=auth, caller=request.user)
1132 # obj.creator = request.user
1133 # obj.save()
Tony Mack53106f32013-04-27 16:43:01 -04001134
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001135 #def delete_model(self, request, obj):
1136 # # update openstack connection to use this site/tenant
1137 # auth = request.session.get('auth', {})
1138 # auth['tenant'] = obj.slice.name
1139 # obj.os_manager = OpenStackManager(auth=auth, caller=request.user)
1140 # obj.delete()
Tony Mackcdec0902013-04-15 00:38:49 -04001141
Siobhan Tully53437282013-04-26 19:30:27 -04001142class UserCreationForm(forms.ModelForm):
1143 """A form for creating new users. Includes all the required
1144 fields, plus a repeated password."""
1145 password1 = forms.CharField(label='Password', widget=forms.PasswordInput)
1146 password2 = forms.CharField(label='Password confirmation', widget=forms.PasswordInput)
1147
1148 class Meta:
Siobhan Tully30fd4292013-05-10 08:59:56 -04001149 model = User
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001150 fields = ('email', 'firstname', 'lastname', 'phone', 'public_key')
Siobhan Tully53437282013-04-26 19:30:27 -04001151
1152 def clean_password2(self):
1153 # Check that the two password entries match
1154 password1 = self.cleaned_data.get("password1")
1155 password2 = self.cleaned_data.get("password2")
1156 if password1 and password2 and password1 != password2:
1157 raise forms.ValidationError("Passwords don't match")
1158 return password2
1159
1160 def save(self, commit=True):
1161 # Save the provided password in hashed format
1162 user = super(UserCreationForm, self).save(commit=False)
Tony Mackf9f4afb2013-05-01 21:02:12 -04001163 user.password = self.cleaned_data["password1"]
1164 #user.set_password(self.cleaned_data["password1"])
Siobhan Tully53437282013-04-26 19:30:27 -04001165 if commit:
1166 user.save()
1167 return user
1168
Siobhan Tully567e3e62013-06-21 18:03:16 -04001169
Siobhan Tully53437282013-04-26 19:30:27 -04001170class UserChangeForm(forms.ModelForm):
1171 """A form for updating users. Includes all the fields on
1172 the user, but replaces the password field with admin's
1173 password hash display field.
1174 """
Siobhan Tully63b7ba42014-01-12 10:35:11 -05001175 password = ReadOnlyPasswordHashField(label='Password',
1176 help_text= '<a href=\"password/\">Change Password</a>.')
Siobhan Tully53437282013-04-26 19:30:27 -04001177
1178 class Meta:
Siobhan Tully30fd4292013-05-10 08:59:56 -04001179 model = User
Scott Baker9f6b8ed2014-11-17 23:44:03 -08001180 widgets = { 'public_key': UploadTextareaWidget, }
Siobhan Tully53437282013-04-26 19:30:27 -04001181
1182 def clean_password(self):
1183 # Regardless of what the user provides, return the initial value.
1184 # This is done here, rather than on the field, because the
1185 # field does not have access to the initial value
1186 return self.initial["password"]
1187
Scott Baker2c3cb642014-05-19 17:55:56 -07001188class UserDashboardViewInline(PlStackTabularInline):
1189 model = UserDashboardView
1190 extra = 0
1191 suit_classes = 'suit-tab suit-tab-dashboards'
1192 fields = ['user', 'dashboardView', 'order']
1193
Scott Baker86c83ab2014-10-03 13:10:47 -07001194class UserAdmin(PermissionCheckingAdminMixin, UserAdmin):
1195 # Note: Make sure PermissionCheckingAdminMixin is listed before
1196 # admin.ModelAdmin in the class declaration.
1197
Siobhan Tully53437282013-04-26 19:30:27 -04001198 class Meta:
1199 app_label = "core"
1200
1201 # The forms to add and change user instances
1202 form = UserChangeForm
1203 add_form = UserCreationForm
1204
1205 # The fields to be used in displaying the User model.
1206 # These override the definitions on the base UserAdmin
1207 # that reference specific fields on auth.User.
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001208 list_display = ('email', 'firstname', 'lastname', 'site', 'last_login')
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001209 list_filter = ('site',)
Tony Mack336e0f92014-11-30 15:53:08 -05001210 inlines = [SlicePrivilegeInline,SitePrivilegeInline,ControllerPrivilegeInline,UserDashboardViewInline]
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001211
Scott Bakercbfb6002014-10-03 00:32:37 -07001212 fieldListLoginDetails = ['backend_status_text', 'email','site','password','is_active','is_readonly','is_admin','public_key']
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001213 fieldListContactInfo = ['firstname','lastname','phone','timezone']
1214
Siobhan Tully53437282013-04-26 19:30:27 -04001215 fieldsets = (
Scott Baker40c00762014-08-21 16:55:59 -07001216 ('Login Details', {'fields': ['backend_status_text', 'email', 'site','password', 'is_active', 'is_readonly', 'is_admin', 'public_key'], 'classes':['suit-tab suit-tab-general']}),
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001217 ('Contact Information', {'fields': ('firstname','lastname','phone', 'timezone'), 'classes':['suit-tab suit-tab-contact']}),
Scott Baker2c3cb642014-05-19 17:55:56 -07001218 #('Dashboard Views', {'fields': ('dashboards',), 'classes':['suit-tab suit-tab-dashboards']}),
Siobhan Tully53437282013-04-26 19:30:27 -04001219 #('Important dates', {'fields': ('last_login',)}),
1220 )
1221 add_fieldsets = (
1222 (None, {
1223 'classes': ('wide',),
Scott Baker0a5633b2014-10-06 17:51:20 -07001224 'fields': ('email', 'firstname', 'lastname', 'is_readonly', 'phone', 'public_key','password1', 'password2')},
Siobhan Tully53437282013-04-26 19:30:27 -04001225 ),
1226 )
Scott Baker40c00762014-08-21 16:55:59 -07001227 readonly_fields = ('backend_status_text', )
Siobhan Tully53437282013-04-26 19:30:27 -04001228 search_fields = ('email',)
1229 ordering = ('email',)
1230 filter_horizontal = ()
1231
Scott Baker3ca51f62014-05-23 12:05:11 -07001232 user_readonly_fields = fieldListLoginDetails + fieldListContactInfo
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001233
Scott Baker0a5633b2014-10-06 17:51:20 -07001234 @property
1235 def suit_form_tabs(self):
1236 if getattr(_thread_locals, "obj", None) is None:
1237 return []
1238 else:
1239 return (('general','Login Details'),
1240 ('contact','Contact Information'),
1241 ('sliceprivileges','Slice Privileges'),
1242 ('siteprivileges','Site Privileges'),
Tony Mack336e0f92014-11-30 15:53:08 -05001243 ('controllerprivileges','Controller Privileges'),
Scott Baker0a5633b2014-10-06 17:51:20 -07001244 ('dashboards','Dashboard Views'))
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001245
Tony Mackc2835a92013-05-28 09:18:49 -04001246 def formfield_for_foreignkey(self, db_field, request, **kwargs):
1247 if db_field.name == 'site':
Tony Mack5b061472014-02-04 07:57:10 -05001248 kwargs['queryset'] = Site.select_by_user(request.user)
Tony Mackc2835a92013-05-28 09:18:49 -04001249
1250 return super(UserAdmin, self).formfield_for_foreignkey(db_field, request, **kwargs)
1251
Tony Mack5b061472014-02-04 07:57:10 -05001252 def queryset(self, request):
1253 return User.select_by_user(request.user)
1254
Scott Baker2c3cb642014-05-19 17:55:56 -07001255class DashboardViewAdmin(PlanetStackBaseAdmin):
1256 fieldsets = [('Dashboard View Details',
Scott Baker40c00762014-08-21 16:55:59 -07001257 {'fields': ['backend_status_text', 'name', 'url'],
Scott Baker2c3cb642014-05-19 17:55:56 -07001258 'classes': ['suit-tab suit-tab-general']})
1259 ]
Scott Baker40c00762014-08-21 16:55:59 -07001260 readonly_fields = ('backend_status_text', )
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001261
Scott Baker2c3cb642014-05-19 17:55:56 -07001262 suit_form_tabs =(('general','Dashboard View Details'),)
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001263
Scott Baker0165fac2014-01-13 11:49:26 -08001264class ServiceResourceInline(PlStackTabularInline):
Scott Baker3de3e372013-05-10 16:50:44 -07001265 model = ServiceResource
1266 extra = 0
1267
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001268class ServiceClassAdmin(PlanetStackBaseAdmin):
Scott Baker63d1a552014-08-21 15:19:07 -07001269 list_display = ('backend_status_icon', 'name', 'commitment', 'membershipFee')
1270 list_display_links = ('backend_status_icon', 'name', )
Scott Baker3de3e372013-05-10 16:50:44 -07001271 inlines = [ServiceResourceInline]
1272
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001273 user_readonly_fields = ['name', 'commitment', 'membershipFee']
1274 user_readonly_inlines = []
1275
Scott Baker0165fac2014-01-13 11:49:26 -08001276class ReservedResourceInline(PlStackTabularInline):
Scott Baker133c9212013-05-17 09:09:11 -07001277 model = ReservedResource
1278 extra = 0
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001279 suit_classes = 'suit-tab suit-tab-reservedresources'
Scott Baker133c9212013-05-17 09:09:11 -07001280
1281 def formfield_for_foreignkey(self, db_field, request=None, **kwargs):
1282 field = super(ReservedResourceInline, self).formfield_for_foreignkey(db_field, request, **kwargs)
1283
1284 if db_field.name == 'resource':
1285 # restrict resources to those that the slice's service class allows
1286 if request._slice is not None:
1287 field.queryset = field.queryset.filter(serviceClass = request._slice.serviceClass, calendarReservable=True)
1288 if len(field.queryset) > 0:
1289 field.initial = field.queryset.all()[0]
1290 else:
1291 field.queryset = field.queryset.none()
1292 elif db_field.name == 'sliver':
1293 # restrict slivers to those that belong to the slice
1294 if request._slice is not None:
1295 field.queryset = field.queryset.filter(slice = request._slice)
1296 else:
1297 field.queryset = field.queryset.none()
1298
1299 return field
1300
Tony Mack5b061472014-02-04 07:57:10 -05001301 def queryset(self, request):
1302 return ReservedResource.select_by_user(request.user)
1303
Scott Baker133c9212013-05-17 09:09:11 -07001304class ReservationChangeForm(forms.ModelForm):
1305 class Meta:
1306 model = Reservation
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001307 widgets = {
1308 'slice' : LinkedSelect
1309 }
Scott Baker133c9212013-05-17 09:09:11 -07001310
1311class ReservationAddForm(forms.ModelForm):
1312 slice = forms.ModelChoiceField(queryset=Slice.objects.all(), widget=forms.Select(attrs={"onChange":"document.getElementById('id_refresh').value=1; submit()"}))
1313 refresh = forms.CharField(widget=forms.HiddenInput())
1314
1315 class Media:
1316 css = {'all': ('planetstack.css',)} # .field-refresh { display: none; }
1317
1318 def clean_slice(self):
1319 slice = self.cleaned_data.get("slice")
1320 x = ServiceResource.objects.filter(serviceClass = slice.serviceClass, calendarReservable=True)
1321 if len(x) == 0:
1322 raise forms.ValidationError("The slice you selected does not have a service class that allows reservations")
1323 return slice
1324
1325 class Meta:
1326 model = Reservation
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001327 widgets = {
1328 'slice' : LinkedSelect
1329 }
1330
Scott Baker133c9212013-05-17 09:09:11 -07001331
1332class ReservationAddRefreshForm(ReservationAddForm):
1333 """ This form is displayed when the Reservation Form receives an update
1334 from the Slice dropdown onChange handler. It doesn't validate the
1335 data and doesn't save the data. This will cause the form to be
1336 redrawn.
1337 """
1338
Scott Baker8737e5f2013-05-17 09:35:32 -07001339 """ don't validate anything other than slice """
1340 dont_validate_fields = ("startTime", "duration")
1341
Scott Baker133c9212013-05-17 09:09:11 -07001342 def full_clean(self):
1343 result = super(ReservationAddForm, self).full_clean()
Scott Baker8737e5f2013-05-17 09:35:32 -07001344
1345 for fieldname in self.dont_validate_fields:
1346 if fieldname in self._errors:
1347 del self._errors[fieldname]
1348
Scott Baker133c9212013-05-17 09:09:11 -07001349 return result
1350
1351 """ don't save anything """
1352 def is_valid(self):
1353 return False
1354
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001355class ReservationAdmin(PlanetStackBaseAdmin):
Scott Baker40c00762014-08-21 16:55:59 -07001356 fieldList = ['backend_status_text', 'slice', 'startTime', 'duration']
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001357 fieldsets = [('Reservation Details', {'fields': fieldList, 'classes': ['suit-tab suit-tab-general']})]
Scott Baker40c00762014-08-21 16:55:59 -07001358 readonly_fields = ('backend_status_text', )
Scott Baker133c9212013-05-17 09:09:11 -07001359 list_display = ('startTime', 'duration')
Scott Baker133c9212013-05-17 09:09:11 -07001360 form = ReservationAddForm
1361
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001362 suit_form_tabs = (('general','Reservation Details'), ('reservedresources','Reserved Resources'))
1363
1364 inlines = [ReservedResourceInline]
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001365 user_readonly_fields = fieldList
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001366
Scott Baker133c9212013-05-17 09:09:11 -07001367 def add_view(self, request, form_url='', extra_context=None):
Scott Bakeracd45142013-05-19 16:19:16 -07001368 timezone.activate(request.user.timezone)
Scott Baker133c9212013-05-17 09:09:11 -07001369 request._refresh = False
1370 request._slice = None
1371 if request.method == 'POST':
Scott Baker8737e5f2013-05-17 09:35:32 -07001372 # "refresh" will be set to "1" if the form was submitted due to
1373 # a change in the Slice dropdown.
Scott Baker133c9212013-05-17 09:09:11 -07001374 if request.POST.get("refresh","1") == "1":
1375 request._refresh = True
1376 request.POST["refresh"] = "0"
Scott Baker8737e5f2013-05-17 09:35:32 -07001377
1378 # Keep track of the slice that was selected, so the
1379 # reservedResource inline can filter items for the slice.
Scott Baker133c9212013-05-17 09:09:11 -07001380 request._slice = request.POST.get("slice",None)
1381 if (request._slice is not None):
1382 request._slice = Slice.objects.get(id=request._slice)
1383
1384 result = super(ReservationAdmin, self).add_view(request, form_url, extra_context)
1385 return result
1386
Scott Bakeracd45142013-05-19 16:19:16 -07001387 def changelist_view(self, request, extra_context = None):
1388 timezone.activate(request.user.timezone)
1389 return super(ReservationAdmin, self).changelist_view(request, extra_context)
1390
Scott Baker133c9212013-05-17 09:09:11 -07001391 def get_form(self, request, obj=None, **kwargs):
Siobhan Tullyd3515752013-06-21 16:34:53 -04001392 request._obj_ = obj
1393 if obj is not None:
1394 # For changes, set request._slice to the slice already set in the
1395 # object.
1396 request._slice = obj.slice
1397 self.form = ReservationChangeForm
1398 else:
1399 if getattr(request, "_refresh", False):
1400 self.form = ReservationAddRefreshForm
1401 else:
1402 self.form = ReservationAddForm
1403 return super(ReservationAdmin, self).get_form(request, obj, **kwargs)
1404
Scott Baker133c9212013-05-17 09:09:11 -07001405 def get_readonly_fields(self, request, obj=None):
Siobhan Tullyd3515752013-06-21 16:34:53 -04001406 if (obj is not None):
1407 # Prevent slice from being changed after the reservation has been
1408 # created.
1409 return ['slice']
1410 else:
Scott Baker133c9212013-05-17 09:09:11 -07001411 return []
Scott Baker3de3e372013-05-10 16:50:44 -07001412
Tony Mack5b061472014-02-04 07:57:10 -05001413 def queryset(self, request):
1414 return Reservation.select_by_user(request.user)
1415
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001416class NetworkParameterTypeAdmin(PlanetStackBaseAdmin):
Scott Baker63d1a552014-08-21 15:19:07 -07001417 list_display = ("backend_status_icon", "name", )
1418 list_display_links = ('backend_status_icon', 'name', )
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001419 user_readonly_fields = ['name']
1420 user_readonly_inlines = []
Scott Baker74d8e622013-07-29 16:04:22 -07001421
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001422class RouterAdmin(PlanetStackBaseAdmin):
Scott Baker63d1a552014-08-21 15:19:07 -07001423 list_display = ("backend_status_icon", "name", )
1424 list_display_links = ('backend_status_icon', 'name', )
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001425 user_readonly_fields = ['name']
1426 user_readonly_inlines = []
1427
Scott Baker0165fac2014-01-13 11:49:26 -08001428class RouterInline(PlStackTabularInline):
Scott Baker74d8e622013-07-29 16:04:22 -07001429 model = Router.networks.through
1430 extra = 0
1431 verbose_name_plural = "Routers"
1432 verbose_name = "Router"
Siobhan Tully2d95e482013-09-06 10:56:06 -04001433 suit_classes = 'suit-tab suit-tab-routers'
Scott Baker74d8e622013-07-29 16:04:22 -07001434
Scott Bakerb27b62c2014-08-15 16:29:16 -07001435class NetworkParameterInline(PlStackGenericTabularInline):
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001436 model = NetworkParameter
Scott Baker618e3792014-08-15 13:42:29 -07001437 extra = 0
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001438 verbose_name_plural = "Parameters"
1439 verbose_name = "Parameter"
1440 suit_classes = 'suit-tab suit-tab-netparams'
Scott Baker40c00762014-08-21 16:55:59 -07001441 fields = ['backend_status_icon', 'parameter', 'value']
1442 readonly_fields = ('backend_status_icon', )
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001443
Scott Baker0165fac2014-01-13 11:49:26 -08001444class NetworkSliversInline(PlStackTabularInline):
Scott Baker40c00762014-08-21 16:55:59 -07001445 fields = ['backend_status_icon', 'network','sliver','ip']
1446 readonly_fields = ("backend_status_icon", "ip", )
Scott Baker74d8e622013-07-29 16:04:22 -07001447 model = NetworkSliver
Scott Baker874936e2014-01-13 18:15:34 -08001448 selflink_fieldname = "sliver"
Scott Baker74d8e622013-07-29 16:04:22 -07001449 extra = 0
1450 verbose_name_plural = "Slivers"
1451 verbose_name = "Sliver"
Siobhan Tully2d95e482013-09-06 10:56:06 -04001452 suit_classes = 'suit-tab suit-tab-networkslivers'
Scott Baker74d8e622013-07-29 16:04:22 -07001453
Scott Baker0165fac2014-01-13 11:49:26 -08001454class NetworkSlicesInline(PlStackTabularInline):
Scott Bakerd7d2a392013-08-06 08:57:30 -07001455 model = NetworkSlice
Scott Baker874936e2014-01-13 18:15:34 -08001456 selflink_fieldname = "slice"
Scott Bakerd7d2a392013-08-06 08:57:30 -07001457 extra = 0
1458 verbose_name_plural = "Slices"
1459 verbose_name = "Slice"
Siobhan Tully2d95e482013-09-06 10:56:06 -04001460 suit_classes = 'suit-tab suit-tab-networkslices'
Scott Baker40c00762014-08-21 16:55:59 -07001461 fields = ['backend_status_icon', 'network','slice']
1462 readonly_fields = ('backend_status_icon', )
Scott Bakerd7d2a392013-08-06 08:57:30 -07001463
Tony Mack336e0f92014-11-30 15:53:08 -05001464class ControllerNetworksInline(PlStackTabularInline):
1465 model = ControllerNetworks
Scott Bakerfbb45862014-10-17 16:27:23 -07001466 extra = 0
Tony Mack336e0f92014-11-30 15:53:08 -05001467 verbose_name_plural = "Controller Networks"
1468 verbose_name = "Controller Network"
Scott Bakerfbb45862014-10-17 16:27:23 -07001469 suit_classes = 'suit-tab suit-tab-admin-only'
Tony Mack336e0f92014-11-30 15:53:08 -05001470 fields = ['backend_status_icon', 'controller','net_id','subnet_id']
Scott Bakerfbb45862014-10-17 16:27:23 -07001471 readonly_fields = ('backend_status_icon', )
1472
Scott Baker9f6b8ed2014-11-17 23:44:03 -08001473class NetworkForm(forms.ModelForm):
1474 class Meta:
1475 model = Network
1476 widgets = {
1477 'topologyParameters': UploadTextareaWidget,
1478 'controllerParameters': UploadTextareaWidget,
1479 }
1480
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001481class NetworkAdmin(PlanetStackBaseAdmin):
Scott Baker63d1a552014-08-21 15:19:07 -07001482 list_display = ("backend_status_icon", "name", "subnet", "ports", "labels")
1483 list_display_links = ('backend_status_icon', 'name', )
Scott Baker74d8e622013-07-29 16:04:22 -07001484 readonly_fields = ("subnet", )
Siobhan Tully2d95e482013-09-06 10:56:06 -04001485
Scott Bakerd7d2a392013-08-06 08:57:30 -07001486 inlines = [NetworkParameterInline, NetworkSliversInline, NetworkSlicesInline, RouterInline]
Tony Mack336e0f92014-11-30 15:53:08 -05001487 admin_inlines = [ControllerNetworksInline]
Scott Baker74d8e622013-07-29 16:04:22 -07001488
Scott Baker9f6b8ed2014-11-17 23:44:03 -08001489 form=NetworkForm
1490
Siobhan Tully2d95e482013-09-06 10:56:06 -04001491 fieldsets = [
Scott Baker40248712014-11-17 16:04:45 -08001492 (None, {'fields': ['backend_status_text', 'name','template','ports','labels','owner','guaranteedBandwidth', 'permitAllSlices','permittedSlices','network_id','router_id','subnet_id','subnet'],
1493 'classes':['suit-tab suit-tab-general']}),
Scott Baker7b6722e2014-11-17 16:18:00 -08001494 (None, {'fields': ['topologyParameters', 'controllerUrl', 'controllerParameters'],
Scott Baker40248712014-11-17 16:04:45 -08001495 'classes':['suit-tab suit-tab-sdn']}),
1496 ]
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001497
Scott Baker40c00762014-08-21 16:55:59 -07001498 readonly_fields = ('backend_status_text', )
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001499 user_readonly_fields = ['name','template','ports','labels','owner','guaranteedBandwidth', 'permitAllSlices','permittedSlices','network_id','router_id','subnet_id','subnet']
Siobhan Tully2d95e482013-09-06 10:56:06 -04001500
Scott Bakerfbb45862014-10-17 16:27:23 -07001501 @property
1502 def suit_form_tabs(self):
1503 tabs=[('general','Network Details'),
Scott Baker40248712014-11-17 16:04:45 -08001504 ('sdn', 'SDN Configuration'),
Scott Bakerfbb45862014-10-17 16:27:23 -07001505 ('netparams', 'Parameters'),
1506 ('networkslivers','Slivers'),
1507 ('networkslices','Slices'),
1508 ('routers','Routers'),
1509 ]
1510
1511 request=getattr(_thread_locals, "request", None)
1512 if request and request.user.is_admin:
1513 tabs.append( ('admin-only', 'Admin-Only') )
1514
1515 return tabs
1516
1517
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001518class NetworkTemplateAdmin(PlanetStackBaseAdmin):
Scott Baker63d1a552014-08-21 15:19:07 -07001519 list_display = ("backend_status_icon", "name", "guaranteedBandwidth", "visibility")
1520 list_display_links = ('backend_status_icon', 'name', )
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001521 user_readonly_fields = ["name", "guaranteedBandwidth", "visibility"]
1522 user_readonly_inlines = []
Scott Baker40248712014-11-17 16:04:45 -08001523 fieldsets = [
1524 (None, {'fields': ['name', 'description', 'guaranteedBandwidth', 'visibility', 'translation', 'sharedNetworkName', 'sharedNetworkId', 'topologyKind', 'controllerKind'],
1525 'classes':['suit-tab suit-tab-general']}),]
1526 suit_form_tabs = (('general','Network Template Details'), )
Scott Baker74d8e622013-07-29 16:04:22 -07001527
Scott Baker37b47902014-09-02 14:37:41 -07001528class FlavorAdmin(PlanetStackBaseAdmin):
1529 list_display = ("backend_status_icon", "name", "flavor", "order", "default")
1530 list_display_links = ("backend_status_icon", "name")
1531 user_readonly_fields = ("name", "flavor")
1532 fields = ("name", "description", "flavor", "order", "default")
1533
Tony Mack31c2b8f2013-04-26 20:01:42 -04001534# register a signal that caches the user's credentials when they log in
1535def cache_credentials(sender, user, request, **kwds):
1536 auth = {'username': request.POST['username'],
1537 'password': request.POST['password']}
1538 request.session['auth'] = auth
1539user_logged_in.connect(cache_credentials)
1540
Scott Baker15cddfa2013-12-09 13:45:19 -08001541def dollar_field(fieldName, short_description):
1542 def newFunc(self, obj):
1543 try:
1544 x= "$ %0.2f" % float(getattr(obj, fieldName, 0.0))
1545 except:
1546 x=getattr(obj, fieldName, 0.0)
1547 return x
1548 newFunc.short_description = short_description
1549 return newFunc
1550
1551def right_dollar_field(fieldName, short_description):
1552 def newFunc(self, obj):
1553 try:
1554 #x= '<div align=right style="width:6em">$ %0.2f</div>' % float(getattr(obj, fieldName, 0.0))
1555 x= '<div align=right>$ %0.2f</div>' % float(getattr(obj, fieldName, 0.0))
1556 except:
1557 x=getattr(obj, fieldName, 0.0)
1558 return x
1559 newFunc.short_description = short_description
1560 newFunc.allow_tags = True
1561 return newFunc
Scott Baker43105042013-12-06 23:23:36 -08001562
Scott Baker0165fac2014-01-13 11:49:26 -08001563class InvoiceChargeInline(PlStackTabularInline):
Scott Baker43105042013-12-06 23:23:36 -08001564 model = Charge
Scott Baker15cddfa2013-12-09 13:45:19 -08001565 extra = 0
Scott Baker43105042013-12-06 23:23:36 -08001566 verbose_name_plural = "Charges"
1567 verbose_name = "Charge"
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001568 exclude = ['account']
Scott Baker9cb88a22013-12-09 18:56:00 -08001569 fields = ["date", "kind", "state", "object", "coreHours", "dollar_amount", "slice"]
1570 readonly_fields = ["date", "kind", "state", "object", "coreHours", "dollar_amount", "slice"]
1571 can_delete = False
1572 max_num = 0
1573
1574 dollar_amount = right_dollar_field("amount", "Amount")
Scott Baker43105042013-12-06 23:23:36 -08001575
1576class InvoiceAdmin(admin.ModelAdmin):
1577 list_display = ("date", "account")
1578
1579 inlines = [InvoiceChargeInline]
1580
Scott Baker9cb88a22013-12-09 18:56:00 -08001581 fields = ["date", "account", "dollar_amount"]
1582 readonly_fields = ["date", "account", "dollar_amount"]
1583
1584 dollar_amount = dollar_field("amount", "Amount")
Scott Baker43105042013-12-06 23:23:36 -08001585
Scott Baker0165fac2014-01-13 11:49:26 -08001586class InvoiceInline(PlStackTabularInline):
Scott Baker15cddfa2013-12-09 13:45:19 -08001587 model = Invoice
1588 extra = 0
1589 verbose_name_plural = "Invoices"
1590 verbose_name = "Invoice"
Scott Baker0165fac2014-01-13 11:49:26 -08001591 fields = ["date", "dollar_amount"]
1592 readonly_fields = ["date", "dollar_amount"]
Scott Baker15cddfa2013-12-09 13:45:19 -08001593 suit_classes = 'suit-tab suit-tab-accountinvoice'
1594 can_delete=False
1595 max_num=0
1596
1597 dollar_amount = right_dollar_field("amount", "Amount")
1598
Scott Baker0165fac2014-01-13 11:49:26 -08001599class PendingChargeInline(PlStackTabularInline):
Scott Baker43105042013-12-06 23:23:36 -08001600 model = Charge
Scott Baker15cddfa2013-12-09 13:45:19 -08001601 extra = 0
Scott Baker43105042013-12-06 23:23:36 -08001602 verbose_name_plural = "Charges"
1603 verbose_name = "Charge"
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001604 exclude = ["invoice"]
Scott Baker15cddfa2013-12-09 13:45:19 -08001605 fields = ["date", "kind", "state", "object", "coreHours", "dollar_amount", "slice"]
1606 readonly_fields = ["date", "kind", "state", "object", "coreHours", "dollar_amount", "slice"]
Scott Baker43105042013-12-06 23:23:36 -08001607 suit_classes = 'suit-tab suit-tab-accountpendingcharges'
Scott Baker15cddfa2013-12-09 13:45:19 -08001608 can_delete=False
1609 max_num=0
Scott Baker43105042013-12-06 23:23:36 -08001610
1611 def queryset(self, request):
1612 qs = super(PendingChargeInline, self).queryset(request)
1613 qs = qs.filter(state="pending")
1614 return qs
1615
Scott Baker15cddfa2013-12-09 13:45:19 -08001616 dollar_amount = right_dollar_field("amount", "Amount")
1617
Scott Baker0165fac2014-01-13 11:49:26 -08001618class PaymentInline(PlStackTabularInline):
Scott Baker43105042013-12-06 23:23:36 -08001619 model=Payment
1620 extra = 1
1621 verbose_name_plural = "Payments"
1622 verbose_name = "Payment"
Scott Baker15cddfa2013-12-09 13:45:19 -08001623 fields = ["date", "dollar_amount"]
1624 readonly_fields = ["date", "dollar_amount"]
Scott Baker43105042013-12-06 23:23:36 -08001625 suit_classes = 'suit-tab suit-tab-accountpayments'
Scott Baker15cddfa2013-12-09 13:45:19 -08001626 can_delete=False
1627 max_num=0
1628
1629 dollar_amount = right_dollar_field("amount", "Amount")
1630
Scott Baker43105042013-12-06 23:23:36 -08001631class AccountAdmin(admin.ModelAdmin):
1632 list_display = ("site", "balance_due")
1633
1634 inlines = [InvoiceInline, PaymentInline, PendingChargeInline]
1635
1636 fieldsets = [
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001637 (None, {'fields': ['site', 'dollar_balance_due', 'dollar_total_invoices', 'dollar_total_payments'],'classes':['suit-tab suit-tab-general']}),]
Scott Baker43105042013-12-06 23:23:36 -08001638
Scott Baker15cddfa2013-12-09 13:45:19 -08001639 readonly_fields = ['site', 'dollar_balance_due', 'dollar_total_invoices', 'dollar_total_payments']
Scott Baker43105042013-12-06 23:23:36 -08001640
1641 suit_form_tabs =(
1642 ('general','Account Details'),
1643 ('accountinvoice', 'Invoices'),
1644 ('accountpayments', 'Payments'),
1645 ('accountpendingcharges','Pending Charges'),
1646 )
1647
Scott Baker15cddfa2013-12-09 13:45:19 -08001648 dollar_balance_due = dollar_field("balance_due", "Balance Due")
1649 dollar_total_invoices = dollar_field("total_invoices", "Total Invoices")
1650 dollar_total_payments = dollar_field("total_payments", "Total Payments")
1651
Siobhan Tully53437282013-04-26 19:30:27 -04001652# Now register the new UserAdmin...
Siobhan Tully30fd4292013-05-10 08:59:56 -04001653admin.site.register(User, UserAdmin)
Siobhan Tully53437282013-04-26 19:30:27 -04001654# ... and, since we're not using Django's builtin permissions,
1655# unregister the Group model from admin.
Siobhan Tullyce652d02013-10-08 21:52:35 -04001656#admin.site.unregister(Group)
Siobhan Tully53437282013-04-26 19:30:27 -04001657
Siobhan Tullybf1153a2013-05-27 20:53:48 -04001658#Do not show django evolution in the admin interface
1659from django_evolution.models import Version, Evolution
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001660#admin.site.unregister(Version)
1661#admin.site.unregister(Evolution)
Siobhan Tullybf1153a2013-05-27 20:53:48 -04001662
1663
1664# When debugging it is often easier to see all the classes, but for regular use
1665# only the top-levels should be displayed
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001666showAll = False
Scott Baker43105042013-12-06 23:23:36 -08001667
Siobhan Tullybf1153a2013-05-27 20:53:48 -04001668admin.site.register(Deployment, DeploymentAdmin)
Tony Mack336e0f92014-11-30 15:53:08 -05001669admin.site.register(Controller, ControllerAdmin)
Siobhan Tully4bc09f22013-04-10 21:15:21 -04001670admin.site.register(Site, SiteAdmin)
Siobhan Tully4bc09f22013-04-10 21:15:21 -04001671admin.site.register(Slice, SliceAdmin)
Siobhan Tullyce652d02013-10-08 21:52:35 -04001672admin.site.register(Service, ServiceAdmin)
smbakera3cf70c2013-06-27 02:01:41 -07001673admin.site.register(Reservation, ReservationAdmin)
Scott Baker74d8e622013-07-29 16:04:22 -07001674admin.site.register(Network, NetworkAdmin)
1675admin.site.register(Router, RouterAdmin)
Scott Baker74d8e622013-07-29 16:04:22 -07001676admin.site.register(NetworkTemplate, NetworkTemplateAdmin)
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001677admin.site.register(Account, AccountAdmin)
1678admin.site.register(Invoice, InvoiceAdmin)
Siobhan Tullybf1153a2013-05-27 20:53:48 -04001679
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001680if True:
1681 admin.site.register(NetworkParameterType, NetworkParameterTypeAdmin)
1682 admin.site.register(ServiceClass, ServiceClassAdmin)
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001683 #admin.site.register(PlanetStack)
Siobhan Tullyd3515752013-06-21 16:34:53 -04001684 admin.site.register(Tag, TagAdmin)
Tony Mack336e0f92014-11-30 15:53:08 -05001685 admin.site.register(ControllerRole)
Siobhan Tullyce652d02013-10-08 21:52:35 -04001686 admin.site.register(SiteRole)
1687 admin.site.register(SliceRole)
1688 admin.site.register(PlanetStackRole)
Siobhan Tullybf1153a2013-05-27 20:53:48 -04001689 admin.site.register(Node, NodeAdmin)
Siobhan Tullybfd11dc2013-09-03 12:59:24 -04001690 #admin.site.register(SlicePrivilege, SlicePrivilegeAdmin)
1691 #admin.site.register(SitePrivilege, SitePrivilegeAdmin)
Siobhan Tullybf1153a2013-05-27 20:53:48 -04001692 admin.site.register(Sliver, SliverAdmin)
Siobhan Tullybf1153a2013-05-27 20:53:48 -04001693 admin.site.register(Image, ImageAdmin)
Scott Baker2c3cb642014-05-19 17:55:56 -07001694 admin.site.register(DashboardView, DashboardViewAdmin)
Scott Baker37b47902014-09-02 14:37:41 -07001695 admin.site.register(Flavor, FlavorAdmin)
Tony Mack7130ac32013-03-22 21:58:00 -04001696