| Scott Baker | f57e559 | 2015-04-14 17:18:51 -0700 | [diff] [blame] | 1 | import json |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 2 | from operator import attrgetter |
| Siobhan Tully | 00353f7 | 2013-10-08 21:53:27 -0400 | [diff] [blame] | 3 | |
| Scott Baker | 80d65b1 | 2016-05-26 20:20:47 -0700 | [diff] [blame] | 4 | from core.models import PlCoreBase, PlCoreBaseManager, SingletonModel, XOS |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 5 | from core.models.plcorebase import StrippedCharField |
| 6 | from django.db import models |
| 7 | from xos.exceptions import * |
| 8 | |
| 9 | COARSE_KIND = "coarse" |
| 10 | |
| Scott Baker | 80d65b1 | 2016-05-26 20:20:47 -0700 | [diff] [blame] | 11 | def get_xos(): |
| 12 | xos = XOS.objects.all() |
| 13 | |
| 14 | if xos: |
| 15 | return xos[0] |
| 16 | else: |
| 17 | return None |
| 18 |
|
| Scott Baker | 2461bec | 2015-08-14 09:10:11 -0700 | [diff] [blame] | 19 | |
| Scott Baker | 82498c5 | 2015-07-13 13:07:27 -0700 | [diff] [blame] | 20 | class AttributeMixin(object): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 21 | # helper for extracting things from a json-encoded |
| 22 | # service_specific_attribute |
| 23 | |
| Scott Baker | 82498c5 | 2015-07-13 13:07:27 -0700 | [diff] [blame] | 24 | def get_attribute(self, name, default=None): |
| 25 | if self.service_specific_attribute: |
| 26 | attributes = json.loads(self.service_specific_attribute) |
| 27 | else: |
| 28 | attributes = {} |
| 29 | return attributes.get(name, default) |
| 30 | |
| 31 | def set_attribute(self, name, value): |
| 32 | if self.service_specific_attribute: |
| 33 | attributes = json.loads(self.service_specific_attribute) |
| 34 | else: |
| 35 | attributes = {} |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 36 | attributes[name] = value |
| Scott Baker | 82498c5 | 2015-07-13 13:07:27 -0700 | [diff] [blame] | 37 | self.service_specific_attribute = json.dumps(attributes) |
| 38 | |
| 39 | def get_initial_attribute(self, name, default=None): |
| 40 | if self._initial["service_specific_attribute"]: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 41 | attributes = json.loads( |
| 42 | self._initial["service_specific_attribute"]) |
| Scott Baker | 82498c5 | 2015-07-13 13:07:27 -0700 | [diff] [blame] | 43 | else: |
| 44 | attributes = {} |
| 45 | return attributes.get(name, default) |
| 46 | |
| Scott Baker | 74404fe | 2015-07-13 13:54:06 -0700 | [diff] [blame] | 47 | @classmethod |
| Scott Baker | 6dfde21 | 2016-03-01 20:10:24 -0800 | [diff] [blame] | 48 | def get_default_attribute(cls, name): |
| 49 | for (attrname, default) in cls.simple_attributes: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 50 | if attrname == name: |
| Scott Baker | 6dfde21 | 2016-03-01 20:10:24 -0800 | [diff] [blame] | 51 | return default |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 52 | if hasattr(cls, "default_attributes"): |
| Scott Baker | f180ff3 | 2016-03-31 11:46:02 -0700 | [diff] [blame] | 53 | if name in cls.default_attributes: |
| 54 | return cls.default_attributes[name] |
| 55 | |
| 56 | return None |
| Scott Baker | 6dfde21 | 2016-03-01 20:10:24 -0800 | [diff] [blame] | 57 | |
| 58 | @classmethod |
| Scott Baker | 74404fe | 2015-07-13 13:54:06 -0700 | [diff] [blame] | 59 | def setup_simple_attributes(cls): |
| 60 | for (attrname, default) in cls.simple_attributes: |
| Scott Baker | 096dce8 | 2015-07-13 14:27:51 -0700 | [diff] [blame] | 61 | setattr(cls, attrname, property(lambda self, attrname=attrname, default=default: self.get_attribute(attrname, default), |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 62 | lambda self, value, attrname=attrname: self.set_attribute( |
| 63 | attrname, value), |
| Scott Baker | 096dce8 | 2015-07-13 14:27:51 -0700 | [diff] [blame] | 64 | None, |
| 65 | attrname)) |
| Scott Baker | 74404fe | 2015-07-13 13:54:06 -0700 | [diff] [blame] | 66 | |
| Scott Baker | 80d65b1 | 2016-05-26 20:20:47 -0700 | [diff] [blame] | 67 | class ServiceController(PlCoreBase): |
| 68 | xos = models.ForeignKey(XOS, related_name='servicecontrolers', help_text="Pointer to XOS", default=get_xos) |
| 69 | name = StrippedCharField(max_length=30, help_text="Service Name") |
| Scott Baker | 1e55982 | 2016-05-26 20:57:10 -0700 | [diff] [blame] | 70 | base_url = StrippedCharField(max_length=1024, help_text="Base URL, allows use of relative URLs for resources", null=True, blank=True) |
| Scott Baker | 80d65b1 | 2016-05-26 20:20:47 -0700 | [diff] [blame] | 71 | |
| Scott Baker | 147cace | 2016-05-26 21:11:37 -0700 | [diff] [blame] | 72 | def __unicode__(self): return u'%s' % (self.name) |
| 73 | |
| Scott Baker | 80d65b1 | 2016-05-26 20:20:47 -0700 | [diff] [blame] | 74 | class ServiceControllerResource(PlCoreBase): |
| 75 | KIND_CHOICES = (('models', 'Models'), |
| 76 | ('admin', 'Admin'), |
| 77 | ('django_library', 'Django Library'), |
| Scott Baker | 1ea8cb5 | 2016-05-26 22:19:35 -0700 | [diff] [blame^] | 78 | ('synchronizer', 'Synchronizer'), |
| Scott Baker | 80d65b1 | 2016-05-26 20:20:47 -0700 | [diff] [blame] | 79 | ('rest', 'REST API'), |
| 80 | ('tosca_custom_types', 'Tosca Custom Types'), |
| 81 | ('tosca_resource', 'Tosca Resource')) |
| 82 | |
| 83 | FORMAT_CHOICES = (('python', 'Python'), |
| 84 | ('manifest', 'Manifest'), |
| 85 | ('docker', 'Docker Container')) |
| 86 | |
| 87 | service_controller = models.ForeignKey(ServiceController, related_name='servicecontrolerresources', |
| 88 | help_text="The Service Controller this resource is associated with") |
| 89 | |
| 90 | name = StrippedCharField(max_length=30, help_text="Object Name") |
| 91 | kind = StrippedCharField(choices=KIND_CHOICES, max_length=30) |
| 92 | format = StrippedCharField(choices=FORMAT_CHOICES, max_length=30) |
| Scott Baker | 1e55982 | 2016-05-26 20:57:10 -0700 | [diff] [blame] | 93 | url = StrippedCharField(max_length=1024, help_text="URL of resource", null=True, blank=True) |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 94 | |
| Scott Baker | 147cace | 2016-05-26 21:11:37 -0700 | [diff] [blame] | 95 | def __unicode__(self): return u'%s' % (self.name) |
| 96 | |
| Scott Baker | 82498c5 | 2015-07-13 13:07:27 -0700 | [diff] [blame] | 97 | class Service(PlCoreBase, AttributeMixin): |
| Scott Baker | 008a996 | 2015-04-15 20:58:20 -0700 | [diff] [blame] | 98 | # when subclassing a service, redefine KIND to describe the new service |
| 99 | KIND = "generic" |
| 100 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 101 | description = models.TextField( |
| 102 | max_length=254, null=True, blank=True, help_text="Description of Service") |
| Siobhan Tully | 00353f7 | 2013-10-08 21:53:27 -0400 | [diff] [blame] | 103 | enabled = models.BooleanField(default=True) |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 104 | kind = StrippedCharField( |
| 105 | max_length=30, help_text="Kind of service", default=KIND) |
| Tony Mack | d84b1ff | 2015-03-09 13:03:56 -0400 | [diff] [blame] | 106 | name = StrippedCharField(max_length=30, help_text="Service Name") |
| Scott Baker | d3e3029 | 2016-05-02 09:38:24 -0700 | [diff] [blame] | 107 | versionNumber = StrippedCharField(blank=True, null=True, |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 108 | max_length=30, help_text="Version of Service Definition") |
| Siobhan Tully | cf04fb6 | 2014-01-11 11:25:57 -0500 | [diff] [blame] | 109 | published = models.BooleanField(default=True) |
| Tony Mack | d84b1ff | 2015-03-09 13:03:56 -0400 | [diff] [blame] | 110 | view_url = StrippedCharField(blank=True, null=True, max_length=1024) |
| 111 | icon_url = StrippedCharField(blank=True, null=True, max_length=1024) |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 112 | public_key = models.TextField( |
| 113 | null=True, blank=True, max_length=1024, help_text="Public key string") |
| Scott Baker | f60c010 | 2015-11-12 16:22:52 -0800 | [diff] [blame] | 114 | private_key_fn = StrippedCharField(blank=True, null=True, max_length=1024) |
| Siobhan Tully | 00353f7 | 2013-10-08 21:53:27 -0400 | [diff] [blame] | 115 | |
| Scott Baker | 2f0828e | 2015-07-13 12:33:28 -0700 | [diff] [blame] | 116 | # Service_specific_attribute and service_specific_id are opaque to XOS |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 117 | service_specific_id = StrippedCharField( |
| 118 | max_length=30, blank=True, null=True) |
| Scott Baker | 2f0828e | 2015-07-13 12:33:28 -0700 | [diff] [blame] | 119 | service_specific_attribute = models.TextField(blank=True, null=True) |
| 120 | |
| Scott Baker | 80d65b1 | 2016-05-26 20:20:47 -0700 | [diff] [blame] | 121 | controller = models.ForeignKey(ServiceController, related_name='services', |
| 122 | help_text="The Service Controller this Service uses") |
| 123 | |
| Scott Baker | 008a996 | 2015-04-15 20:58:20 -0700 | [diff] [blame] | 124 | def __init__(self, *args, **kwargs): |
| 125 | # for subclasses, set the default kind appropriately |
| 126 | self._meta.get_field("kind").default = self.KIND |
| 127 | super(Service, self).__init__(*args, **kwargs) |
| 128 | |
| 129 | @classmethod |
| 130 | def get_service_objects(cls): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 131 | return cls.objects.filter(kind=cls.KIND) |
| Scott Baker | 008a996 | 2015-04-15 20:58:20 -0700 | [diff] [blame] | 132 | |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 133 | @classmethod |
| Scott Baker | 542cd6f | 2015-10-19 21:18:53 -0700 | [diff] [blame] | 134 | def get_deleted_service_objects(cls): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 135 | return cls.deleted_objects.filter(kind=cls.KIND) |
| Scott Baker | 542cd6f | 2015-10-19 21:18:53 -0700 | [diff] [blame] | 136 | |
| 137 | @classmethod |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 138 | def get_service_objects_by_user(cls, user): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 139 | return cls.select_by_user(user).filter(kind=cls.KIND) |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 140 | |
| 141 | @classmethod |
| 142 | def select_by_user(cls, user): |
| 143 | if user.is_admin: |
| 144 | return cls.objects.all() |
| 145 | else: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 146 | service_ids = [ |
| 147 | sp.slice.id for sp in ServicePrivilege.objects.filter(user=user)] |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 148 | return cls.objects.filter(id__in=service_ids) |
| 149 | |
| Scott Baker | cce158d | 2015-12-07 22:20:40 -0800 | [diff] [blame] | 150 | @property |
| 151 | def serviceattribute_dict(self): |
| 152 | attrs = {} |
| 153 | for attr in self.serviceattributes.all(): |
| 154 | attrs[attr.name] = attr.value |
| 155 | return attrs |
| 156 | |
| Siobhan Tully | 00353f7 | 2013-10-08 21:53:27 -0400 | [diff] [blame] | 157 | def __unicode__(self): return u'%s' % (self.name) |
| 158 | |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 159 | def can_update(self, user): |
| 160 | return user.can_update_service(self, allow=['admin']) |
| Scott Baker | 618a489 | 2015-07-06 14:27:31 -0700 | [diff] [blame] | 161 | |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 162 | def get_scalable_nodes(self, slice, max_per_node=None, exclusive_slices=[]): |
| 163 | """ |
| 164 | Get a list of nodes that can be used to scale up a slice. |
| 165 | |
| 166 | slice - slice to scale up |
| Tony Mack | 3de59e3 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 167 | max_per_node - maximum numbers of instances that 'slice' can have on a single node |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 168 | exclusive_slices - list of slices that must have no nodes in common with 'slice'. |
| 169 | """ |
| 170 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 171 | # late import to get around order-of-imports constraint in __init__.py |
| 172 | from core.models import Node, Instance |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 173 | |
| 174 | nodes = list(Node.objects.all()) |
| 175 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 176 | conflicting_instances = Instance.objects.filter( |
| 177 | slice__in=exclusive_slices) |
| 178 | conflicting_nodes = Node.objects.filter( |
| 179 | instances__in=conflicting_instances) |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 180 | |
| 181 | nodes = [x for x in nodes if x not in conflicting_nodes] |
| 182 | |
| Tony Mack | 3de59e3 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 183 | # If max_per_node is set, then limit the number of instances this slice |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 184 | # can have on a single node. |
| 185 | if max_per_node: |
| 186 | acceptable_nodes = [] |
| 187 | for node in nodes: |
| Tony Mack | 3de59e3 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 188 | existing_count = node.instances.filter(slice=slice).count() |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 189 | if existing_count < max_per_node: |
| 190 | acceptable_nodes.append(node) |
| 191 | nodes = acceptable_nodes |
| 192 | |
| 193 | return nodes |
| 194 | |
| 195 | def pick_node(self, slice, max_per_node=None, exclusive_slices=[]): |
| 196 | # Pick the best node to scale up a slice. |
| 197 | |
| 198 | nodes = self.get_scalable_nodes(slice, max_per_node, exclusive_slices) |
| Tony Mack | 3de59e3 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 199 | nodes = sorted(nodes, key=lambda node: node.instances.all().count()) |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 200 | if not nodes: |
| 201 | return None |
| 202 | return nodes[0] |
| 203 | |
| 204 | def adjust_scale(self, slice_hint, scale, max_per_node=None, exclusive_slices=[]): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 205 | # late import to get around order-of-imports constraint in __init__.py |
| 206 | from core.models import Instance |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 207 | |
| 208 | slices = [x for x in self.slices.all() if slice_hint in x.name] |
| 209 | for slice in slices: |
| Tony Mack | 3de59e3 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 210 | while slice.instances.all().count() > scale: |
| 211 | s = slice.instances.all()[0] |
| 212 | # print "drop instance", s |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 213 | s.delete() |
| 214 | |
| Tony Mack | 3de59e3 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 215 | while slice.instances.all().count() < scale: |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 216 | node = self.pick_node(slice, max_per_node, exclusive_slices) |
| 217 | if not node: |
| 218 | # no more available nodes |
| 219 | break |
| 220 | |
| 221 | image = slice.default_image |
| 222 | if not image: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 223 | raise XOSConfigurationError( |
| 224 | "No default_image for slice %s" % slice.name) |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 225 | |
| 226 | flavor = slice.default_flavor |
| 227 | if not flavor: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 228 | raise XOSConfigurationError( |
| 229 | "No default_flavor for slice %s" % slice.name) |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 230 | |
| Tony Mack | 3de59e3 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 231 | s = Instance(slice=slice, |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 232 | node=node, |
| 233 | creator=slice.creator, |
| 234 | image=image, |
| 235 | flavor=flavor, |
| 236 | deployment=node.site_deployment.deployment) |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 237 | s.save() |
| 238 | |
| Tony Mack | 3de59e3 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 239 | # print "add instance", s |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 240 | |
| Scott Baker | 8a67ae7 | 2015-12-09 22:54:52 -0800 | [diff] [blame] | 241 | def get_vtn_src_nets(self): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 242 | nets = [] |
| Scott Baker | 8a67ae7 | 2015-12-09 22:54:52 -0800 | [diff] [blame] | 243 | for slice in self.slices.all(): |
| 244 | for ns in slice.networkslices.all(): |
| 245 | if not ns.network: |
| 246 | continue |
| Scott Baker | f63cf1c | 2016-01-12 19:59:12 -0800 | [diff] [blame] | 247 | # if ns.network.template.access in ["direct", "indirect"]: |
| 248 | # # skip access networks; we want to use the private network |
| 249 | # continue |
| Scott Baker | e6e7855 | 2016-04-11 21:26:39 -0700 | [diff] [blame] | 250 | if "management" in ns.network.name: |
| 251 | # don't try to connect the management network to anything |
| 252 | continue |
| Scott Baker | 8a67ae7 | 2015-12-09 22:54:52 -0800 | [diff] [blame] | 253 | if ns.network.name in ["wan_network", "lan_network"]: |
| 254 | # we don't want to attach to the vCPE's lan or wan network |
| 255 | # we only want to attach to its private network |
| 256 | # TODO: fix hard-coding of network name |
| 257 | continue |
| 258 | for cn in ns.network.controllernetworks.all(): |
| 259 | if cn.net_id: |
| 260 | net = {"name": ns.network.name, "net_id": cn.net_id} |
| 261 | nets.append(net) |
| 262 | return nets |
| 263 | |
| Scott Baker | cbd718e | 2015-12-08 21:31:18 -0800 | [diff] [blame] | 264 | def get_vtn_nets(self): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 265 | nets = [] |
| Scott Baker | cbd718e | 2015-12-08 21:31:18 -0800 | [diff] [blame] | 266 | for slice in self.slices.all(): |
| 267 | for ns in slice.networkslices.all(): |
| 268 | if not ns.network: |
| 269 | continue |
| Scott Baker | 8a67ae7 | 2015-12-09 22:54:52 -0800 | [diff] [blame] | 270 | if ns.network.template.access not in ["direct", "indirect"]: |
| 271 | # skip anything that's not an access network |
| 272 | continue |
| Scott Baker | cbd718e | 2015-12-08 21:31:18 -0800 | [diff] [blame] | 273 | for cn in ns.network.controllernetworks.all(): |
| 274 | if cn.net_id: |
| 275 | net = {"name": ns.network.name, "net_id": cn.net_id} |
| 276 | nets.append(net) |
| 277 | return nets |
| 278 | |
| 279 | def get_vtn_dependencies_nets(self): |
| 280 | provider_nets = [] |
| Scott Baker | 5380a44 | 2015-12-08 19:27:50 -0800 | [diff] [blame] | 281 | for tenant in self.subscribed_tenants.all(): |
| 282 | if tenant.provider_service: |
| Scott Baker | cbd718e | 2015-12-08 21:31:18 -0800 | [diff] [blame] | 283 | for net in tenant.provider_service.get_vtn_nets(): |
| 284 | if not net in provider_nets: |
| 285 | provider_nets.append(net) |
| 286 | return provider_nets |
| 287 | |
| 288 | def get_vtn_dependencies_ids(self): |
| 289 | return [x["net_id"] for x in self.get_vtn_dependencies_nets()] |
| 290 | |
| 291 | def get_vtn_dependencies_names(self): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 292 | return [x["name"] + "_" + x["net_id"] for x in self.get_vtn_dependencies_nets()] |
| Scott Baker | cbd718e | 2015-12-08 21:31:18 -0800 | [diff] [blame] | 293 | |
| Scott Baker | 8a67ae7 | 2015-12-09 22:54:52 -0800 | [diff] [blame] | 294 | def get_vtn_src_ids(self): |
| 295 | return [x["net_id"] for x in self.get_vtn_src_nets()] |
| Scott Baker | cbd718e | 2015-12-08 21:31:18 -0800 | [diff] [blame] | 296 | |
| Scott Baker | 8a67ae7 | 2015-12-09 22:54:52 -0800 | [diff] [blame] | 297 | def get_vtn_src_names(self): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 298 | return [x["name"] + "_" + x["net_id"] for x in self.get_vtn_src_nets()] |
| Scott Baker | 5380a44 | 2015-12-08 19:27:50 -0800 | [diff] [blame] | 299 | |
| 300 | |
| Siobhan Tully | 00353f7 | 2013-10-08 21:53:27 -0400 | [diff] [blame] | 301 | class ServiceAttribute(PlCoreBase): |
| Scott Baker | cce158d | 2015-12-07 22:20:40 -0800 | [diff] [blame] | 302 | name = models.CharField(help_text="Attribute Name", max_length=128) |
| Tony Mack | d84b1ff | 2015-03-09 13:03:56 -0400 | [diff] [blame] | 303 | value = StrippedCharField(help_text="Attribute Value", max_length=1024) |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 304 | service = models.ForeignKey(Service, related_name='serviceattributes', |
| 305 | help_text="The Service this attribute is associated with") |
| 306 | |
| Siobhan Tully | 00353f7 | 2013-10-08 21:53:27 -0400 | [diff] [blame] | 307 | |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 308 | class ServiceRole(PlCoreBase): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 309 | ROLE_CHOICES = (('admin', 'Admin'),) |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 310 | role = StrippedCharField(choices=ROLE_CHOICES, unique=True, max_length=30) |
| 311 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 312 | def __unicode__(self): return u'%s' % (self.role) |
| 313 | |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 314 | |
| 315 | class ServicePrivilege(PlCoreBase): |
| 316 | user = models.ForeignKey('User', related_name='serviceprivileges') |
| 317 | service = models.ForeignKey('Service', related_name='serviceprivileges') |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 318 | role = models.ForeignKey('ServiceRole', related_name='serviceprivileges') |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 319 | |
| 320 | class Meta: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 321 | unique_together = ('user', 'service', 'role') |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 322 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 323 | def __unicode__(self): return u'%s %s %s' % ( |
| 324 | self.service, self.user, self.role) |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 325 | |
| 326 | def can_update(self, user): |
| 327 | if not self.service.enabled: |
| 328 | raise PermissionDenied, "Cannot modify permission(s) of a disabled service" |
| 329 | return self.service.can_update(user) |
| 330 | |
| 331 | def save(self, *args, **kwds): |
| 332 | if not self.service.enabled: |
| 333 | raise PermissionDenied, "Cannot modify permission(s) of a disabled service" |
| 334 | super(ServicePrivilege, self).save(*args, **kwds) |
| 335 | |
| 336 | def delete(self, *args, **kwds): |
| 337 | if not self.service.enabled: |
| 338 | raise PermissionDenied, "Cannot modify permission(s) of a disabled service" |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 339 | super(ServicePrivilege, self).delete(*args, **kwds) |
| 340 | |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 341 | @classmethod |
| 342 | def select_by_user(cls, user): |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 343 | if user.is_admin: |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 344 | qs = cls.objects.all() |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 345 | else: |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 346 | qs = cls.objects.filter(user=user) |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 347 | return qs |
| 348 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 349 | |
| Scott Baker | 82498c5 | 2015-07-13 13:07:27 -0700 | [diff] [blame] | 350 | class TenantRoot(PlCoreBase, AttributeMixin): |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 351 | """ A tenantRoot is one of the things that can sit at the root of a chain |
| 352 | of tenancy. This object represents a node. |
| 353 | """ |
| 354 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 355 | KIND = "generic" |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 356 | kind = StrippedCharField(max_length=30, default=KIND) |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 357 | name = StrippedCharField( |
| 358 | max_length=255, help_text="name", blank=True, null=True) |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 359 | |
| Scott Baker | efcec63 | 2015-07-07 12:12:42 -0700 | [diff] [blame] | 360 | service_specific_attribute = models.TextField(blank=True, null=True) |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 361 | service_specific_id = StrippedCharField( |
| 362 | max_length=30, blank=True, null=True) |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 363 | |
| Scott Baker | db66fd3 | 2015-07-07 17:59:44 -0700 | [diff] [blame] | 364 | def __init__(self, *args, **kwargs): |
| 365 | # for subclasses, set the default kind appropriately |
| 366 | self._meta.get_field("kind").default = self.KIND |
| 367 | super(TenantRoot, self).__init__(*args, **kwargs) |
| 368 | |
| Scott Baker | 618a489 | 2015-07-06 14:27:31 -0700 | [diff] [blame] | 369 | def __unicode__(self): |
| 370 | if not self.name: |
| 371 | return u"%s-tenant_root-#%s" % (str(self.kind), str(self.id)) |
| 372 | else: |
| 373 | return self.name |
| 374 | |
| 375 | def can_update(self, user): |
| 376 | return user.can_update_tenant_root(self, allow=['admin']) |
| 377 | |
| Scott Baker | efcec63 | 2015-07-07 12:12:42 -0700 | [diff] [blame] | 378 | def get_subscribed_tenants(self, tenant_class): |
| 379 | ids = self.subscribed_tenants.filter(kind=tenant_class.KIND) |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 380 | return tenant_class.objects.filter(id__in=ids) |
| Scott Baker | efcec63 | 2015-07-07 12:12:42 -0700 | [diff] [blame] | 381 | |
| 382 | def get_newest_subscribed_tenant(self, kind): |
| 383 | st = list(self.get_subscribed_tenants(kind)) |
| 384 | if not st: |
| 385 | return None |
| 386 | return sorted(st, key=attrgetter('id'))[0] |
| 387 | |
| 388 | @classmethod |
| 389 | def get_tenant_objects(cls): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 390 | return cls.objects.filter(kind=cls.KIND) |
| Scott Baker | efcec63 | 2015-07-07 12:12:42 -0700 | [diff] [blame] | 391 | |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 392 | @classmethod |
| 393 | def get_tenant_objects_by_user(cls, user): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 394 | return cls.select_by_user(user).filter(kind=cls.KIND) |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 395 | |
| 396 | @classmethod |
| 397 | def select_by_user(cls, user): |
| 398 | if user.is_admin: |
| 399 | return cls.objects.all() |
| 400 | else: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 401 | tr_ids = [ |
| 402 | trp.tenant_root.id for trp in TenantRootPrivilege.objects.filter(user=user)] |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 403 | return cls.objects.filter(id__in=tr_ids) |
| 404 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 405 | # helper function to be used in subclasses that want to ensure |
| 406 | # service_specific_id is unique |
| Scott Baker | dc880f3 | 2016-03-31 14:45:31 -0700 | [diff] [blame] | 407 | def validate_unique_service_specific_id(self, none_okay=False): |
| 408 | if not none_okay and (self.service_specific_id is None): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 409 | raise XOSMissingField("subscriber_specific_id is None, and it's a required field", fields={ |
| 410 | "service_specific_id": "cannot be none"}) |
| Scott Baker | dc880f3 | 2016-03-31 14:45:31 -0700 | [diff] [blame] | 411 | |
| 412 | if self.service_specific_id: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 413 | conflicts = self.get_tenant_objects().filter( |
| 414 | service_specific_id=self.service_specific_id) |
| Scott Baker | dc880f3 | 2016-03-31 14:45:31 -0700 | [diff] [blame] | 415 | if self.pk: |
| Scott Baker | e4c9d87 | 2016-04-01 16:28:41 -0700 | [diff] [blame] | 416 | conflicts = conflicts.exclude(pk=self.pk) |
| Scott Baker | dc880f3 | 2016-03-31 14:45:31 -0700 | [diff] [blame] | 417 | if conflicts: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 418 | raise XOSDuplicateKey("service_specific_id %s already exists" % self.service_specific_id, fields={ |
| 419 | "service_specific_id": "duplicate key"}) |
| 420 | |
| Scott Baker | dc880f3 | 2016-03-31 14:45:31 -0700 | [diff] [blame] | 421 | |
| Scott Baker | 82498c5 | 2015-07-13 13:07:27 -0700 | [diff] [blame] | 422 | class Tenant(PlCoreBase, AttributeMixin): |
| Scott Baker | 8103d0f | 2015-04-10 16:42:26 -0700 | [diff] [blame] | 423 | """ A tenant is a relationship between two entities, a subscriber and a |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 424 | provider. This object represents an edge. |
| Scott Baker | 8103d0f | 2015-04-10 16:42:26 -0700 | [diff] [blame] | 425 | |
| 426 | The subscriber can be a User, a Service, or a Tenant. |
| 427 | |
| 428 | The provider is always a Service. |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 429 | |
| 430 | TODO: rename "Tenant" to "Tenancy" |
| Scott Baker | 8103d0f | 2015-04-10 16:42:26 -0700 | [diff] [blame] | 431 | """ |
| Scott Baker | 008a996 | 2015-04-15 20:58:20 -0700 | [diff] [blame] | 432 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 433 | CONNECTIVITY_CHOICES = (('public', 'Public'), |
| 434 | ('private', 'Private'), ('na', 'Not Applicable')) |
| Scott Baker | 925a8fa | 2015-04-26 20:30:40 -0700 | [diff] [blame] | 435 | |
| Scott Baker | 008a996 | 2015-04-15 20:58:20 -0700 | [diff] [blame] | 436 | # when subclassing a service, redefine KIND to describe the new service |
| 437 | KIND = "generic" |
| 438 | |
| 439 | kind = StrippedCharField(max_length=30, default=KIND) |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 440 | provider_service = models.ForeignKey( |
| 441 | Service, related_name='provided_tenants') |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 442 | |
| 443 | # The next four things are the various type of objects that can be subscribers of this Tenancy |
| 444 | # relationship. One and only one can be used at a time. |
| Scott Baker | 97ca910 | 2016-04-13 16:56:39 -0700 | [diff] [blame] | 445 | # XXX these should really be changed to GenericForeignKey |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 446 | subscriber_service = models.ForeignKey( |
| 447 | Service, related_name='subscribed_tenants', blank=True, null=True) |
| 448 | subscriber_tenant = models.ForeignKey( |
| 449 | "Tenant", related_name='subscribed_tenants', blank=True, null=True) |
| 450 | subscriber_user = models.ForeignKey( |
| 451 | "User", related_name='subscribed_tenants', blank=True, null=True) |
| 452 | subscriber_root = models.ForeignKey( |
| 453 | "TenantRoot", related_name="subscribed_tenants", blank=True, null=True) |
| 454 | subscriber_network = models.ForeignKey( |
| 455 | "Network", related_name="subscribed_tenants", blank=True, null=True) |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 456 | |
| 457 | # Service_specific_attribute and service_specific_id are opaque to XOS |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 458 | service_specific_id = StrippedCharField( |
| 459 | max_length=30, blank=True, null=True) |
| Scott Baker | 76934d8 | 2015-05-06 19:49:31 -0700 | [diff] [blame] | 460 | service_specific_attribute = models.TextField(blank=True, null=True) |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 461 | |
| 462 | # Connect_method is only used by Coarse tenants |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 463 | connect_method = models.CharField( |
| 464 | null=False, blank=False, max_length=30, choices=CONNECTIVITY_CHOICES, default="na") |
| Scott Baker | 8103d0f | 2015-04-10 16:42:26 -0700 | [diff] [blame] | 465 | |
| Scott Baker | 008a996 | 2015-04-15 20:58:20 -0700 | [diff] [blame] | 466 | def __init__(self, *args, **kwargs): |
| 467 | # for subclasses, set the default kind appropriately |
| 468 | self._meta.get_field("kind").default = self.KIND |
| 469 | super(Tenant, self).__init__(*args, **kwargs) |
| 470 | |
| Scott Baker | 8103d0f | 2015-04-10 16:42:26 -0700 | [diff] [blame] | 471 | def __unicode__(self): |
| Scott Baker | f996b76 | 2015-05-20 20:42:04 -0700 | [diff] [blame] | 472 | return u"%s-tenant-%s" % (str(self.kind), str(self.id)) |
| Scott Baker | 8103d0f | 2015-04-10 16:42:26 -0700 | [diff] [blame] | 473 | |
| Scott Baker | 008a996 | 2015-04-15 20:58:20 -0700 | [diff] [blame] | 474 | @classmethod |
| 475 | def get_tenant_objects(cls): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 476 | return cls.objects.filter(kind=cls.KIND) |
| Scott Baker | 008a996 | 2015-04-15 20:58:20 -0700 | [diff] [blame] | 477 | |
| Scott Baker | e7fc9f5 | 2015-05-05 17:52:03 -0700 | [diff] [blame] | 478 | @classmethod |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 479 | def get_tenant_objects_by_user(cls, user): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 480 | return cls.select_by_user(user).filter(kind=cls.KIND) |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 481 | |
| 482 | @classmethod |
| Scott Baker | e7fc9f5 | 2015-05-05 17:52:03 -0700 | [diff] [blame] | 483 | def get_deleted_tenant_objects(cls): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 484 | return cls.deleted_objects.filter(kind=cls.KIND) |
| Scott Baker | e7fc9f5 | 2015-05-05 17:52:03 -0700 | [diff] [blame] | 485 | |
| Scott Baker | cce158d | 2015-12-07 22:20:40 -0800 | [diff] [blame] | 486 | @property |
| 487 | def tenantattribute_dict(self): |
| 488 | attrs = {} |
| 489 | for attr in self.tenantattributes.all(): |
| 490 | attrs[attr.name] = attr.value |
| 491 | return attrs |
| 492 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 493 | # helper function to be used in subclasses that want to ensure |
| 494 | # service_specific_id is unique |
| Scott Baker | d921e1c | 2015-04-20 14:24:29 -0700 | [diff] [blame] | 495 | def validate_unique_service_specific_id(self): |
| 496 | if self.pk is None: |
| 497 | if self.service_specific_id is None: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 498 | raise XOSMissingField("subscriber_specific_id is None, and it's a required field", fields={ |
| 499 | "service_specific_id": "cannot be none"}) |
| Scott Baker | d921e1c | 2015-04-20 14:24:29 -0700 | [diff] [blame] | 500 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 501 | conflicts = self.get_tenant_objects().filter( |
| 502 | service_specific_id=self.service_specific_id) |
| Scott Baker | d921e1c | 2015-04-20 14:24:29 -0700 | [diff] [blame] | 503 | if conflicts: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 504 | raise XOSDuplicateKey("service_specific_id %s already exists" % self.service_specific_id, fields={ |
| 505 | "service_specific_id": "duplicate key"}) |
| Scott Baker | d921e1c | 2015-04-20 14:24:29 -0700 | [diff] [blame] | 506 | |
| Scott Baker | 618a489 | 2015-07-06 14:27:31 -0700 | [diff] [blame] | 507 | def save(self, *args, **kwargs): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 508 | subCount = sum([1 for e in [self.subscriber_service, self.subscriber_tenant, |
| 509 | self.subscriber_user, self.subscriber_root] if e is not None]) |
| Scott Baker | 618a489 | 2015-07-06 14:27:31 -0700 | [diff] [blame] | 510 | if (subCount > 1): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 511 | raise XOSConflictingField( |
| 512 | "Only one of subscriber_service, subscriber_tenant, subscriber_user, subscriber_root should be set") |
| Scott Baker | 618a489 | 2015-07-06 14:27:31 -0700 | [diff] [blame] | 513 | |
| 514 | super(Tenant, self).save(*args, **kwargs) |
| 515 | |
| 516 | def get_subscribed_tenants(self, tenant_class): |
| 517 | ids = self.subscribed_tenants.filter(kind=tenant_class.KIND) |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 518 | return tenant_class.objects.filter(id__in=ids) |
| Scott Baker | 618a489 | 2015-07-06 14:27:31 -0700 | [diff] [blame] | 519 | |
| 520 | def get_newest_subscribed_tenant(self, kind): |
| 521 | st = list(self.get_subscribed_tenants(kind)) |
| 522 | if not st: |
| 523 | return None |
| 524 | return sorted(st, key=attrgetter('id'))[0] |
| 525 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 526 | |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 527 | class Scheduler(object): |
| 528 | # XOS Scheduler Abstract Base Class |
| 529 | # Used to implement schedulers that pick which node to put instances on |
| 530 | |
| 531 | def __init__(self, slice): |
| 532 | self.slice = slice |
| 533 | |
| 534 | def pick(self): |
| 535 | # this method should return a tuple (node, parent) |
| 536 | # node is the node to instantiate on |
| 537 | # parent is for container_vm instances only, and is the VM that will |
| 538 | # hold the container |
| 539 | |
| 540 | raise Exception("Abstract Base") |
| 541 | |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 542 | |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 543 | class LeastLoadedNodeScheduler(Scheduler): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 544 | # This scheduler always return the node with the fewest number of |
| 545 | # instances. |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 546 | |
| Scott Baker | 6526f55 | 2016-03-04 11:29:02 -0800 | [diff] [blame] | 547 | def __init__(self, slice, label=None): |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 548 | super(LeastLoadedNodeScheduler, self).__init__(slice) |
| Scott Baker | 6526f55 | 2016-03-04 11:29:02 -0800 | [diff] [blame] | 549 | self.label = label |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 550 | |
| 551 | def pick(self): |
| 552 | from core.models import Node |
| Pingping Lin | fa30bae | 2016-03-03 09:52:24 -0800 | [diff] [blame] | 553 | if not self.slice.default_node: |
| 554 | nodes = list(Node.objects.all()) |
| 555 | nodes = sorted(nodes, key=lambda node: node.instances.all().count()) |
| 556 | else: |
| 557 | nodes = list(Node.objects.filter(name = self.slice.default_node)) |
| Matteo Scandolo | 1d7ff81 | 2016-04-21 09:02:08 -0700 | [diff] [blame] | 558 | |
| Scott Baker | 6526f55 | 2016-03-04 11:29:02 -0800 | [diff] [blame] | 559 | if self.label: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 560 | nodes = nodes.filter(nodelabels__name=self.label) |
| Scott Baker | 6526f55 | 2016-03-04 11:29:02 -0800 | [diff] [blame] | 561 | |
| 562 | nodes = list(nodes) |
| 563 | |
| 564 | if not nodes: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 565 | raise Exception( |
| 566 | "LeastLoadedNodeScheduler: No suitable nodes to pick from") |
| Scott Baker | cce158d | 2015-12-07 22:20:40 -0800 | [diff] [blame] | 567 | |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 568 | # TODO: logic to filter nodes by which nodes are up, and which |
| 569 | # nodes the slice can instantiate on. |
| Pingping Lin | fa30bae | 2016-03-03 09:52:24 -0800 | [diff] [blame] | 570 | # nodes = sorted(nodes, key=lambda node: node.instances.all().count()) |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 571 | return [nodes[0], None] |
| 572 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 573 | |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 574 | class ContainerVmScheduler(Scheduler): |
| 575 | # This scheduler picks a VM in the slice with the fewest containers inside |
| 576 | # of it. If no VMs are suitable, then it creates a VM. |
| 577 | |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 578 | MAX_VM_PER_CONTAINER = 10 |
| 579 | |
| 580 | def __init__(self, slice): |
| 581 | super(ContainerVmScheduler, self).__init__(slice) |
| 582 | |
| 583 | @property |
| 584 | def image(self): |
| 585 | from core.models import Image |
| 586 | |
| Scott Baker | a8188a7 | 2016-05-19 17:54:52 -0700 | [diff] [blame] | 587 | # If slice has default_image set then use it |
| 588 | if self.slice.default_image: |
| 589 | return self.slice.default_image |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 590 | |
| Scott Baker | a8188a7 | 2016-05-19 17:54:52 -0700 | [diff] [blame] | 591 | raise XOPSProgrammingError("Please set a default image for %s" % self.slice.name) |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 592 | |
| 593 | def make_new_instance(self): |
| 594 | from core.models import Instance, Flavor |
| 595 | |
| 596 | flavors = Flavor.objects.filter(name="m1.small") |
| 597 | if not flavors: |
| 598 | raise XOSConfigurationError("No m1.small flavor") |
| 599 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 600 | (node, parent) = LeastLoadedNodeScheduler(self.slice).pick() |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 601 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 602 | instance = Instance(slice=self.slice, |
| 603 | node=node, |
| 604 | image=self.image, |
| 605 | creator=self.slice.creator, |
| 606 | deployment=node.site_deployment.deployment, |
| 607 | flavor=flavors[0], |
| 608 | isolation="vm", |
| 609 | parent=parent) |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 610 | instance.save() |
| 611 | # We rely on a special naming convention to identify the VMs that will |
| 612 | # hole containers. |
| 613 | instance.name = "%s-outer-%s" % (instance.slice.name, instance.id) |
| 614 | instance.save() |
| 615 | return instance |
| 616 | |
| 617 | def pick(self): |
| 618 | from core.models import Instance, Flavor |
| 619 | |
| 620 | for vm in self.slice.instances.filter(isolation="vm"): |
| 621 | avail_vms = [] |
| 622 | if (vm.name.startswith("%s-outer-" % self.slice.name)): |
| 623 | container_count = Instance.objects.filter(parent=vm).count() |
| 624 | if (container_count < self.MAX_VM_PER_CONTAINER): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 625 | avail_vms.append((vm, container_count)) |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 626 | # sort by least containers-per-vm |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 627 | avail_vms = sorted(avail_vms, key=lambda x: x[1]) |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 628 | print "XXX", avail_vms |
| 629 | if avail_vms: |
| 630 | instance = avail_vms[0][0] |
| 631 | return (instance.node, instance) |
| 632 | |
| 633 | instance = self.make_new_instance() |
| 634 | return (instance.node, instance) |
| 635 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 636 | |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 637 | class TenantWithContainer(Tenant): |
| 638 | """ A tenant that manages a container """ |
| 639 | |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 640 | class Meta: |
| 641 | proxy = True |
| 642 | |
| 643 | def __init__(self, *args, **kwargs): |
| 644 | super(TenantWithContainer, self).__init__(*args, **kwargs) |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 645 | self.cached_instance = None |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 646 | self.orig_instance_id = self.get_initial_attribute("instance_id") |
| Scott Baker | e274f6b | 2015-11-02 20:54:28 -0800 | [diff] [blame] | 647 | |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 648 | @property |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 649 | def instance(self): |
| 650 | from core.models import Instance |
| 651 | if getattr(self, "cached_instance", None): |
| 652 | return self.cached_instance |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 653 | instance_id = self.get_attribute("instance_id") |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 654 | if not instance_id: |
| 655 | return None |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 656 | instances = Instance.objects.filter(id=instance_id) |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 657 | if not instances: |
| 658 | return None |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 659 | instance = instances[0] |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 660 | instance.caller = self.creator |
| 661 | self.cached_instance = instance |
| 662 | return instance |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 663 | |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 664 | @instance.setter |
| 665 | def instance(self, value): |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 666 | if value: |
| 667 | value = value.id |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 668 | if (value != self.get_attribute("instance_id", None)): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 669 | self.cached_instance = None |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 670 | self.set_attribute("instance_id", value) |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 671 | |
| Scott Baker | e274f6b | 2015-11-02 20:54:28 -0800 | [diff] [blame] | 672 | @property |
| Scott Baker | bdda0be | 2016-02-10 12:23:53 -0800 | [diff] [blame] | 673 | def external_hostname(self): |
| 674 | return self.get_attribute("external_hostname", "") |
| 675 | |
| 676 | @external_hostname.setter |
| 677 | def external_hostname(self, value): |
| 678 | self.set_attribute("external_hostname", value) |
| 679 | |
| 680 | @property |
| 681 | def external_container(self): |
| 682 | return self.get_attribute("external_container", "") |
| 683 | |
| 684 | @external_container.setter |
| 685 | def external_container(self, value): |
| 686 | self.set_attribute("external_container", value) |
| 687 | |
| 688 | @property |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 689 | def creator(self): |
| 690 | from core.models import User |
| 691 | if getattr(self, "cached_creator", None): |
| 692 | return self.cached_creator |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 693 | creator_id = self.get_attribute("creator_id") |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 694 | if not creator_id: |
| 695 | return None |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 696 | users = User.objects.filter(id=creator_id) |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 697 | if not users: |
| 698 | return None |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 699 | user = users[0] |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 700 | self.cached_creator = users[0] |
| 701 | return user |
| 702 | |
| 703 | @creator.setter |
| 704 | def creator(self, value): |
| 705 | if value: |
| 706 | value = value.id |
| 707 | if (value != self.get_attribute("creator_id", None)): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 708 | self.cached_creator = None |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 709 | self.set_attribute("creator_id", value) |
| 710 | |
| 711 | @property |
| 712 | def image(self): |
| 713 | from core.models import Image |
| 714 | # Implement the logic here to pick the image that should be used when |
| 715 | # instantiating the VM that will hold the container. |
| Scott Baker | abad4c9 | 2016-05-25 10:02:29 -0700 | [diff] [blame] | 716 | |
| Scott Baker | cfaf0d3 | 2015-11-16 22:51:02 -0800 | [diff] [blame] | 717 | slice = self.provider_service.slices.all() |
| 718 | if not slice: |
| 719 | raise XOSProgrammingError("provider service has no slice") |
| 720 | slice = slice[0] |
| 721 | |
| Scott Baker | abad4c9 | 2016-05-25 10:02:29 -0700 | [diff] [blame] | 722 | # If slice has default_image set then use it |
| 723 | if slice.default_image: |
| 724 | return slice.default_image |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 725 | |
| Scott Baker | a8188a7 | 2016-05-19 17:54:52 -0700 | [diff] [blame] | 726 | raise XOPSProgrammingError("Please set a default image for %s" % self.slice.name) |
| Scott Baker | e274f6b | 2015-11-02 20:54:28 -0800 | [diff] [blame] | 727 | |
| Scott Baker | cfaf0d3 | 2015-11-16 22:51:02 -0800 | [diff] [blame] | 728 | def save_instance(self, instance): |
| 729 | # Override this function to do custom pre-save or post-save processing, |
| 730 | # such as creating ports for containers. |
| 731 | instance.save() |
| Scott Baker | e274f6b | 2015-11-02 20:54:28 -0800 | [diff] [blame] | 732 | |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 733 | def pick_least_loaded_instance_in_slice(self, slices): |
| 734 | for slice in slices: |
| 735 | if slice.instances.all().count() > 0: |
| 736 | for instance in slice.instances.all(): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 737 | # Pick the first instance that has lesser than 5 tenants |
| 738 | if self.count_of_tenants_of_an_instance(instance) < 5: |
| 739 | return instance |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 740 | return None |
| 741 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 742 | # TODO: Ideally the tenant count for an instance should be maintained using a |
| 743 | # many-to-one relationship attribute, however this model being proxy, it does |
| 744 | # not permit any new attributes to be defined. Find if any better solutions |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 745 | def count_of_tenants_of_an_instance(self, instance): |
| 746 | tenant_count = 0 |
| 747 | for tenant in self.get_tenant_objects().all(): |
| 748 | if tenant.get_attribute("instance_id", None) == instance.id: |
| 749 | tenant_count += 1 |
| 750 | return tenant_count |
| 751 | |
| Scott Baker | cfaf0d3 | 2015-11-16 22:51:02 -0800 | [diff] [blame] | 752 | def manage_container(self): |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 753 | from core.models import Instance, Flavor |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 754 | |
| 755 | if self.deleted: |
| 756 | return |
| 757 | |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 758 | if (self.instance is not None) and (self.instance.image != self.image): |
| 759 | self.instance.delete() |
| 760 | self.instance = None |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 761 | |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 762 | if self.instance is None: |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 763 | if not self.provider_service.slices.count(): |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 764 | raise XOSConfigurationError("The service has no slices") |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 765 | |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 766 | new_instance_created = False |
| 767 | instance = None |
| 768 | if self.get_attribute("use_same_instance_for_multiple_tenants", default=False): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 769 | # Find if any existing instances can be used for this tenant |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 770 | slices = self.provider_service.slices.all() |
| 771 | instance = self.pick_least_loaded_instance_in_slice(slices) |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 772 | |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 773 | if not instance: |
| Srikanth Vavilapalli | 71120a9 | 2015-11-17 13:41:38 -0600 | [diff] [blame] | 774 | slice = self.provider_service.slices.all()[0] |
| Scott Baker | cfaf0d3 | 2015-11-16 22:51:02 -0800 | [diff] [blame] | 775 | |
| Srikanth Vavilapalli | 2ab2d07 | 2016-02-15 01:18:09 -0500 | [diff] [blame] | 776 | flavor = slice.default_flavor |
| 777 | if not flavor: |
| 778 | flavors = Flavor.objects.filter(name="m1.small") |
| 779 | if not flavors: |
| 780 | raise XOSConfigurationError("No m1.small flavor") |
| 781 | flavor = flavors[0] |
| 782 | |
| Srikanth Vavilapalli | 71120a9 | 2015-11-17 13:41:38 -0600 | [diff] [blame] | 783 | if slice.default_isolation == "container_vm": |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 784 | (node, parent) = ContainerVmScheduler(slice).pick() |
| Srikanth Vavilapalli | 71120a9 | 2015-11-17 13:41:38 -0600 | [diff] [blame] | 785 | else: |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 786 | (node, parent) = LeastLoadedNodeScheduler(slice).pick() |
| Scott Baker | cfaf0d3 | 2015-11-16 22:51:02 -0800 | [diff] [blame] | 787 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 788 | instance = Instance(slice=slice, |
| 789 | node=node, |
| 790 | image=self.image, |
| 791 | creator=self.creator, |
| 792 | deployment=node.site_deployment.deployment, |
| 793 | flavor=flavor, |
| 794 | isolation=slice.default_isolation, |
| 795 | parent=parent) |
| Srikanth Vavilapalli | 71120a9 | 2015-11-17 13:41:38 -0600 | [diff] [blame] | 796 | self.save_instance(instance) |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 797 | new_instance_created = True |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 798 | |
| 799 | try: |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 800 | self.instance = instance |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 801 | super(TenantWithContainer, self).save() |
| 802 | except: |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 803 | if new_instance_created: |
| 804 | instance.delete() |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 805 | raise |
| 806 | |
| 807 | def cleanup_container(self): |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 808 | if self.instance: |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 809 | if self.get_attribute("use_same_instance_for_multiple_tenants", default=False): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 810 | # Delete the instance only if this is last tenant in that |
| 811 | # instance |
| 812 | tenant_count = self.count_of_tenants_of_an_instance( |
| 813 | self.instance) |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 814 | if tenant_count == 0: |
| 815 | self.instance.delete() |
| 816 | else: |
| 817 | self.instance.delete() |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 818 | self.instance = None |
| Scott Baker | 618a489 | 2015-07-06 14:27:31 -0700 | [diff] [blame] | 819 | |
| Scott Baker | ee83f2a | 2015-12-10 23:23:07 -0800 | [diff] [blame] | 820 | def save(self, *args, **kwargs): |
| 821 | if (not self.creator) and (hasattr(self, "caller")) and (self.caller): |
| 822 | self.creator = self.caller |
| 823 | super(TenantWithContainer, self).save(*args, **kwargs) |
| 824 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 825 | |
| Scott Baker | 925a8fa | 2015-04-26 20:30:40 -0700 | [diff] [blame] | 826 | class CoarseTenant(Tenant): |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 827 | """ TODO: rename "CoarseTenant" --> "StaticTenant" """ |
| Scott Baker | 925a8fa | 2015-04-26 20:30:40 -0700 | [diff] [blame] | 828 | class Meta: |
| 829 | proxy = True |
| Siobhan Tully | 00353f7 | 2013-10-08 21:53:27 -0400 | [diff] [blame] | 830 | |
| Scott Baker | 2461bec | 2015-08-14 09:10:11 -0700 | [diff] [blame] | 831 | KIND = COARSE_KIND |
| Scott Baker | 925a8fa | 2015-04-26 20:30:40 -0700 | [diff] [blame] | 832 | |
| 833 | def save(self, *args, **kwargs): |
| 834 | if (not self.subscriber_service): |
| 835 | raise XOSValidationError("subscriber_service cannot be null") |
| 836 | if (self.subscriber_tenant or self.subscriber_user): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 837 | raise XOSValidationError( |
| 838 | "subscriber_tenant and subscriber_user must be null") |
| Scott Baker | 925a8fa | 2015-04-26 20:30:40 -0700 | [diff] [blame] | 839 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 840 | super(CoarseTenant, self).save() |
| 841 | |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 842 | |
| 843 | class Subscriber(TenantRoot): |
| 844 | """ Intermediate class for TenantRoots that are to be Subscribers """ |
| 845 | |
| 846 | class Meta: |
| 847 | proxy = True |
| 848 | |
| 849 | KIND = "Subscriber" |
| 850 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 851 | |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 852 | class Provider(TenantRoot): |
| 853 | """ Intermediate class for TenantRoots that are to be Providers """ |
| 854 | |
| 855 | class Meta: |
| 856 | proxy = True |
| 857 | |
| 858 | KIND = "Provider" |
| 859 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 860 | |
| Scott Baker | 462a1d9 | 2015-10-15 15:59:19 -0700 | [diff] [blame] | 861 | class TenantAttribute(PlCoreBase): |
| Scott Baker | 907f991 | 2015-10-20 17:12:36 -0700 | [diff] [blame] | 862 | name = models.CharField(help_text="Attribute Name", max_length=128) |
| Scott Baker | 462a1d9 | 2015-10-15 15:59:19 -0700 | [diff] [blame] | 863 | value = models.TextField(help_text="Attribute Value") |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 864 | tenant = models.ForeignKey(Tenant, related_name='tenantattributes', |
| 865 | help_text="The Tenant this attribute is associated with") |
| Scott Baker | 462a1d9 | 2015-10-15 15:59:19 -0700 | [diff] [blame] | 866 | |
| Scott Baker | 1271a4d | 2016-04-06 14:34:49 -0700 | [diff] [blame] | 867 | def __unicode__(self): return u'%s-%s' % (self.name, self.id) |
| 868 | |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 869 | |
| 870 | class TenantRootRole(PlCoreBase): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 871 | ROLE_CHOICES = (('admin', 'Admin'), ('access', 'Access')) |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 872 | |
| 873 | role = StrippedCharField(choices=ROLE_CHOICES, unique=True, max_length=30) |
| 874 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 875 | def __unicode__(self): return u'%s' % (self.role) |
| 876 | |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 877 | |
| 878 | class TenantRootPrivilege(PlCoreBase): |
| 879 | user = models.ForeignKey('User', related_name="tenant_root_privileges") |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 880 | tenant_root = models.ForeignKey( |
| 881 | 'TenantRoot', related_name="tenant_root_privileges") |
| 882 | role = models.ForeignKey( |
| 883 | 'TenantRootRole', related_name="tenant_root_privileges") |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 884 | |
| 885 | class Meta: |
| 886 | unique_together = ('user', 'tenant_root', 'role') |
| 887 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 888 | def __unicode__(self): return u'%s %s %s' % ( |
| 889 | self.tenant_root, self.user, self.role) |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 890 | |
| 891 | def save(self, *args, **kwds): |
| 892 | if not self.user.is_active: |
| 893 | raise PermissionDenied, "Cannot modify role(s) of a disabled user" |
| Scott Baker | 335882a | 2015-07-24 10:15:31 -0700 | [diff] [blame] | 894 | super(TenantRootPrivilege, self).save(*args, **kwds) |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 895 | |
| 896 | def can_update(self, user): |
| Scott Baker | 335882a | 2015-07-24 10:15:31 -0700 | [diff] [blame] | 897 | return user.can_update_tenant_root_privilege(self) |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 898 | |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 899 | @classmethod |
| 900 | def select_by_user(cls, user): |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 901 | if user.is_admin: |
| Scott Baker | 5116b30 | 2015-07-24 15:48:03 -0700 | [diff] [blame] | 902 | return cls.objects.all() |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 903 | else: |
| Scott Baker | 5116b30 | 2015-07-24 15:48:03 -0700 | [diff] [blame] | 904 | # User can see his own privilege |
| 905 | trp_ids = [trp.id for trp in cls.objects.filter(user=user)] |
| 906 | |
| 907 | # A slice admin can see the SlicePrivileges for his Slice |
| 908 | for priv in cls.objects.filter(user=user, role__role="admin"): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 909 | trp_ids.extend( |
| 910 | [trp.id for trp in cls.objects.filter(tenant_root=priv.tenant_root)]) |
| Scott Baker | 5116b30 | 2015-07-24 15:48:03 -0700 | [diff] [blame] | 911 | |
| 912 | return cls.objects.filter(id__in=trp_ids) |
| 913 | |
| Jeremy Mowery | 9563513 | 2016-04-15 17:39:49 -0700 | [diff] [blame] | 914 | |
| Jeremy Mowery | b31bd9e | 2016-03-14 23:59:11 -0700 | [diff] [blame] | 915 | class TenantRole(PlCoreBase): |
| Jeremy Mowery | 98e97d7 | 2016-04-15 00:05:27 -0700 | [diff] [blame] | 916 | """A TenantRole option.""" |
| Jeremy Mowery | 9563513 | 2016-04-15 17:39:49 -0700 | [diff] [blame] | 917 | ROLE_CHOICES = (('admin', 'Admin'), ('access', 'Access')) |
| Jeremy Mowery | b31bd9e | 2016-03-14 23:59:11 -0700 | [diff] [blame] | 918 | role = StrippedCharField(choices=ROLE_CHOICES, unique=True, max_length=30) |
| Jeremy Mowery | 9563513 | 2016-04-15 17:39:49 -0700 | [diff] [blame] | 919 | |
| 920 | def __unicode__(self): return u'%s' % (self.role) |
| 921 | |
| Jeremy Mowery | b31bd9e | 2016-03-14 23:59:11 -0700 | [diff] [blame] | 922 | |
| 923 | class TenantPrivilege(PlCoreBase): |
| Jeremy Mowery | 98e97d7 | 2016-04-15 00:05:27 -0700 | [diff] [blame] | 924 | """"A TenantPrivilege which defines how users can access a particular Tenant. |
| 925 | |
| 926 | Attributes: |
| 927 | id (models.AutoField): The ID of the privilege. |
| 928 | user (models.ForeignKey): A Foreign Key to the a User. |
| 929 | tenant (models.ForeignKey): A ForeignKey to the Tenant. |
| 930 | role (models.ForeignKey): A ForeignKey to the TenantRole. |
| 931 | """ |
| Jeremy Mowery | c86352a | 2016-04-10 23:00:54 -0700 | [diff] [blame] | 932 | id = models.AutoField(primary_key=True) |
| Jeremy Mowery | 5d06a23 | 2016-04-04 22:30:44 -0700 | [diff] [blame] | 933 | user = models.ForeignKey('User', related_name="tenantprivileges") |
| 934 | tenant = models.ForeignKey('Tenant', related_name="tenantprivileges") |
| 935 | role = models.ForeignKey('TenantRole', related_name="tenantprivileges") |
| Jeremy Mowery | b31bd9e | 2016-03-14 23:59:11 -0700 | [diff] [blame] | 936 | |
| Jeremy Mowery | 9563513 | 2016-04-15 17:39:49 -0700 | [diff] [blame] | 937 | def __unicode__(self): return u'%s %s %s' % ( |
| 938 | self.tenant, self.user, self.role) |
| Jeremy Mowery | b31bd9e | 2016-03-14 23:59:11 -0700 | [diff] [blame] | 939 | |
| 940 | def save(self, *args, **kwds): |
| 941 | if not self.user.is_active: |
| 942 | raise PermissionDenied, "Cannot modify role(s) of a disabled user" |
| 943 | super(TenantPrivilege, self).save(*args, **kwds) |
| 944 | |
| 945 | def can_update(self, user): |
| 946 | return user.can_update_tenant_privilege(self) |
| 947 | |
| 948 | @classmethod |
| 949 | def select_by_user(cls, user): |
| 950 | if user.is_admin: |
| 951 | return cls.objects.all() |
| 952 | else: |
| 953 | # User can see his own privilege |
| 954 | trp_ids = [trp.id for trp in cls.objects.filter(user=user)] |
| 955 | |
| 956 | # A tenant admin can see the TenantPrivileges for their Tenants |
| 957 | for priv in cls.objects.filter(user=user, role__role="admin"): |
| Jeremy Mowery | 9563513 | 2016-04-15 17:39:49 -0700 | [diff] [blame] | 958 | trp_ids.extend( |
| 959 | [trp.id for trp in cls.objects.filter(tenant=priv.tenant)]) |
| Jeremy Mowery | b31bd9e | 2016-03-14 23:59:11 -0700 | [diff] [blame] | 960 | |
| 961 | return cls.objects.filter(id__in=trp_ids) |