blob: a27199f2de11df608688b67bf4a5e03a917deec7 [file] [log] [blame]
Scott Bakerf57e5592015-04-14 17:18:51 -07001import json
Jeremy Mowery11ed8992016-04-17 20:56:54 -07002from operator import attrgetter
Siobhan Tully00353f72013-10-08 21:53:27 -04003
Scott Baker80d65b12016-05-26 20:20:47 -07004from core.models import PlCoreBase, PlCoreBaseManager, SingletonModel, XOS
Jeremy Mowery11ed8992016-04-17 20:56:54 -07005from core.models.plcorebase import StrippedCharField
6from django.db import models
7from xos.exceptions import *
Scott Bakerb0eb23e2016-06-01 16:08:04 -07008import urlparse
Jeremy Mowery11ed8992016-04-17 20:56:54 -07009
10COARSE_KIND = "coarse"
11
Scott Baker80d65b12016-05-26 20:20:47 -070012def get_xos():
13 xos = XOS.objects.all()
14
15 if xos:
16 return xos[0]
17 else:
18 return None
19
Scott Baker2461bec2015-08-14 09:10:11 -070020
Scott Baker82498c52015-07-13 13:07:27 -070021class AttributeMixin(object):
Jeremy Mowery11ed8992016-04-17 20:56:54 -070022 # helper for extracting things from a json-encoded
23 # service_specific_attribute
24
Scott Baker82498c52015-07-13 13:07:27 -070025 def get_attribute(self, name, default=None):
26 if self.service_specific_attribute:
27 attributes = json.loads(self.service_specific_attribute)
28 else:
29 attributes = {}
30 return attributes.get(name, default)
31
32 def set_attribute(self, name, value):
33 if self.service_specific_attribute:
34 attributes = json.loads(self.service_specific_attribute)
35 else:
36 attributes = {}
Jeremy Mowery11ed8992016-04-17 20:56:54 -070037 attributes[name] = value
Scott Baker82498c52015-07-13 13:07:27 -070038 self.service_specific_attribute = json.dumps(attributes)
39
40 def get_initial_attribute(self, name, default=None):
41 if self._initial["service_specific_attribute"]:
Jeremy Mowery11ed8992016-04-17 20:56:54 -070042 attributes = json.loads(
43 self._initial["service_specific_attribute"])
Scott Baker82498c52015-07-13 13:07:27 -070044 else:
45 attributes = {}
46 return attributes.get(name, default)
47
Scott Baker74404fe2015-07-13 13:54:06 -070048 @classmethod
Scott Baker6dfde212016-03-01 20:10:24 -080049 def get_default_attribute(cls, name):
50 for (attrname, default) in cls.simple_attributes:
Jeremy Mowery11ed8992016-04-17 20:56:54 -070051 if attrname == name:
Scott Baker6dfde212016-03-01 20:10:24 -080052 return default
Jeremy Mowery11ed8992016-04-17 20:56:54 -070053 if hasattr(cls, "default_attributes"):
Scott Bakerf180ff32016-03-31 11:46:02 -070054 if name in cls.default_attributes:
55 return cls.default_attributes[name]
56
57 return None
Scott Baker6dfde212016-03-01 20:10:24 -080058
59 @classmethod
Scott Baker74404fe2015-07-13 13:54:06 -070060 def setup_simple_attributes(cls):
61 for (attrname, default) in cls.simple_attributes:
Scott Baker096dce82015-07-13 14:27:51 -070062 setattr(cls, attrname, property(lambda self, attrname=attrname, default=default: self.get_attribute(attrname, default),
Jeremy Mowery11ed8992016-04-17 20:56:54 -070063 lambda self, value, attrname=attrname: self.set_attribute(
64 attrname, value),
Scott Baker096dce82015-07-13 14:27:51 -070065 None,
66 attrname))
Scott Baker74404fe2015-07-13 13:54:06 -070067
Scott Baker80d65b12016-05-26 20:20:47 -070068class ServiceController(PlCoreBase):
Scott Bakerb0eb23e2016-06-01 16:08:04 -070069 xos = models.ForeignKey(XOS, related_name='service_controllers', help_text="Pointer to XOS", default=get_xos)
Scott Baker80d65b12016-05-26 20:20:47 -070070 name = StrippedCharField(max_length=30, help_text="Service Name")
Scott Baker1e559822016-05-26 20:57:10 -070071 base_url = StrippedCharField(max_length=1024, help_text="Base URL, allows use of relative URLs for resources", null=True, blank=True)
Scott Baker80d65b12016-05-26 20:20:47 -070072
Scott Baker147cace2016-05-26 21:11:37 -070073 def __unicode__(self): return u'%s' % (self.name)
74
Scott Baker80d65b12016-05-26 20:20:47 -070075class ServiceControllerResource(PlCoreBase):
76 KIND_CHOICES = (('models', 'Models'),
77 ('admin', 'Admin'),
78 ('django_library', 'Django Library'),
Scott Baker1ea8cb52016-05-26 22:19:35 -070079 ('synchronizer', 'Synchronizer'),
Scott Baker80d65b12016-05-26 20:20:47 -070080 ('rest', 'REST API'),
81 ('tosca_custom_types', 'Tosca Custom Types'),
82 ('tosca_resource', 'Tosca Resource'))
83
84 FORMAT_CHOICES = (('python', 'Python'),
85 ('manifest', 'Manifest'),
Scott Baker411c4e92016-05-26 22:50:42 -070086 ('docker', 'Docker Container'),
87 ('yaml', 'YAML'))
Scott Baker80d65b12016-05-26 20:20:47 -070088
Scott Bakerb0eb23e2016-06-01 16:08:04 -070089 service_controller = models.ForeignKey(ServiceController, related_name='service_controller_resources',
Scott Baker80d65b12016-05-26 20:20:47 -070090 help_text="The Service Controller this resource is associated with")
91
92 name = StrippedCharField(max_length=30, help_text="Object Name")
93 kind = StrippedCharField(choices=KIND_CHOICES, max_length=30)
94 format = StrippedCharField(choices=FORMAT_CHOICES, max_length=30)
Scott Baker1e559822016-05-26 20:57:10 -070095 url = StrippedCharField(max_length=1024, help_text="URL of resource", null=True, blank=True)
Jeremy Mowery11ed8992016-04-17 20:56:54 -070096
Scott Baker147cace2016-05-26 21:11:37 -070097 def __unicode__(self): return u'%s' % (self.name)
98
Scott Bakerb0eb23e2016-06-01 16:08:04 -070099 @property
100 def full_url(self):
101 if self.service_controller and self.service_controller.base_url:
102 return urlparse.urljoin(self.service_controller.base_url, self.url)
103 else:
104 return self.url
105
Scott Baker82498c52015-07-13 13:07:27 -0700106class Service(PlCoreBase, AttributeMixin):
Scott Baker008a9962015-04-15 20:58:20 -0700107 # when subclassing a service, redefine KIND to describe the new service
108 KIND = "generic"
109
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700110 description = models.TextField(
111 max_length=254, null=True, blank=True, help_text="Description of Service")
Siobhan Tully00353f72013-10-08 21:53:27 -0400112 enabled = models.BooleanField(default=True)
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700113 kind = StrippedCharField(
114 max_length=30, help_text="Kind of service", default=KIND)
Tony Mackd84b1ff2015-03-09 13:03:56 -0400115 name = StrippedCharField(max_length=30, help_text="Service Name")
Scott Bakerd3e30292016-05-02 09:38:24 -0700116 versionNumber = StrippedCharField(blank=True, null=True,
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700117 max_length=30, help_text="Version of Service Definition")
Siobhan Tullycf04fb62014-01-11 11:25:57 -0500118 published = models.BooleanField(default=True)
Tony Mackd84b1ff2015-03-09 13:03:56 -0400119 view_url = StrippedCharField(blank=True, null=True, max_length=1024)
120 icon_url = StrippedCharField(blank=True, null=True, max_length=1024)
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700121 public_key = models.TextField(
122 null=True, blank=True, max_length=1024, help_text="Public key string")
Scott Bakerf60c0102015-11-12 16:22:52 -0800123 private_key_fn = StrippedCharField(blank=True, null=True, max_length=1024)
Siobhan Tully00353f72013-10-08 21:53:27 -0400124
Scott Baker2f0828e2015-07-13 12:33:28 -0700125 # Service_specific_attribute and service_specific_id are opaque to XOS
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700126 service_specific_id = StrippedCharField(
127 max_length=30, blank=True, null=True)
Scott Baker2f0828e2015-07-13 12:33:28 -0700128 service_specific_attribute = models.TextField(blank=True, null=True)
129
Scott Baker80d65b12016-05-26 20:20:47 -0700130 controller = models.ForeignKey(ServiceController, related_name='services',
131 help_text="The Service Controller this Service uses")
132
Scott Baker008a9962015-04-15 20:58:20 -0700133 def __init__(self, *args, **kwargs):
134 # for subclasses, set the default kind appropriately
135 self._meta.get_field("kind").default = self.KIND
136 super(Service, self).__init__(*args, **kwargs)
137
138 @classmethod
139 def get_service_objects(cls):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700140 return cls.objects.filter(kind=cls.KIND)
Scott Baker008a9962015-04-15 20:58:20 -0700141
Scott Baker16573d32015-07-24 15:36:02 -0700142 @classmethod
Scott Baker542cd6f2015-10-19 21:18:53 -0700143 def get_deleted_service_objects(cls):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700144 return cls.deleted_objects.filter(kind=cls.KIND)
Scott Baker542cd6f2015-10-19 21:18:53 -0700145
146 @classmethod
Scott Baker16573d32015-07-24 15:36:02 -0700147 def get_service_objects_by_user(cls, user):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700148 return cls.select_by_user(user).filter(kind=cls.KIND)
Scott Baker16573d32015-07-24 15:36:02 -0700149
150 @classmethod
151 def select_by_user(cls, user):
152 if user.is_admin:
153 return cls.objects.all()
154 else:
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700155 service_ids = [
156 sp.slice.id for sp in ServicePrivilege.objects.filter(user=user)]
Scott Baker16573d32015-07-24 15:36:02 -0700157 return cls.objects.filter(id__in=service_ids)
158
Scott Bakercce158d2015-12-07 22:20:40 -0800159 @property
160 def serviceattribute_dict(self):
161 attrs = {}
162 for attr in self.serviceattributes.all():
163 attrs[attr.name] = attr.value
164 return attrs
165
Siobhan Tully00353f72013-10-08 21:53:27 -0400166 def __unicode__(self): return u'%s' % (self.name)
167
Tony Mack9d2ea092015-04-29 12:23:10 -0400168 def can_update(self, user):
169 return user.can_update_service(self, allow=['admin'])
Scott Baker618a4892015-07-06 14:27:31 -0700170
Scott Baker98436732015-05-11 16:36:41 -0700171 def get_scalable_nodes(self, slice, max_per_node=None, exclusive_slices=[]):
172 """
173 Get a list of nodes that can be used to scale up a slice.
174
175 slice - slice to scale up
Tony Mack3de59e32015-08-19 11:58:18 -0400176 max_per_node - maximum numbers of instances that 'slice' can have on a single node
Scott Baker98436732015-05-11 16:36:41 -0700177 exclusive_slices - list of slices that must have no nodes in common with 'slice'.
178 """
179
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700180 # late import to get around order-of-imports constraint in __init__.py
181 from core.models import Node, Instance
Scott Baker98436732015-05-11 16:36:41 -0700182
183 nodes = list(Node.objects.all())
184
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700185 conflicting_instances = Instance.objects.filter(
186 slice__in=exclusive_slices)
187 conflicting_nodes = Node.objects.filter(
188 instances__in=conflicting_instances)
Scott Baker98436732015-05-11 16:36:41 -0700189
190 nodes = [x for x in nodes if x not in conflicting_nodes]
191
Tony Mack3de59e32015-08-19 11:58:18 -0400192 # If max_per_node is set, then limit the number of instances this slice
Scott Baker98436732015-05-11 16:36:41 -0700193 # can have on a single node.
194 if max_per_node:
195 acceptable_nodes = []
196 for node in nodes:
Tony Mack3de59e32015-08-19 11:58:18 -0400197 existing_count = node.instances.filter(slice=slice).count()
Scott Baker98436732015-05-11 16:36:41 -0700198 if existing_count < max_per_node:
199 acceptable_nodes.append(node)
200 nodes = acceptable_nodes
201
202 return nodes
203
204 def pick_node(self, slice, max_per_node=None, exclusive_slices=[]):
205 # Pick the best node to scale up a slice.
206
207 nodes = self.get_scalable_nodes(slice, max_per_node, exclusive_slices)
Tony Mack3de59e32015-08-19 11:58:18 -0400208 nodes = sorted(nodes, key=lambda node: node.instances.all().count())
Scott Baker98436732015-05-11 16:36:41 -0700209 if not nodes:
210 return None
211 return nodes[0]
212
213 def adjust_scale(self, slice_hint, scale, max_per_node=None, exclusive_slices=[]):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700214 # late import to get around order-of-imports constraint in __init__.py
215 from core.models import Instance
Scott Baker98436732015-05-11 16:36:41 -0700216
217 slices = [x for x in self.slices.all() if slice_hint in x.name]
218 for slice in slices:
Tony Mack3de59e32015-08-19 11:58:18 -0400219 while slice.instances.all().count() > scale:
220 s = slice.instances.all()[0]
221 # print "drop instance", s
Scott Baker98436732015-05-11 16:36:41 -0700222 s.delete()
223
Tony Mack3de59e32015-08-19 11:58:18 -0400224 while slice.instances.all().count() < scale:
Scott Baker98436732015-05-11 16:36:41 -0700225 node = self.pick_node(slice, max_per_node, exclusive_slices)
226 if not node:
227 # no more available nodes
228 break
229
230 image = slice.default_image
231 if not image:
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700232 raise XOSConfigurationError(
233 "No default_image for slice %s" % slice.name)
Scott Baker98436732015-05-11 16:36:41 -0700234
235 flavor = slice.default_flavor
236 if not flavor:
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700237 raise XOSConfigurationError(
238 "No default_flavor for slice %s" % slice.name)
Scott Baker98436732015-05-11 16:36:41 -0700239
Tony Mack3de59e32015-08-19 11:58:18 -0400240 s = Instance(slice=slice,
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700241 node=node,
242 creator=slice.creator,
243 image=image,
244 flavor=flavor,
245 deployment=node.site_deployment.deployment)
Scott Baker98436732015-05-11 16:36:41 -0700246 s.save()
247
Tony Mack3de59e32015-08-19 11:58:18 -0400248 # print "add instance", s
Tony Mack9d2ea092015-04-29 12:23:10 -0400249
Scott Baker8a67ae72015-12-09 22:54:52 -0800250 def get_vtn_src_nets(self):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700251 nets = []
Scott Baker8a67ae72015-12-09 22:54:52 -0800252 for slice in self.slices.all():
253 for ns in slice.networkslices.all():
254 if not ns.network:
255 continue
Scott Bakerf63cf1c2016-01-12 19:59:12 -0800256# if ns.network.template.access in ["direct", "indirect"]:
257# # skip access networks; we want to use the private network
258# continue
Scott Bakere6e78552016-04-11 21:26:39 -0700259 if "management" in ns.network.name:
260 # don't try to connect the management network to anything
261 continue
Scott Baker8a67ae72015-12-09 22:54:52 -0800262 if ns.network.name in ["wan_network", "lan_network"]:
263 # we don't want to attach to the vCPE's lan or wan network
264 # we only want to attach to its private network
265 # TODO: fix hard-coding of network name
266 continue
267 for cn in ns.network.controllernetworks.all():
268 if cn.net_id:
269 net = {"name": ns.network.name, "net_id": cn.net_id}
270 nets.append(net)
271 return nets
272
Scott Bakercbd718e2015-12-08 21:31:18 -0800273 def get_vtn_nets(self):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700274 nets = []
Scott Bakercbd718e2015-12-08 21:31:18 -0800275 for slice in self.slices.all():
276 for ns in slice.networkslices.all():
277 if not ns.network:
278 continue
Scott Baker8a67ae72015-12-09 22:54:52 -0800279 if ns.network.template.access not in ["direct", "indirect"]:
280 # skip anything that's not an access network
281 continue
Scott Bakercbd718e2015-12-08 21:31:18 -0800282 for cn in ns.network.controllernetworks.all():
283 if cn.net_id:
284 net = {"name": ns.network.name, "net_id": cn.net_id}
285 nets.append(net)
286 return nets
287
288 def get_vtn_dependencies_nets(self):
289 provider_nets = []
Scott Baker5380a442015-12-08 19:27:50 -0800290 for tenant in self.subscribed_tenants.all():
291 if tenant.provider_service:
Scott Bakercbd718e2015-12-08 21:31:18 -0800292 for net in tenant.provider_service.get_vtn_nets():
293 if not net in provider_nets:
294 provider_nets.append(net)
295 return provider_nets
296
297 def get_vtn_dependencies_ids(self):
298 return [x["net_id"] for x in self.get_vtn_dependencies_nets()]
299
300 def get_vtn_dependencies_names(self):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700301 return [x["name"] + "_" + x["net_id"] for x in self.get_vtn_dependencies_nets()]
Scott Bakercbd718e2015-12-08 21:31:18 -0800302
Scott Baker8a67ae72015-12-09 22:54:52 -0800303 def get_vtn_src_ids(self):
304 return [x["net_id"] for x in self.get_vtn_src_nets()]
Scott Bakercbd718e2015-12-08 21:31:18 -0800305
Scott Baker8a67ae72015-12-09 22:54:52 -0800306 def get_vtn_src_names(self):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700307 return [x["name"] + "_" + x["net_id"] for x in self.get_vtn_src_nets()]
Scott Baker5380a442015-12-08 19:27:50 -0800308
309
Siobhan Tully00353f72013-10-08 21:53:27 -0400310class ServiceAttribute(PlCoreBase):
Scott Bakercce158d2015-12-07 22:20:40 -0800311 name = models.CharField(help_text="Attribute Name", max_length=128)
Tony Mackd84b1ff2015-03-09 13:03:56 -0400312 value = StrippedCharField(help_text="Attribute Value", max_length=1024)
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700313 service = models.ForeignKey(Service, related_name='serviceattributes',
314 help_text="The Service this attribute is associated with")
315
Siobhan Tully00353f72013-10-08 21:53:27 -0400316
Tony Mack9d2ea092015-04-29 12:23:10 -0400317class ServiceRole(PlCoreBase):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700318 ROLE_CHOICES = (('admin', 'Admin'),)
Tony Mack9d2ea092015-04-29 12:23:10 -0400319 role = StrippedCharField(choices=ROLE_CHOICES, unique=True, max_length=30)
320
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700321 def __unicode__(self): return u'%s' % (self.role)
322
Tony Mack9d2ea092015-04-29 12:23:10 -0400323
324class ServicePrivilege(PlCoreBase):
325 user = models.ForeignKey('User', related_name='serviceprivileges')
326 service = models.ForeignKey('Service', related_name='serviceprivileges')
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700327 role = models.ForeignKey('ServiceRole', related_name='serviceprivileges')
Tony Mack9d2ea092015-04-29 12:23:10 -0400328
329 class Meta:
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700330 unique_together = ('user', 'service', 'role')
Tony Mack9d2ea092015-04-29 12:23:10 -0400331
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700332 def __unicode__(self): return u'%s %s %s' % (
333 self.service, self.user, self.role)
Tony Mack9d2ea092015-04-29 12:23:10 -0400334
335 def can_update(self, user):
336 if not self.service.enabled:
337 raise PermissionDenied, "Cannot modify permission(s) of a disabled service"
338 return self.service.can_update(user)
339
340 def save(self, *args, **kwds):
341 if not self.service.enabled:
342 raise PermissionDenied, "Cannot modify permission(s) of a disabled service"
343 super(ServicePrivilege, self).save(*args, **kwds)
344
345 def delete(self, *args, **kwds):
346 if not self.service.enabled:
347 raise PermissionDenied, "Cannot modify permission(s) of a disabled service"
Scott Baker4587b822015-07-01 18:29:08 -0700348 super(ServicePrivilege, self).delete(*args, **kwds)
349
Scott Baker16573d32015-07-24 15:36:02 -0700350 @classmethod
351 def select_by_user(cls, user):
Tony Mack9d2ea092015-04-29 12:23:10 -0400352 if user.is_admin:
Scott Baker16573d32015-07-24 15:36:02 -0700353 qs = cls.objects.all()
Tony Mack9d2ea092015-04-29 12:23:10 -0400354 else:
Scott Baker16573d32015-07-24 15:36:02 -0700355 qs = cls.objects.filter(user=user)
Scott Baker4587b822015-07-01 18:29:08 -0700356 return qs
357
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700358
Scott Baker82498c52015-07-13 13:07:27 -0700359class TenantRoot(PlCoreBase, AttributeMixin):
Scott Baker4587b822015-07-01 18:29:08 -0700360 """ A tenantRoot is one of the things that can sit at the root of a chain
361 of tenancy. This object represents a node.
362 """
363
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700364 KIND = "generic"
Scott Baker4587b822015-07-01 18:29:08 -0700365 kind = StrippedCharField(max_length=30, default=KIND)
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700366 name = StrippedCharField(
367 max_length=255, help_text="name", blank=True, null=True)
Scott Baker4587b822015-07-01 18:29:08 -0700368
Scott Bakerefcec632015-07-07 12:12:42 -0700369 service_specific_attribute = models.TextField(blank=True, null=True)
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700370 service_specific_id = StrippedCharField(
371 max_length=30, blank=True, null=True)
Scott Baker4587b822015-07-01 18:29:08 -0700372
Scott Bakerdb66fd32015-07-07 17:59:44 -0700373 def __init__(self, *args, **kwargs):
374 # for subclasses, set the default kind appropriately
375 self._meta.get_field("kind").default = self.KIND
376 super(TenantRoot, self).__init__(*args, **kwargs)
377
Scott Baker618a4892015-07-06 14:27:31 -0700378 def __unicode__(self):
379 if not self.name:
380 return u"%s-tenant_root-#%s" % (str(self.kind), str(self.id))
381 else:
382 return self.name
383
384 def can_update(self, user):
385 return user.can_update_tenant_root(self, allow=['admin'])
386
Scott Bakerefcec632015-07-07 12:12:42 -0700387 def get_subscribed_tenants(self, tenant_class):
388 ids = self.subscribed_tenants.filter(kind=tenant_class.KIND)
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700389 return tenant_class.objects.filter(id__in=ids)
Scott Bakerefcec632015-07-07 12:12:42 -0700390
391 def get_newest_subscribed_tenant(self, kind):
392 st = list(self.get_subscribed_tenants(kind))
393 if not st:
394 return None
395 return sorted(st, key=attrgetter('id'))[0]
396
397 @classmethod
398 def get_tenant_objects(cls):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700399 return cls.objects.filter(kind=cls.KIND)
Scott Bakerefcec632015-07-07 12:12:42 -0700400
Scott Baker16573d32015-07-24 15:36:02 -0700401 @classmethod
402 def get_tenant_objects_by_user(cls, user):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700403 return cls.select_by_user(user).filter(kind=cls.KIND)
Scott Baker16573d32015-07-24 15:36:02 -0700404
405 @classmethod
406 def select_by_user(cls, user):
407 if user.is_admin:
408 return cls.objects.all()
409 else:
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700410 tr_ids = [
411 trp.tenant_root.id for trp in TenantRootPrivilege.objects.filter(user=user)]
Scott Baker16573d32015-07-24 15:36:02 -0700412 return cls.objects.filter(id__in=tr_ids)
413
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700414 # helper function to be used in subclasses that want to ensure
415 # service_specific_id is unique
Scott Bakerdc880f32016-03-31 14:45:31 -0700416 def validate_unique_service_specific_id(self, none_okay=False):
417 if not none_okay and (self.service_specific_id is None):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700418 raise XOSMissingField("subscriber_specific_id is None, and it's a required field", fields={
419 "service_specific_id": "cannot be none"})
Scott Bakerdc880f32016-03-31 14:45:31 -0700420
421 if self.service_specific_id:
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700422 conflicts = self.get_tenant_objects().filter(
423 service_specific_id=self.service_specific_id)
Scott Bakerdc880f32016-03-31 14:45:31 -0700424 if self.pk:
Scott Bakere4c9d872016-04-01 16:28:41 -0700425 conflicts = conflicts.exclude(pk=self.pk)
Scott Bakerdc880f32016-03-31 14:45:31 -0700426 if conflicts:
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700427 raise XOSDuplicateKey("service_specific_id %s already exists" % self.service_specific_id, fields={
428 "service_specific_id": "duplicate key"})
429
Scott Bakerdc880f32016-03-31 14:45:31 -0700430
Scott Baker82498c52015-07-13 13:07:27 -0700431class Tenant(PlCoreBase, AttributeMixin):
Scott Baker8103d0f2015-04-10 16:42:26 -0700432 """ A tenant is a relationship between two entities, a subscriber and a
Scott Baker4587b822015-07-01 18:29:08 -0700433 provider. This object represents an edge.
Scott Baker8103d0f2015-04-10 16:42:26 -0700434
435 The subscriber can be a User, a Service, or a Tenant.
436
437 The provider is always a Service.
Scott Baker4587b822015-07-01 18:29:08 -0700438
439 TODO: rename "Tenant" to "Tenancy"
Scott Baker8103d0f2015-04-10 16:42:26 -0700440 """
Scott Baker008a9962015-04-15 20:58:20 -0700441
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700442 CONNECTIVITY_CHOICES = (('public', 'Public'),
443 ('private', 'Private'), ('na', 'Not Applicable'))
Scott Baker925a8fa2015-04-26 20:30:40 -0700444
Scott Baker008a9962015-04-15 20:58:20 -0700445 # when subclassing a service, redefine KIND to describe the new service
446 KIND = "generic"
447
448 kind = StrippedCharField(max_length=30, default=KIND)
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700449 provider_service = models.ForeignKey(
450 Service, related_name='provided_tenants')
Scott Baker4587b822015-07-01 18:29:08 -0700451
452 # The next four things are the various type of objects that can be subscribers of this Tenancy
453 # relationship. One and only one can be used at a time.
Scott Baker97ca9102016-04-13 16:56:39 -0700454 # XXX these should really be changed to GenericForeignKey
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700455 subscriber_service = models.ForeignKey(
456 Service, related_name='subscribed_tenants', blank=True, null=True)
457 subscriber_tenant = models.ForeignKey(
458 "Tenant", related_name='subscribed_tenants', blank=True, null=True)
459 subscriber_user = models.ForeignKey(
460 "User", related_name='subscribed_tenants', blank=True, null=True)
461 subscriber_root = models.ForeignKey(
462 "TenantRoot", related_name="subscribed_tenants", blank=True, null=True)
463 subscriber_network = models.ForeignKey(
464 "Network", related_name="subscribed_tenants", blank=True, null=True)
Scott Baker4587b822015-07-01 18:29:08 -0700465
466 # Service_specific_attribute and service_specific_id are opaque to XOS
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700467 service_specific_id = StrippedCharField(
468 max_length=30, blank=True, null=True)
Scott Baker76934d82015-05-06 19:49:31 -0700469 service_specific_attribute = models.TextField(blank=True, null=True)
Scott Baker4587b822015-07-01 18:29:08 -0700470
471 # Connect_method is only used by Coarse tenants
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700472 connect_method = models.CharField(
473 null=False, blank=False, max_length=30, choices=CONNECTIVITY_CHOICES, default="na")
Scott Baker8103d0f2015-04-10 16:42:26 -0700474
Scott Baker008a9962015-04-15 20:58:20 -0700475 def __init__(self, *args, **kwargs):
476 # for subclasses, set the default kind appropriately
477 self._meta.get_field("kind").default = self.KIND
478 super(Tenant, self).__init__(*args, **kwargs)
479
Scott Baker8103d0f2015-04-10 16:42:26 -0700480 def __unicode__(self):
Scott Bakerf996b762015-05-20 20:42:04 -0700481 return u"%s-tenant-%s" % (str(self.kind), str(self.id))
Scott Baker8103d0f2015-04-10 16:42:26 -0700482
Scott Baker008a9962015-04-15 20:58:20 -0700483 @classmethod
484 def get_tenant_objects(cls):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700485 return cls.objects.filter(kind=cls.KIND)
Scott Baker008a9962015-04-15 20:58:20 -0700486
Scott Bakere7fc9f52015-05-05 17:52:03 -0700487 @classmethod
Scott Baker16573d32015-07-24 15:36:02 -0700488 def get_tenant_objects_by_user(cls, user):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700489 return cls.select_by_user(user).filter(kind=cls.KIND)
Scott Baker16573d32015-07-24 15:36:02 -0700490
491 @classmethod
Scott Bakere7fc9f52015-05-05 17:52:03 -0700492 def get_deleted_tenant_objects(cls):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700493 return cls.deleted_objects.filter(kind=cls.KIND)
Scott Bakere7fc9f52015-05-05 17:52:03 -0700494
Scott Bakercce158d2015-12-07 22:20:40 -0800495 @property
496 def tenantattribute_dict(self):
497 attrs = {}
498 for attr in self.tenantattributes.all():
499 attrs[attr.name] = attr.value
500 return attrs
501
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700502 # helper function to be used in subclasses that want to ensure
503 # service_specific_id is unique
Scott Bakerd921e1c2015-04-20 14:24:29 -0700504 def validate_unique_service_specific_id(self):
505 if self.pk is None:
506 if self.service_specific_id is None:
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700507 raise XOSMissingField("subscriber_specific_id is None, and it's a required field", fields={
508 "service_specific_id": "cannot be none"})
Scott Bakerd921e1c2015-04-20 14:24:29 -0700509
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700510 conflicts = self.get_tenant_objects().filter(
511 service_specific_id=self.service_specific_id)
Scott Bakerd921e1c2015-04-20 14:24:29 -0700512 if conflicts:
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700513 raise XOSDuplicateKey("service_specific_id %s already exists" % self.service_specific_id, fields={
514 "service_specific_id": "duplicate key"})
Scott Bakerd921e1c2015-04-20 14:24:29 -0700515
Scott Baker618a4892015-07-06 14:27:31 -0700516 def save(self, *args, **kwargs):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700517 subCount = sum([1 for e in [self.subscriber_service, self.subscriber_tenant,
518 self.subscriber_user, self.subscriber_root] if e is not None])
Scott Baker618a4892015-07-06 14:27:31 -0700519 if (subCount > 1):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700520 raise XOSConflictingField(
521 "Only one of subscriber_service, subscriber_tenant, subscriber_user, subscriber_root should be set")
Scott Baker618a4892015-07-06 14:27:31 -0700522
523 super(Tenant, self).save(*args, **kwargs)
524
525 def get_subscribed_tenants(self, tenant_class):
526 ids = self.subscribed_tenants.filter(kind=tenant_class.KIND)
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700527 return tenant_class.objects.filter(id__in=ids)
Scott Baker618a4892015-07-06 14:27:31 -0700528
529 def get_newest_subscribed_tenant(self, kind):
530 st = list(self.get_subscribed_tenants(kind))
531 if not st:
532 return None
533 return sorted(st, key=attrgetter('id'))[0]
534
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700535
Scott Baker534dd2a2015-11-18 20:58:08 -0800536class Scheduler(object):
537 # XOS Scheduler Abstract Base Class
538 # Used to implement schedulers that pick which node to put instances on
539
540 def __init__(self, slice):
541 self.slice = slice
542
543 def pick(self):
544 # this method should return a tuple (node, parent)
545 # node is the node to instantiate on
546 # parent is for container_vm instances only, and is the VM that will
547 # hold the container
548
549 raise Exception("Abstract Base")
550
Scott Baker534dd2a2015-11-18 20:58:08 -0800551
Scott Baker534dd2a2015-11-18 20:58:08 -0800552class LeastLoadedNodeScheduler(Scheduler):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700553 # This scheduler always return the node with the fewest number of
554 # instances.
Scott Baker534dd2a2015-11-18 20:58:08 -0800555
Scott Baker6526f552016-03-04 11:29:02 -0800556 def __init__(self, slice, label=None):
Scott Baker534dd2a2015-11-18 20:58:08 -0800557 super(LeastLoadedNodeScheduler, self).__init__(slice)
Scott Baker6526f552016-03-04 11:29:02 -0800558 self.label = label
Scott Baker534dd2a2015-11-18 20:58:08 -0800559
560 def pick(self):
561 from core.models import Node
Pingping Linfa30bae2016-03-03 09:52:24 -0800562 if not self.slice.default_node:
563 nodes = list(Node.objects.all())
564 nodes = sorted(nodes, key=lambda node: node.instances.all().count())
565 else:
566 nodes = list(Node.objects.filter(name = self.slice.default_node))
Matteo Scandolo1d7ff812016-04-21 09:02:08 -0700567
Scott Baker6526f552016-03-04 11:29:02 -0800568 if self.label:
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700569 nodes = nodes.filter(nodelabels__name=self.label)
Scott Baker6526f552016-03-04 11:29:02 -0800570
571 nodes = list(nodes)
572
573 if not nodes:
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700574 raise Exception(
575 "LeastLoadedNodeScheduler: No suitable nodes to pick from")
Scott Bakercce158d2015-12-07 22:20:40 -0800576
Scott Baker534dd2a2015-11-18 20:58:08 -0800577 # TODO: logic to filter nodes by which nodes are up, and which
578 # nodes the slice can instantiate on.
Pingping Linfa30bae2016-03-03 09:52:24 -0800579# nodes = sorted(nodes, key=lambda node: node.instances.all().count())
Scott Baker534dd2a2015-11-18 20:58:08 -0800580 return [nodes[0], None]
581
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700582
Scott Baker534dd2a2015-11-18 20:58:08 -0800583class ContainerVmScheduler(Scheduler):
584 # This scheduler picks a VM in the slice with the fewest containers inside
585 # of it. If no VMs are suitable, then it creates a VM.
586
Scott Baker534dd2a2015-11-18 20:58:08 -0800587 MAX_VM_PER_CONTAINER = 10
588
589 def __init__(self, slice):
590 super(ContainerVmScheduler, self).__init__(slice)
591
592 @property
593 def image(self):
594 from core.models import Image
595
Scott Bakera8188a72016-05-19 17:54:52 -0700596 # If slice has default_image set then use it
597 if self.slice.default_image:
598 return self.slice.default_image
Scott Baker534dd2a2015-11-18 20:58:08 -0800599
Scott Bakera8188a72016-05-19 17:54:52 -0700600 raise XOPSProgrammingError("Please set a default image for %s" % self.slice.name)
Scott Baker534dd2a2015-11-18 20:58:08 -0800601
602 def make_new_instance(self):
603 from core.models import Instance, Flavor
604
605 flavors = Flavor.objects.filter(name="m1.small")
606 if not flavors:
607 raise XOSConfigurationError("No m1.small flavor")
608
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700609 (node, parent) = LeastLoadedNodeScheduler(self.slice).pick()
Scott Baker534dd2a2015-11-18 20:58:08 -0800610
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700611 instance = Instance(slice=self.slice,
612 node=node,
613 image=self.image,
614 creator=self.slice.creator,
615 deployment=node.site_deployment.deployment,
616 flavor=flavors[0],
617 isolation="vm",
618 parent=parent)
Scott Baker534dd2a2015-11-18 20:58:08 -0800619 instance.save()
620 # We rely on a special naming convention to identify the VMs that will
621 # hole containers.
622 instance.name = "%s-outer-%s" % (instance.slice.name, instance.id)
623 instance.save()
624 return instance
625
626 def pick(self):
627 from core.models import Instance, Flavor
628
629 for vm in self.slice.instances.filter(isolation="vm"):
630 avail_vms = []
631 if (vm.name.startswith("%s-outer-" % self.slice.name)):
632 container_count = Instance.objects.filter(parent=vm).count()
633 if (container_count < self.MAX_VM_PER_CONTAINER):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700634 avail_vms.append((vm, container_count))
Scott Baker534dd2a2015-11-18 20:58:08 -0800635 # sort by least containers-per-vm
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700636 avail_vms = sorted(avail_vms, key=lambda x: x[1])
Scott Baker534dd2a2015-11-18 20:58:08 -0800637 print "XXX", avail_vms
638 if avail_vms:
639 instance = avail_vms[0][0]
640 return (instance.node, instance)
641
642 instance = self.make_new_instance()
643 return (instance.node, instance)
644
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700645
Scott Bakere458afd2015-09-09 16:36:06 -0700646class TenantWithContainer(Tenant):
647 """ A tenant that manages a container """
648
Scott Bakere458afd2015-09-09 16:36:06 -0700649 class Meta:
650 proxy = True
651
652 def __init__(self, *args, **kwargs):
653 super(TenantWithContainer, self).__init__(*args, **kwargs)
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700654 self.cached_instance = None
Tony Mack6a782f92015-09-13 22:50:39 +0000655 self.orig_instance_id = self.get_initial_attribute("instance_id")
Scott Bakere274f6b2015-11-02 20:54:28 -0800656
Scott Bakere458afd2015-09-09 16:36:06 -0700657 @property
Tony Mack6a782f92015-09-13 22:50:39 +0000658 def instance(self):
659 from core.models import Instance
660 if getattr(self, "cached_instance", None):
661 return self.cached_instance
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700662 instance_id = self.get_attribute("instance_id")
Srikanth Vavilapalli1d14ff02015-11-17 12:21:02 -0600663 if not instance_id:
664 return None
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700665 instances = Instance.objects.filter(id=instance_id)
Srikanth Vavilapalli1d14ff02015-11-17 12:21:02 -0600666 if not instances:
667 return None
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700668 instance = instances[0]
Tony Mack6a782f92015-09-13 22:50:39 +0000669 instance.caller = self.creator
670 self.cached_instance = instance
671 return instance
Scott Bakere458afd2015-09-09 16:36:06 -0700672
Tony Mack6a782f92015-09-13 22:50:39 +0000673 @instance.setter
674 def instance(self, value):
Scott Bakere458afd2015-09-09 16:36:06 -0700675 if value:
676 value = value.id
Tony Mack6a782f92015-09-13 22:50:39 +0000677 if (value != self.get_attribute("instance_id", None)):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700678 self.cached_instance = None
Tony Mack6a782f92015-09-13 22:50:39 +0000679 self.set_attribute("instance_id", value)
Scott Bakere458afd2015-09-09 16:36:06 -0700680
Scott Bakere274f6b2015-11-02 20:54:28 -0800681 @property
Scott Bakerbdda0be2016-02-10 12:23:53 -0800682 def external_hostname(self):
683 return self.get_attribute("external_hostname", "")
684
685 @external_hostname.setter
686 def external_hostname(self, value):
687 self.set_attribute("external_hostname", value)
688
689 @property
690 def external_container(self):
691 return self.get_attribute("external_container", "")
692
693 @external_container.setter
694 def external_container(self, value):
695 self.set_attribute("external_container", value)
696
697 @property
Scott Bakere458afd2015-09-09 16:36:06 -0700698 def creator(self):
699 from core.models import User
700 if getattr(self, "cached_creator", None):
701 return self.cached_creator
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700702 creator_id = self.get_attribute("creator_id")
Scott Bakere458afd2015-09-09 16:36:06 -0700703 if not creator_id:
704 return None
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700705 users = User.objects.filter(id=creator_id)
Scott Bakere458afd2015-09-09 16:36:06 -0700706 if not users:
707 return None
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700708 user = users[0]
Scott Bakere458afd2015-09-09 16:36:06 -0700709 self.cached_creator = users[0]
710 return user
711
712 @creator.setter
713 def creator(self, value):
714 if value:
715 value = value.id
716 if (value != self.get_attribute("creator_id", None)):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700717 self.cached_creator = None
Scott Bakere458afd2015-09-09 16:36:06 -0700718 self.set_attribute("creator_id", value)
719
720 @property
721 def image(self):
722 from core.models import Image
723 # Implement the logic here to pick the image that should be used when
724 # instantiating the VM that will hold the container.
Scott Bakerabad4c92016-05-25 10:02:29 -0700725
Scott Bakercfaf0d32015-11-16 22:51:02 -0800726 slice = self.provider_service.slices.all()
727 if not slice:
728 raise XOSProgrammingError("provider service has no slice")
729 slice = slice[0]
730
Scott Bakerabad4c92016-05-25 10:02:29 -0700731 # If slice has default_image set then use it
732 if slice.default_image:
733 return slice.default_image
Scott Bakere458afd2015-09-09 16:36:06 -0700734
Scott Bakera8188a72016-05-19 17:54:52 -0700735 raise XOPSProgrammingError("Please set a default image for %s" % self.slice.name)
Scott Bakere274f6b2015-11-02 20:54:28 -0800736
Scott Bakercfaf0d32015-11-16 22:51:02 -0800737 def save_instance(self, instance):
738 # Override this function to do custom pre-save or post-save processing,
739 # such as creating ports for containers.
740 instance.save()
Scott Bakere274f6b2015-11-02 20:54:28 -0800741
Srikanth Vavilapalli1d14ff02015-11-17 12:21:02 -0600742 def pick_least_loaded_instance_in_slice(self, slices):
743 for slice in slices:
744 if slice.instances.all().count() > 0:
745 for instance in slice.instances.all():
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700746 # Pick the first instance that has lesser than 5 tenants
747 if self.count_of_tenants_of_an_instance(instance) < 5:
748 return instance
Srikanth Vavilapalli1d14ff02015-11-17 12:21:02 -0600749 return None
750
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700751 # TODO: Ideally the tenant count for an instance should be maintained using a
752 # many-to-one relationship attribute, however this model being proxy, it does
753 # not permit any new attributes to be defined. Find if any better solutions
Srikanth Vavilapalli1d14ff02015-11-17 12:21:02 -0600754 def count_of_tenants_of_an_instance(self, instance):
755 tenant_count = 0
756 for tenant in self.get_tenant_objects().all():
757 if tenant.get_attribute("instance_id", None) == instance.id:
758 tenant_count += 1
759 return tenant_count
760
Scott Bakercfaf0d32015-11-16 22:51:02 -0800761 def manage_container(self):
Tony Mack6a782f92015-09-13 22:50:39 +0000762 from core.models import Instance, Flavor
Scott Bakere458afd2015-09-09 16:36:06 -0700763
764 if self.deleted:
765 return
766
Tony Mack6a782f92015-09-13 22:50:39 +0000767 if (self.instance is not None) and (self.instance.image != self.image):
768 self.instance.delete()
769 self.instance = None
Scott Bakere458afd2015-09-09 16:36:06 -0700770
Tony Mack6a782f92015-09-13 22:50:39 +0000771 if self.instance is None:
Scott Bakere458afd2015-09-09 16:36:06 -0700772 if not self.provider_service.slices.count():
Srikanth Vavilapalli1d14ff02015-11-17 12:21:02 -0600773 raise XOSConfigurationError("The service has no slices")
Scott Bakere458afd2015-09-09 16:36:06 -0700774
Srikanth Vavilapalli1d14ff02015-11-17 12:21:02 -0600775 new_instance_created = False
776 instance = None
777 if self.get_attribute("use_same_instance_for_multiple_tenants", default=False):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700778 # Find if any existing instances can be used for this tenant
Srikanth Vavilapalli1d14ff02015-11-17 12:21:02 -0600779 slices = self.provider_service.slices.all()
780 instance = self.pick_least_loaded_instance_in_slice(slices)
Scott Bakere458afd2015-09-09 16:36:06 -0700781
Srikanth Vavilapalli1d14ff02015-11-17 12:21:02 -0600782 if not instance:
Srikanth Vavilapalli71120a92015-11-17 13:41:38 -0600783 slice = self.provider_service.slices.all()[0]
Scott Bakercfaf0d32015-11-16 22:51:02 -0800784
Srikanth Vavilapalli2ab2d072016-02-15 01:18:09 -0500785 flavor = slice.default_flavor
786 if not flavor:
787 flavors = Flavor.objects.filter(name="m1.small")
788 if not flavors:
789 raise XOSConfigurationError("No m1.small flavor")
790 flavor = flavors[0]
791
Srikanth Vavilapalli71120a92015-11-17 13:41:38 -0600792 if slice.default_isolation == "container_vm":
Scott Baker534dd2a2015-11-18 20:58:08 -0800793 (node, parent) = ContainerVmScheduler(slice).pick()
Srikanth Vavilapalli71120a92015-11-17 13:41:38 -0600794 else:
Scott Baker534dd2a2015-11-18 20:58:08 -0800795 (node, parent) = LeastLoadedNodeScheduler(slice).pick()
Scott Bakercfaf0d32015-11-16 22:51:02 -0800796
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700797 instance = Instance(slice=slice,
798 node=node,
799 image=self.image,
800 creator=self.creator,
801 deployment=node.site_deployment.deployment,
802 flavor=flavor,
803 isolation=slice.default_isolation,
804 parent=parent)
Srikanth Vavilapalli71120a92015-11-17 13:41:38 -0600805 self.save_instance(instance)
Srikanth Vavilapalli1d14ff02015-11-17 12:21:02 -0600806 new_instance_created = True
Scott Bakere458afd2015-09-09 16:36:06 -0700807
808 try:
Tony Mack6a782f92015-09-13 22:50:39 +0000809 self.instance = instance
Scott Bakere458afd2015-09-09 16:36:06 -0700810 super(TenantWithContainer, self).save()
811 except:
Srikanth Vavilapalli1d14ff02015-11-17 12:21:02 -0600812 if new_instance_created:
813 instance.delete()
Scott Bakere458afd2015-09-09 16:36:06 -0700814 raise
815
816 def cleanup_container(self):
Tony Mack6a782f92015-09-13 22:50:39 +0000817 if self.instance:
Srikanth Vavilapalli1d14ff02015-11-17 12:21:02 -0600818 if self.get_attribute("use_same_instance_for_multiple_tenants", default=False):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700819 # Delete the instance only if this is last tenant in that
820 # instance
821 tenant_count = self.count_of_tenants_of_an_instance(
822 self.instance)
Srikanth Vavilapalli1d14ff02015-11-17 12:21:02 -0600823 if tenant_count == 0:
824 self.instance.delete()
825 else:
826 self.instance.delete()
Tony Mack6a782f92015-09-13 22:50:39 +0000827 self.instance = None
Scott Baker618a4892015-07-06 14:27:31 -0700828
Scott Bakeree83f2a2015-12-10 23:23:07 -0800829 def save(self, *args, **kwargs):
830 if (not self.creator) and (hasattr(self, "caller")) and (self.caller):
831 self.creator = self.caller
832 super(TenantWithContainer, self).save(*args, **kwargs)
833
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700834
Scott Baker925a8fa2015-04-26 20:30:40 -0700835class CoarseTenant(Tenant):
Scott Baker4587b822015-07-01 18:29:08 -0700836 """ TODO: rename "CoarseTenant" --> "StaticTenant" """
Scott Baker925a8fa2015-04-26 20:30:40 -0700837 class Meta:
838 proxy = True
Siobhan Tully00353f72013-10-08 21:53:27 -0400839
Scott Baker2461bec2015-08-14 09:10:11 -0700840 KIND = COARSE_KIND
Scott Baker925a8fa2015-04-26 20:30:40 -0700841
842 def save(self, *args, **kwargs):
843 if (not self.subscriber_service):
844 raise XOSValidationError("subscriber_service cannot be null")
845 if (self.subscriber_tenant or self.subscriber_user):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700846 raise XOSValidationError(
847 "subscriber_tenant and subscriber_user must be null")
Scott Baker925a8fa2015-04-26 20:30:40 -0700848
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700849 super(CoarseTenant, self).save()
850
Scott Baker4587b822015-07-01 18:29:08 -0700851
852class Subscriber(TenantRoot):
853 """ Intermediate class for TenantRoots that are to be Subscribers """
854
855 class Meta:
856 proxy = True
857
858 KIND = "Subscriber"
859
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700860
Scott Baker4587b822015-07-01 18:29:08 -0700861class Provider(TenantRoot):
862 """ Intermediate class for TenantRoots that are to be Providers """
863
864 class Meta:
865 proxy = True
866
867 KIND = "Provider"
868
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700869
Scott Baker462a1d92015-10-15 15:59:19 -0700870class TenantAttribute(PlCoreBase):
Scott Baker907f9912015-10-20 17:12:36 -0700871 name = models.CharField(help_text="Attribute Name", max_length=128)
Scott Baker462a1d92015-10-15 15:59:19 -0700872 value = models.TextField(help_text="Attribute Value")
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700873 tenant = models.ForeignKey(Tenant, related_name='tenantattributes',
874 help_text="The Tenant this attribute is associated with")
Scott Baker462a1d92015-10-15 15:59:19 -0700875
Scott Baker1271a4d2016-04-06 14:34:49 -0700876 def __unicode__(self): return u'%s-%s' % (self.name, self.id)
877
Scott Baker4587b822015-07-01 18:29:08 -0700878
879class TenantRootRole(PlCoreBase):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700880 ROLE_CHOICES = (('admin', 'Admin'), ('access', 'Access'))
Scott Baker4587b822015-07-01 18:29:08 -0700881
882 role = StrippedCharField(choices=ROLE_CHOICES, unique=True, max_length=30)
883
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700884 def __unicode__(self): return u'%s' % (self.role)
885
Scott Baker4587b822015-07-01 18:29:08 -0700886
887class TenantRootPrivilege(PlCoreBase):
888 user = models.ForeignKey('User', related_name="tenant_root_privileges")
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700889 tenant_root = models.ForeignKey(
890 'TenantRoot', related_name="tenant_root_privileges")
891 role = models.ForeignKey(
892 'TenantRootRole', related_name="tenant_root_privileges")
Scott Baker4587b822015-07-01 18:29:08 -0700893
894 class Meta:
895 unique_together = ('user', 'tenant_root', 'role')
896
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700897 def __unicode__(self): return u'%s %s %s' % (
898 self.tenant_root, self.user, self.role)
Scott Baker4587b822015-07-01 18:29:08 -0700899
900 def save(self, *args, **kwds):
901 if not self.user.is_active:
902 raise PermissionDenied, "Cannot modify role(s) of a disabled user"
Scott Baker335882a2015-07-24 10:15:31 -0700903 super(TenantRootPrivilege, self).save(*args, **kwds)
Scott Baker4587b822015-07-01 18:29:08 -0700904
905 def can_update(self, user):
Scott Baker335882a2015-07-24 10:15:31 -0700906 return user.can_update_tenant_root_privilege(self)
Scott Baker4587b822015-07-01 18:29:08 -0700907
Scott Baker16573d32015-07-24 15:36:02 -0700908 @classmethod
909 def select_by_user(cls, user):
Scott Baker4587b822015-07-01 18:29:08 -0700910 if user.is_admin:
Scott Baker5116b302015-07-24 15:48:03 -0700911 return cls.objects.all()
Scott Baker4587b822015-07-01 18:29:08 -0700912 else:
Scott Baker5116b302015-07-24 15:48:03 -0700913 # User can see his own privilege
914 trp_ids = [trp.id for trp in cls.objects.filter(user=user)]
915
916 # A slice admin can see the SlicePrivileges for his Slice
917 for priv in cls.objects.filter(user=user, role__role="admin"):
Jeremy Mowery11ed8992016-04-17 20:56:54 -0700918 trp_ids.extend(
919 [trp.id for trp in cls.objects.filter(tenant_root=priv.tenant_root)])
Scott Baker5116b302015-07-24 15:48:03 -0700920
921 return cls.objects.filter(id__in=trp_ids)
922
Jeremy Mowery95635132016-04-15 17:39:49 -0700923
Jeremy Moweryb31bd9e2016-03-14 23:59:11 -0700924class TenantRole(PlCoreBase):
Jeremy Mowery98e97d72016-04-15 00:05:27 -0700925 """A TenantRole option."""
Jeremy Mowery95635132016-04-15 17:39:49 -0700926 ROLE_CHOICES = (('admin', 'Admin'), ('access', 'Access'))
Jeremy Moweryb31bd9e2016-03-14 23:59:11 -0700927 role = StrippedCharField(choices=ROLE_CHOICES, unique=True, max_length=30)
Jeremy Mowery95635132016-04-15 17:39:49 -0700928
929 def __unicode__(self): return u'%s' % (self.role)
930
Jeremy Moweryb31bd9e2016-03-14 23:59:11 -0700931
932class TenantPrivilege(PlCoreBase):
Jeremy Mowery98e97d72016-04-15 00:05:27 -0700933 """"A TenantPrivilege which defines how users can access a particular Tenant.
934
935 Attributes:
936 id (models.AutoField): The ID of the privilege.
937 user (models.ForeignKey): A Foreign Key to the a User.
938 tenant (models.ForeignKey): A ForeignKey to the Tenant.
939 role (models.ForeignKey): A ForeignKey to the TenantRole.
940 """
Jeremy Moweryc86352a2016-04-10 23:00:54 -0700941 id = models.AutoField(primary_key=True)
Jeremy Mowery5d06a232016-04-04 22:30:44 -0700942 user = models.ForeignKey('User', related_name="tenantprivileges")
943 tenant = models.ForeignKey('Tenant', related_name="tenantprivileges")
944 role = models.ForeignKey('TenantRole', related_name="tenantprivileges")
Jeremy Moweryb31bd9e2016-03-14 23:59:11 -0700945
Jeremy Mowery95635132016-04-15 17:39:49 -0700946 def __unicode__(self): return u'%s %s %s' % (
947 self.tenant, self.user, self.role)
Jeremy Moweryb31bd9e2016-03-14 23:59:11 -0700948
949 def save(self, *args, **kwds):
950 if not self.user.is_active:
951 raise PermissionDenied, "Cannot modify role(s) of a disabled user"
952 super(TenantPrivilege, self).save(*args, **kwds)
953
954 def can_update(self, user):
955 return user.can_update_tenant_privilege(self)
956
957 @classmethod
958 def select_by_user(cls, user):
959 if user.is_admin:
960 return cls.objects.all()
961 else:
962 # User can see his own privilege
963 trp_ids = [trp.id for trp in cls.objects.filter(user=user)]
964
965 # A tenant admin can see the TenantPrivileges for their Tenants
966 for priv in cls.objects.filter(user=user, role__role="admin"):
Jeremy Mowery95635132016-04-15 17:39:49 -0700967 trp_ids.extend(
968 [trp.id for trp in cls.objects.filter(tenant=priv.tenant)])
Jeremy Moweryb31bd9e2016-03-14 23:59:11 -0700969
970 return cls.objects.filter(id__in=trp_ids)