| 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 * |
| Scott Baker | b0eb23e | 2016-06-01 16:08:04 -0700 | [diff] [blame^] | 8 | import urlparse |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 9 | |
| 10 | COARSE_KIND = "coarse" |
| 11 | |
| Scott Baker | 80d65b1 | 2016-05-26 20:20:47 -0700 | [diff] [blame] | 12 | def get_xos(): |
| 13 | xos = XOS.objects.all() |
| 14 | |
| 15 | if xos: |
| 16 | return xos[0] |
| 17 | else: |
| 18 | return None |
| 19 |
|
| Scott Baker | 2461bec | 2015-08-14 09:10:11 -0700 | [diff] [blame] | 20 | |
| Scott Baker | 82498c5 | 2015-07-13 13:07:27 -0700 | [diff] [blame] | 21 | class AttributeMixin(object): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 22 | # helper for extracting things from a json-encoded |
| 23 | # service_specific_attribute |
| 24 | |
| Scott Baker | 82498c5 | 2015-07-13 13:07:27 -0700 | [diff] [blame] | 25 | 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 37 | attributes[name] = value |
| Scott Baker | 82498c5 | 2015-07-13 13:07:27 -0700 | [diff] [blame] | 38 | 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 42 | attributes = json.loads( |
| 43 | self._initial["service_specific_attribute"]) |
| Scott Baker | 82498c5 | 2015-07-13 13:07:27 -0700 | [diff] [blame] | 44 | else: |
| 45 | attributes = {} |
| 46 | return attributes.get(name, default) |
| 47 | |
| Scott Baker | 74404fe | 2015-07-13 13:54:06 -0700 | [diff] [blame] | 48 | @classmethod |
| Scott Baker | 6dfde21 | 2016-03-01 20:10:24 -0800 | [diff] [blame] | 49 | def get_default_attribute(cls, name): |
| 50 | for (attrname, default) in cls.simple_attributes: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 51 | if attrname == name: |
| Scott Baker | 6dfde21 | 2016-03-01 20:10:24 -0800 | [diff] [blame] | 52 | return default |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 53 | if hasattr(cls, "default_attributes"): |
| Scott Baker | f180ff3 | 2016-03-31 11:46:02 -0700 | [diff] [blame] | 54 | if name in cls.default_attributes: |
| 55 | return cls.default_attributes[name] |
| 56 | |
| 57 | return None |
| Scott Baker | 6dfde21 | 2016-03-01 20:10:24 -0800 | [diff] [blame] | 58 | |
| 59 | @classmethod |
| Scott Baker | 74404fe | 2015-07-13 13:54:06 -0700 | [diff] [blame] | 60 | def setup_simple_attributes(cls): |
| 61 | for (attrname, default) in cls.simple_attributes: |
| Scott Baker | 096dce8 | 2015-07-13 14:27:51 -0700 | [diff] [blame] | 62 | 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] | 63 | lambda self, value, attrname=attrname: self.set_attribute( |
| 64 | attrname, value), |
| Scott Baker | 096dce8 | 2015-07-13 14:27:51 -0700 | [diff] [blame] | 65 | None, |
| 66 | attrname)) |
| Scott Baker | 74404fe | 2015-07-13 13:54:06 -0700 | [diff] [blame] | 67 | |
| Scott Baker | 80d65b1 | 2016-05-26 20:20:47 -0700 | [diff] [blame] | 68 | class ServiceController(PlCoreBase): |
| Scott Baker | b0eb23e | 2016-06-01 16:08:04 -0700 | [diff] [blame^] | 69 | xos = models.ForeignKey(XOS, related_name='service_controllers', help_text="Pointer to XOS", default=get_xos) |
| Scott Baker | 80d65b1 | 2016-05-26 20:20:47 -0700 | [diff] [blame] | 70 | name = StrippedCharField(max_length=30, help_text="Service Name") |
| Scott Baker | 1e55982 | 2016-05-26 20:57:10 -0700 | [diff] [blame] | 71 | 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] | 72 | |
| Scott Baker | 147cace | 2016-05-26 21:11:37 -0700 | [diff] [blame] | 73 | def __unicode__(self): return u'%s' % (self.name) |
| 74 | |
| Scott Baker | 80d65b1 | 2016-05-26 20:20:47 -0700 | [diff] [blame] | 75 | class ServiceControllerResource(PlCoreBase): |
| 76 | KIND_CHOICES = (('models', 'Models'), |
| 77 | ('admin', 'Admin'), |
| 78 | ('django_library', 'Django Library'), |
| Scott Baker | 1ea8cb5 | 2016-05-26 22:19:35 -0700 | [diff] [blame] | 79 | ('synchronizer', 'Synchronizer'), |
| Scott Baker | 80d65b1 | 2016-05-26 20:20:47 -0700 | [diff] [blame] | 80 | ('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 Baker | 411c4e9 | 2016-05-26 22:50:42 -0700 | [diff] [blame] | 86 | ('docker', 'Docker Container'), |
| 87 | ('yaml', 'YAML')) |
| Scott Baker | 80d65b1 | 2016-05-26 20:20:47 -0700 | [diff] [blame] | 88 | |
| Scott Baker | b0eb23e | 2016-06-01 16:08:04 -0700 | [diff] [blame^] | 89 | service_controller = models.ForeignKey(ServiceController, related_name='service_controller_resources', |
| Scott Baker | 80d65b1 | 2016-05-26 20:20:47 -0700 | [diff] [blame] | 90 | 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 Baker | 1e55982 | 2016-05-26 20:57:10 -0700 | [diff] [blame] | 95 | 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] | 96 | |
| Scott Baker | 147cace | 2016-05-26 21:11:37 -0700 | [diff] [blame] | 97 | def __unicode__(self): return u'%s' % (self.name) |
| 98 | |
| Scott Baker | b0eb23e | 2016-06-01 16:08:04 -0700 | [diff] [blame^] | 99 | @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 Baker | 82498c5 | 2015-07-13 13:07:27 -0700 | [diff] [blame] | 106 | class Service(PlCoreBase, AttributeMixin): |
| Scott Baker | 008a996 | 2015-04-15 20:58:20 -0700 | [diff] [blame] | 107 | # when subclassing a service, redefine KIND to describe the new service |
| 108 | KIND = "generic" |
| 109 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 110 | description = models.TextField( |
| 111 | max_length=254, null=True, blank=True, help_text="Description of Service") |
| Siobhan Tully | 00353f7 | 2013-10-08 21:53:27 -0400 | [diff] [blame] | 112 | enabled = models.BooleanField(default=True) |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 113 | kind = StrippedCharField( |
| 114 | max_length=30, help_text="Kind of service", default=KIND) |
| Tony Mack | d84b1ff | 2015-03-09 13:03:56 -0400 | [diff] [blame] | 115 | name = StrippedCharField(max_length=30, help_text="Service Name") |
| Scott Baker | d3e3029 | 2016-05-02 09:38:24 -0700 | [diff] [blame] | 116 | versionNumber = StrippedCharField(blank=True, null=True, |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 117 | max_length=30, help_text="Version of Service Definition") |
| Siobhan Tully | cf04fb6 | 2014-01-11 11:25:57 -0500 | [diff] [blame] | 118 | published = models.BooleanField(default=True) |
| Tony Mack | d84b1ff | 2015-03-09 13:03:56 -0400 | [diff] [blame] | 119 | view_url = StrippedCharField(blank=True, null=True, max_length=1024) |
| 120 | icon_url = StrippedCharField(blank=True, null=True, max_length=1024) |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 121 | public_key = models.TextField( |
| 122 | null=True, blank=True, max_length=1024, help_text="Public key string") |
| Scott Baker | f60c010 | 2015-11-12 16:22:52 -0800 | [diff] [blame] | 123 | private_key_fn = StrippedCharField(blank=True, null=True, max_length=1024) |
| Siobhan Tully | 00353f7 | 2013-10-08 21:53:27 -0400 | [diff] [blame] | 124 | |
| Scott Baker | 2f0828e | 2015-07-13 12:33:28 -0700 | [diff] [blame] | 125 | # Service_specific_attribute and service_specific_id are opaque to XOS |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 126 | service_specific_id = StrippedCharField( |
| 127 | max_length=30, blank=True, null=True) |
| Scott Baker | 2f0828e | 2015-07-13 12:33:28 -0700 | [diff] [blame] | 128 | service_specific_attribute = models.TextField(blank=True, null=True) |
| 129 | |
| Scott Baker | 80d65b1 | 2016-05-26 20:20:47 -0700 | [diff] [blame] | 130 | controller = models.ForeignKey(ServiceController, related_name='services', |
| 131 | help_text="The Service Controller this Service uses") |
| 132 | |
| Scott Baker | 008a996 | 2015-04-15 20:58:20 -0700 | [diff] [blame] | 133 | 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 140 | return cls.objects.filter(kind=cls.KIND) |
| Scott Baker | 008a996 | 2015-04-15 20:58:20 -0700 | [diff] [blame] | 141 | |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 142 | @classmethod |
| Scott Baker | 542cd6f | 2015-10-19 21:18:53 -0700 | [diff] [blame] | 143 | def get_deleted_service_objects(cls): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 144 | return cls.deleted_objects.filter(kind=cls.KIND) |
| Scott Baker | 542cd6f | 2015-10-19 21:18:53 -0700 | [diff] [blame] | 145 | |
| 146 | @classmethod |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 147 | def get_service_objects_by_user(cls, user): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 148 | return cls.select_by_user(user).filter(kind=cls.KIND) |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 149 | |
| 150 | @classmethod |
| 151 | def select_by_user(cls, user): |
| 152 | if user.is_admin: |
| 153 | return cls.objects.all() |
| 154 | else: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 155 | service_ids = [ |
| 156 | sp.slice.id for sp in ServicePrivilege.objects.filter(user=user)] |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 157 | return cls.objects.filter(id__in=service_ids) |
| 158 | |
| Scott Baker | cce158d | 2015-12-07 22:20:40 -0800 | [diff] [blame] | 159 | @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 Tully | 00353f7 | 2013-10-08 21:53:27 -0400 | [diff] [blame] | 166 | def __unicode__(self): return u'%s' % (self.name) |
| 167 | |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 168 | def can_update(self, user): |
| 169 | return user.can_update_service(self, allow=['admin']) |
| Scott Baker | 618a489 | 2015-07-06 14:27:31 -0700 | [diff] [blame] | 170 | |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 171 | 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 Mack | 3de59e3 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 176 | 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] | 177 | exclusive_slices - list of slices that must have no nodes in common with 'slice'. |
| 178 | """ |
| 179 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 180 | # late import to get around order-of-imports constraint in __init__.py |
| 181 | from core.models import Node, Instance |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 182 | |
| 183 | nodes = list(Node.objects.all()) |
| 184 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 185 | conflicting_instances = Instance.objects.filter( |
| 186 | slice__in=exclusive_slices) |
| 187 | conflicting_nodes = Node.objects.filter( |
| 188 | instances__in=conflicting_instances) |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 189 | |
| 190 | nodes = [x for x in nodes if x not in conflicting_nodes] |
| 191 | |
| Tony Mack | 3de59e3 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 192 | # 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] | 193 | # can have on a single node. |
| 194 | if max_per_node: |
| 195 | acceptable_nodes = [] |
| 196 | for node in nodes: |
| Tony Mack | 3de59e3 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 197 | existing_count = node.instances.filter(slice=slice).count() |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 198 | 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 Mack | 3de59e3 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 208 | nodes = sorted(nodes, key=lambda node: node.instances.all().count()) |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 209 | 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 214 | # late import to get around order-of-imports constraint in __init__.py |
| 215 | from core.models import Instance |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 216 | |
| 217 | slices = [x for x in self.slices.all() if slice_hint in x.name] |
| 218 | for slice in slices: |
| Tony Mack | 3de59e3 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 219 | while slice.instances.all().count() > scale: |
| 220 | s = slice.instances.all()[0] |
| 221 | # print "drop instance", s |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 222 | s.delete() |
| 223 | |
| Tony Mack | 3de59e3 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 224 | while slice.instances.all().count() < scale: |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 225 | 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 232 | raise XOSConfigurationError( |
| 233 | "No default_image for slice %s" % slice.name) |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 234 | |
| 235 | flavor = slice.default_flavor |
| 236 | if not flavor: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 237 | raise XOSConfigurationError( |
| 238 | "No default_flavor for slice %s" % slice.name) |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 239 | |
| Tony Mack | 3de59e3 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 240 | s = Instance(slice=slice, |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 241 | node=node, |
| 242 | creator=slice.creator, |
| 243 | image=image, |
| 244 | flavor=flavor, |
| 245 | deployment=node.site_deployment.deployment) |
| Scott Baker | 9843673 | 2015-05-11 16:36:41 -0700 | [diff] [blame] | 246 | s.save() |
| 247 | |
| Tony Mack | 3de59e3 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 248 | # print "add instance", s |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 249 | |
| Scott Baker | 8a67ae7 | 2015-12-09 22:54:52 -0800 | [diff] [blame] | 250 | def get_vtn_src_nets(self): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 251 | nets = [] |
| Scott Baker | 8a67ae7 | 2015-12-09 22:54:52 -0800 | [diff] [blame] | 252 | for slice in self.slices.all(): |
| 253 | for ns in slice.networkslices.all(): |
| 254 | if not ns.network: |
| 255 | continue |
| Scott Baker | f63cf1c | 2016-01-12 19:59:12 -0800 | [diff] [blame] | 256 | # if ns.network.template.access in ["direct", "indirect"]: |
| 257 | # # skip access networks; we want to use the private network |
| 258 | # continue |
| Scott Baker | e6e7855 | 2016-04-11 21:26:39 -0700 | [diff] [blame] | 259 | if "management" in ns.network.name: |
| 260 | # don't try to connect the management network to anything |
| 261 | continue |
| Scott Baker | 8a67ae7 | 2015-12-09 22:54:52 -0800 | [diff] [blame] | 262 | 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 Baker | cbd718e | 2015-12-08 21:31:18 -0800 | [diff] [blame] | 273 | def get_vtn_nets(self): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 274 | nets = [] |
| Scott Baker | cbd718e | 2015-12-08 21:31:18 -0800 | [diff] [blame] | 275 | for slice in self.slices.all(): |
| 276 | for ns in slice.networkslices.all(): |
| 277 | if not ns.network: |
| 278 | continue |
| Scott Baker | 8a67ae7 | 2015-12-09 22:54:52 -0800 | [diff] [blame] | 279 | if ns.network.template.access not in ["direct", "indirect"]: |
| 280 | # skip anything that's not an access network |
| 281 | continue |
| Scott Baker | cbd718e | 2015-12-08 21:31:18 -0800 | [diff] [blame] | 282 | 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 Baker | 5380a44 | 2015-12-08 19:27:50 -0800 | [diff] [blame] | 290 | for tenant in self.subscribed_tenants.all(): |
| 291 | if tenant.provider_service: |
| Scott Baker | cbd718e | 2015-12-08 21:31:18 -0800 | [diff] [blame] | 292 | 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 301 | 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] | 302 | |
| Scott Baker | 8a67ae7 | 2015-12-09 22:54:52 -0800 | [diff] [blame] | 303 | def get_vtn_src_ids(self): |
| 304 | return [x["net_id"] for x in self.get_vtn_src_nets()] |
| Scott Baker | cbd718e | 2015-12-08 21:31:18 -0800 | [diff] [blame] | 305 | |
| Scott Baker | 8a67ae7 | 2015-12-09 22:54:52 -0800 | [diff] [blame] | 306 | def get_vtn_src_names(self): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 307 | 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] | 308 | |
| 309 | |
| Siobhan Tully | 00353f7 | 2013-10-08 21:53:27 -0400 | [diff] [blame] | 310 | class ServiceAttribute(PlCoreBase): |
| Scott Baker | cce158d | 2015-12-07 22:20:40 -0800 | [diff] [blame] | 311 | name = models.CharField(help_text="Attribute Name", max_length=128) |
| Tony Mack | d84b1ff | 2015-03-09 13:03:56 -0400 | [diff] [blame] | 312 | value = StrippedCharField(help_text="Attribute Value", max_length=1024) |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 313 | service = models.ForeignKey(Service, related_name='serviceattributes', |
| 314 | help_text="The Service this attribute is associated with") |
| 315 | |
| Siobhan Tully | 00353f7 | 2013-10-08 21:53:27 -0400 | [diff] [blame] | 316 | |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 317 | class ServiceRole(PlCoreBase): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 318 | ROLE_CHOICES = (('admin', 'Admin'),) |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 319 | role = StrippedCharField(choices=ROLE_CHOICES, unique=True, max_length=30) |
| 320 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 321 | def __unicode__(self): return u'%s' % (self.role) |
| 322 | |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 323 | |
| 324 | class ServicePrivilege(PlCoreBase): |
| 325 | user = models.ForeignKey('User', related_name='serviceprivileges') |
| 326 | service = models.ForeignKey('Service', related_name='serviceprivileges') |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 327 | role = models.ForeignKey('ServiceRole', related_name='serviceprivileges') |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 328 | |
| 329 | class Meta: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 330 | unique_together = ('user', 'service', 'role') |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 331 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 332 | def __unicode__(self): return u'%s %s %s' % ( |
| 333 | self.service, self.user, self.role) |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 334 | |
| 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 Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 348 | super(ServicePrivilege, self).delete(*args, **kwds) |
| 349 | |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 350 | @classmethod |
| 351 | def select_by_user(cls, user): |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 352 | if user.is_admin: |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 353 | qs = cls.objects.all() |
| Tony Mack | 9d2ea09 | 2015-04-29 12:23:10 -0400 | [diff] [blame] | 354 | else: |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 355 | qs = cls.objects.filter(user=user) |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 356 | return qs |
| 357 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 358 | |
| Scott Baker | 82498c5 | 2015-07-13 13:07:27 -0700 | [diff] [blame] | 359 | class TenantRoot(PlCoreBase, AttributeMixin): |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 360 | """ 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 364 | KIND = "generic" |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 365 | kind = StrippedCharField(max_length=30, default=KIND) |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 366 | name = StrippedCharField( |
| 367 | max_length=255, help_text="name", blank=True, null=True) |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 368 | |
| Scott Baker | efcec63 | 2015-07-07 12:12:42 -0700 | [diff] [blame] | 369 | service_specific_attribute = models.TextField(blank=True, null=True) |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 370 | service_specific_id = StrippedCharField( |
| 371 | max_length=30, blank=True, null=True) |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 372 | |
| Scott Baker | db66fd3 | 2015-07-07 17:59:44 -0700 | [diff] [blame] | 373 | 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 Baker | 618a489 | 2015-07-06 14:27:31 -0700 | [diff] [blame] | 378 | 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 Baker | efcec63 | 2015-07-07 12:12:42 -0700 | [diff] [blame] | 387 | def get_subscribed_tenants(self, tenant_class): |
| 388 | ids = self.subscribed_tenants.filter(kind=tenant_class.KIND) |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 389 | return tenant_class.objects.filter(id__in=ids) |
| Scott Baker | efcec63 | 2015-07-07 12:12:42 -0700 | [diff] [blame] | 390 | |
| 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 399 | return cls.objects.filter(kind=cls.KIND) |
| Scott Baker | efcec63 | 2015-07-07 12:12:42 -0700 | [diff] [blame] | 400 | |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 401 | @classmethod |
| 402 | def get_tenant_objects_by_user(cls, user): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 403 | return cls.select_by_user(user).filter(kind=cls.KIND) |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 404 | |
| 405 | @classmethod |
| 406 | def select_by_user(cls, user): |
| 407 | if user.is_admin: |
| 408 | return cls.objects.all() |
| 409 | else: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 410 | tr_ids = [ |
| 411 | trp.tenant_root.id for trp in TenantRootPrivilege.objects.filter(user=user)] |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 412 | return cls.objects.filter(id__in=tr_ids) |
| 413 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 414 | # helper function to be used in subclasses that want to ensure |
| 415 | # service_specific_id is unique |
| Scott Baker | dc880f3 | 2016-03-31 14:45:31 -0700 | [diff] [blame] | 416 | def validate_unique_service_specific_id(self, none_okay=False): |
| 417 | if not none_okay and (self.service_specific_id is None): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 418 | raise XOSMissingField("subscriber_specific_id is None, and it's a required field", fields={ |
| 419 | "service_specific_id": "cannot be none"}) |
| Scott Baker | dc880f3 | 2016-03-31 14:45:31 -0700 | [diff] [blame] | 420 | |
| 421 | if self.service_specific_id: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 422 | conflicts = self.get_tenant_objects().filter( |
| 423 | service_specific_id=self.service_specific_id) |
| Scott Baker | dc880f3 | 2016-03-31 14:45:31 -0700 | [diff] [blame] | 424 | if self.pk: |
| Scott Baker | e4c9d87 | 2016-04-01 16:28:41 -0700 | [diff] [blame] | 425 | conflicts = conflicts.exclude(pk=self.pk) |
| Scott Baker | dc880f3 | 2016-03-31 14:45:31 -0700 | [diff] [blame] | 426 | if conflicts: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 427 | raise XOSDuplicateKey("service_specific_id %s already exists" % self.service_specific_id, fields={ |
| 428 | "service_specific_id": "duplicate key"}) |
| 429 | |
| Scott Baker | dc880f3 | 2016-03-31 14:45:31 -0700 | [diff] [blame] | 430 | |
| Scott Baker | 82498c5 | 2015-07-13 13:07:27 -0700 | [diff] [blame] | 431 | class Tenant(PlCoreBase, AttributeMixin): |
| Scott Baker | 8103d0f | 2015-04-10 16:42:26 -0700 | [diff] [blame] | 432 | """ A tenant is a relationship between two entities, a subscriber and a |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 433 | provider. This object represents an edge. |
| Scott Baker | 8103d0f | 2015-04-10 16:42:26 -0700 | [diff] [blame] | 434 | |
| 435 | The subscriber can be a User, a Service, or a Tenant. |
| 436 | |
| 437 | The provider is always a Service. |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 438 | |
| 439 | TODO: rename "Tenant" to "Tenancy" |
| Scott Baker | 8103d0f | 2015-04-10 16:42:26 -0700 | [diff] [blame] | 440 | """ |
| Scott Baker | 008a996 | 2015-04-15 20:58:20 -0700 | [diff] [blame] | 441 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 442 | CONNECTIVITY_CHOICES = (('public', 'Public'), |
| 443 | ('private', 'Private'), ('na', 'Not Applicable')) |
| Scott Baker | 925a8fa | 2015-04-26 20:30:40 -0700 | [diff] [blame] | 444 | |
| Scott Baker | 008a996 | 2015-04-15 20:58:20 -0700 | [diff] [blame] | 445 | # when subclassing a service, redefine KIND to describe the new service |
| 446 | KIND = "generic" |
| 447 | |
| 448 | kind = StrippedCharField(max_length=30, default=KIND) |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 449 | provider_service = models.ForeignKey( |
| 450 | Service, related_name='provided_tenants') |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 451 | |
| 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 Baker | 97ca910 | 2016-04-13 16:56:39 -0700 | [diff] [blame] | 454 | # XXX these should really be changed to GenericForeignKey |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 455 | 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 Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 465 | |
| 466 | # Service_specific_attribute and service_specific_id are opaque to XOS |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 467 | service_specific_id = StrippedCharField( |
| 468 | max_length=30, blank=True, null=True) |
| Scott Baker | 76934d8 | 2015-05-06 19:49:31 -0700 | [diff] [blame] | 469 | service_specific_attribute = models.TextField(blank=True, null=True) |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 470 | |
| 471 | # Connect_method is only used by Coarse tenants |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 472 | connect_method = models.CharField( |
| 473 | null=False, blank=False, max_length=30, choices=CONNECTIVITY_CHOICES, default="na") |
| Scott Baker | 8103d0f | 2015-04-10 16:42:26 -0700 | [diff] [blame] | 474 | |
| Scott Baker | 008a996 | 2015-04-15 20:58:20 -0700 | [diff] [blame] | 475 | 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 Baker | 8103d0f | 2015-04-10 16:42:26 -0700 | [diff] [blame] | 480 | def __unicode__(self): |
| Scott Baker | f996b76 | 2015-05-20 20:42:04 -0700 | [diff] [blame] | 481 | return u"%s-tenant-%s" % (str(self.kind), str(self.id)) |
| Scott Baker | 8103d0f | 2015-04-10 16:42:26 -0700 | [diff] [blame] | 482 | |
| Scott Baker | 008a996 | 2015-04-15 20:58:20 -0700 | [diff] [blame] | 483 | @classmethod |
| 484 | def get_tenant_objects(cls): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 485 | return cls.objects.filter(kind=cls.KIND) |
| Scott Baker | 008a996 | 2015-04-15 20:58:20 -0700 | [diff] [blame] | 486 | |
| Scott Baker | e7fc9f5 | 2015-05-05 17:52:03 -0700 | [diff] [blame] | 487 | @classmethod |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 488 | def get_tenant_objects_by_user(cls, user): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 489 | return cls.select_by_user(user).filter(kind=cls.KIND) |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 490 | |
| 491 | @classmethod |
| Scott Baker | e7fc9f5 | 2015-05-05 17:52:03 -0700 | [diff] [blame] | 492 | def get_deleted_tenant_objects(cls): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 493 | return cls.deleted_objects.filter(kind=cls.KIND) |
| Scott Baker | e7fc9f5 | 2015-05-05 17:52:03 -0700 | [diff] [blame] | 494 | |
| Scott Baker | cce158d | 2015-12-07 22:20:40 -0800 | [diff] [blame] | 495 | @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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 502 | # helper function to be used in subclasses that want to ensure |
| 503 | # service_specific_id is unique |
| Scott Baker | d921e1c | 2015-04-20 14:24:29 -0700 | [diff] [blame] | 504 | def validate_unique_service_specific_id(self): |
| 505 | if self.pk is None: |
| 506 | if self.service_specific_id is None: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 507 | raise XOSMissingField("subscriber_specific_id is None, and it's a required field", fields={ |
| 508 | "service_specific_id": "cannot be none"}) |
| Scott Baker | d921e1c | 2015-04-20 14:24:29 -0700 | [diff] [blame] | 509 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 510 | conflicts = self.get_tenant_objects().filter( |
| 511 | service_specific_id=self.service_specific_id) |
| Scott Baker | d921e1c | 2015-04-20 14:24:29 -0700 | [diff] [blame] | 512 | if conflicts: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 513 | raise XOSDuplicateKey("service_specific_id %s already exists" % self.service_specific_id, fields={ |
| 514 | "service_specific_id": "duplicate key"}) |
| Scott Baker | d921e1c | 2015-04-20 14:24:29 -0700 | [diff] [blame] | 515 | |
| Scott Baker | 618a489 | 2015-07-06 14:27:31 -0700 | [diff] [blame] | 516 | def save(self, *args, **kwargs): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 517 | 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 Baker | 618a489 | 2015-07-06 14:27:31 -0700 | [diff] [blame] | 519 | if (subCount > 1): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 520 | raise XOSConflictingField( |
| 521 | "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] | 522 | |
| 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 527 | return tenant_class.objects.filter(id__in=ids) |
| Scott Baker | 618a489 | 2015-07-06 14:27:31 -0700 | [diff] [blame] | 528 | |
| 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 535 | |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 536 | class 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 Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 551 | |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 552 | class LeastLoadedNodeScheduler(Scheduler): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 553 | # This scheduler always return the node with the fewest number of |
| 554 | # instances. |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 555 | |
| Scott Baker | 6526f55 | 2016-03-04 11:29:02 -0800 | [diff] [blame] | 556 | def __init__(self, slice, label=None): |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 557 | super(LeastLoadedNodeScheduler, self).__init__(slice) |
| Scott Baker | 6526f55 | 2016-03-04 11:29:02 -0800 | [diff] [blame] | 558 | self.label = label |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 559 | |
| 560 | def pick(self): |
| 561 | from core.models import Node |
| Pingping Lin | fa30bae | 2016-03-03 09:52:24 -0800 | [diff] [blame] | 562 | 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 Scandolo | 1d7ff81 | 2016-04-21 09:02:08 -0700 | [diff] [blame] | 567 | |
| Scott Baker | 6526f55 | 2016-03-04 11:29:02 -0800 | [diff] [blame] | 568 | if self.label: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 569 | nodes = nodes.filter(nodelabels__name=self.label) |
| Scott Baker | 6526f55 | 2016-03-04 11:29:02 -0800 | [diff] [blame] | 570 | |
| 571 | nodes = list(nodes) |
| 572 | |
| 573 | if not nodes: |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 574 | raise Exception( |
| 575 | "LeastLoadedNodeScheduler: No suitable nodes to pick from") |
| Scott Baker | cce158d | 2015-12-07 22:20:40 -0800 | [diff] [blame] | 576 | |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 577 | # TODO: logic to filter nodes by which nodes are up, and which |
| 578 | # nodes the slice can instantiate on. |
| Pingping Lin | fa30bae | 2016-03-03 09:52:24 -0800 | [diff] [blame] | 579 | # nodes = sorted(nodes, key=lambda node: node.instances.all().count()) |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 580 | return [nodes[0], None] |
| 581 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 582 | |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 583 | class 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 Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 587 | 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 Baker | a8188a7 | 2016-05-19 17:54:52 -0700 | [diff] [blame] | 596 | # If slice has default_image set then use it |
| 597 | if self.slice.default_image: |
| 598 | return self.slice.default_image |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 599 | |
| Scott Baker | a8188a7 | 2016-05-19 17:54:52 -0700 | [diff] [blame] | 600 | raise XOPSProgrammingError("Please set a default image for %s" % self.slice.name) |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 601 | |
| 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 609 | (node, parent) = LeastLoadedNodeScheduler(self.slice).pick() |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 610 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 611 | 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 Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 619 | 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 634 | avail_vms.append((vm, container_count)) |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 635 | # sort by least containers-per-vm |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 636 | avail_vms = sorted(avail_vms, key=lambda x: x[1]) |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 637 | 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 645 | |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 646 | class TenantWithContainer(Tenant): |
| 647 | """ A tenant that manages a container """ |
| 648 | |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 649 | class Meta: |
| 650 | proxy = True |
| 651 | |
| 652 | def __init__(self, *args, **kwargs): |
| 653 | super(TenantWithContainer, self).__init__(*args, **kwargs) |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 654 | self.cached_instance = None |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 655 | self.orig_instance_id = self.get_initial_attribute("instance_id") |
| Scott Baker | e274f6b | 2015-11-02 20:54:28 -0800 | [diff] [blame] | 656 | |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 657 | @property |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 658 | def instance(self): |
| 659 | from core.models import Instance |
| 660 | if getattr(self, "cached_instance", None): |
| 661 | return self.cached_instance |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 662 | instance_id = self.get_attribute("instance_id") |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 663 | if not instance_id: |
| 664 | return None |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 665 | instances = Instance.objects.filter(id=instance_id) |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 666 | if not instances: |
| 667 | return None |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 668 | instance = instances[0] |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 669 | instance.caller = self.creator |
| 670 | self.cached_instance = instance |
| 671 | return instance |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 672 | |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 673 | @instance.setter |
| 674 | def instance(self, value): |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 675 | if value: |
| 676 | value = value.id |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 677 | if (value != self.get_attribute("instance_id", None)): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 678 | self.cached_instance = None |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 679 | self.set_attribute("instance_id", value) |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 680 | |
| Scott Baker | e274f6b | 2015-11-02 20:54:28 -0800 | [diff] [blame] | 681 | @property |
| Scott Baker | bdda0be | 2016-02-10 12:23:53 -0800 | [diff] [blame] | 682 | 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 Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 698 | def creator(self): |
| 699 | from core.models import User |
| 700 | if getattr(self, "cached_creator", None): |
| 701 | return self.cached_creator |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 702 | creator_id = self.get_attribute("creator_id") |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 703 | if not creator_id: |
| 704 | return None |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 705 | users = User.objects.filter(id=creator_id) |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 706 | if not users: |
| 707 | return None |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 708 | user = users[0] |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 709 | 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 717 | self.cached_creator = None |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 718 | 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 Baker | abad4c9 | 2016-05-25 10:02:29 -0700 | [diff] [blame] | 725 | |
| Scott Baker | cfaf0d3 | 2015-11-16 22:51:02 -0800 | [diff] [blame] | 726 | slice = self.provider_service.slices.all() |
| 727 | if not slice: |
| 728 | raise XOSProgrammingError("provider service has no slice") |
| 729 | slice = slice[0] |
| 730 | |
| Scott Baker | abad4c9 | 2016-05-25 10:02:29 -0700 | [diff] [blame] | 731 | # If slice has default_image set then use it |
| 732 | if slice.default_image: |
| 733 | return slice.default_image |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 734 | |
| Scott Baker | a8188a7 | 2016-05-19 17:54:52 -0700 | [diff] [blame] | 735 | raise XOPSProgrammingError("Please set a default image for %s" % self.slice.name) |
| Scott Baker | e274f6b | 2015-11-02 20:54:28 -0800 | [diff] [blame] | 736 | |
| Scott Baker | cfaf0d3 | 2015-11-16 22:51:02 -0800 | [diff] [blame] | 737 | 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 Baker | e274f6b | 2015-11-02 20:54:28 -0800 | [diff] [blame] | 741 | |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 742 | 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 746 | # 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 Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 749 | return None |
| 750 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 751 | # 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 Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 754 | 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 Baker | cfaf0d3 | 2015-11-16 22:51:02 -0800 | [diff] [blame] | 761 | def manage_container(self): |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 762 | from core.models import Instance, Flavor |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 763 | |
| 764 | if self.deleted: |
| 765 | return |
| 766 | |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 767 | if (self.instance is not None) and (self.instance.image != self.image): |
| 768 | self.instance.delete() |
| 769 | self.instance = None |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 770 | |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 771 | if self.instance is None: |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 772 | if not self.provider_service.slices.count(): |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 773 | raise XOSConfigurationError("The service has no slices") |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 774 | |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 775 | new_instance_created = False |
| 776 | instance = None |
| 777 | if self.get_attribute("use_same_instance_for_multiple_tenants", default=False): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 778 | # Find if any existing instances can be used for this tenant |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 779 | slices = self.provider_service.slices.all() |
| 780 | instance = self.pick_least_loaded_instance_in_slice(slices) |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 781 | |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 782 | if not instance: |
| Srikanth Vavilapalli | 71120a9 | 2015-11-17 13:41:38 -0600 | [diff] [blame] | 783 | slice = self.provider_service.slices.all()[0] |
| Scott Baker | cfaf0d3 | 2015-11-16 22:51:02 -0800 | [diff] [blame] | 784 | |
| Srikanth Vavilapalli | 2ab2d07 | 2016-02-15 01:18:09 -0500 | [diff] [blame] | 785 | 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 Vavilapalli | 71120a9 | 2015-11-17 13:41:38 -0600 | [diff] [blame] | 792 | if slice.default_isolation == "container_vm": |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 793 | (node, parent) = ContainerVmScheduler(slice).pick() |
| Srikanth Vavilapalli | 71120a9 | 2015-11-17 13:41:38 -0600 | [diff] [blame] | 794 | else: |
| Scott Baker | 534dd2a | 2015-11-18 20:58:08 -0800 | [diff] [blame] | 795 | (node, parent) = LeastLoadedNodeScheduler(slice).pick() |
| Scott Baker | cfaf0d3 | 2015-11-16 22:51:02 -0800 | [diff] [blame] | 796 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 797 | 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 Vavilapalli | 71120a9 | 2015-11-17 13:41:38 -0600 | [diff] [blame] | 805 | self.save_instance(instance) |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 806 | new_instance_created = True |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 807 | |
| 808 | try: |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 809 | self.instance = instance |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 810 | super(TenantWithContainer, self).save() |
| 811 | except: |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 812 | if new_instance_created: |
| 813 | instance.delete() |
| Scott Baker | e458afd | 2015-09-09 16:36:06 -0700 | [diff] [blame] | 814 | raise |
| 815 | |
| 816 | def cleanup_container(self): |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 817 | if self.instance: |
| Srikanth Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 818 | if self.get_attribute("use_same_instance_for_multiple_tenants", default=False): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 819 | # 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 Vavilapalli | 1d14ff0 | 2015-11-17 12:21:02 -0600 | [diff] [blame] | 823 | if tenant_count == 0: |
| 824 | self.instance.delete() |
| 825 | else: |
| 826 | self.instance.delete() |
| Tony Mack | 6a782f9 | 2015-09-13 22:50:39 +0000 | [diff] [blame] | 827 | self.instance = None |
| Scott Baker | 618a489 | 2015-07-06 14:27:31 -0700 | [diff] [blame] | 828 | |
| Scott Baker | ee83f2a | 2015-12-10 23:23:07 -0800 | [diff] [blame] | 829 | 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 834 | |
| Scott Baker | 925a8fa | 2015-04-26 20:30:40 -0700 | [diff] [blame] | 835 | class CoarseTenant(Tenant): |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 836 | """ TODO: rename "CoarseTenant" --> "StaticTenant" """ |
| Scott Baker | 925a8fa | 2015-04-26 20:30:40 -0700 | [diff] [blame] | 837 | class Meta: |
| 838 | proxy = True |
| Siobhan Tully | 00353f7 | 2013-10-08 21:53:27 -0400 | [diff] [blame] | 839 | |
| Scott Baker | 2461bec | 2015-08-14 09:10:11 -0700 | [diff] [blame] | 840 | KIND = COARSE_KIND |
| Scott Baker | 925a8fa | 2015-04-26 20:30:40 -0700 | [diff] [blame] | 841 | |
| 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 846 | raise XOSValidationError( |
| 847 | "subscriber_tenant and subscriber_user must be null") |
| Scott Baker | 925a8fa | 2015-04-26 20:30:40 -0700 | [diff] [blame] | 848 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 849 | super(CoarseTenant, self).save() |
| 850 | |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 851 | |
| 852 | class 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 860 | |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 861 | class 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 869 | |
| Scott Baker | 462a1d9 | 2015-10-15 15:59:19 -0700 | [diff] [blame] | 870 | class TenantAttribute(PlCoreBase): |
| Scott Baker | 907f991 | 2015-10-20 17:12:36 -0700 | [diff] [blame] | 871 | name = models.CharField(help_text="Attribute Name", max_length=128) |
| Scott Baker | 462a1d9 | 2015-10-15 15:59:19 -0700 | [diff] [blame] | 872 | value = models.TextField(help_text="Attribute Value") |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 873 | tenant = models.ForeignKey(Tenant, related_name='tenantattributes', |
| 874 | help_text="The Tenant this attribute is associated with") |
| Scott Baker | 462a1d9 | 2015-10-15 15:59:19 -0700 | [diff] [blame] | 875 | |
| Scott Baker | 1271a4d | 2016-04-06 14:34:49 -0700 | [diff] [blame] | 876 | def __unicode__(self): return u'%s-%s' % (self.name, self.id) |
| 877 | |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 878 | |
| 879 | class TenantRootRole(PlCoreBase): |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 880 | ROLE_CHOICES = (('admin', 'Admin'), ('access', 'Access')) |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 881 | |
| 882 | role = StrippedCharField(choices=ROLE_CHOICES, unique=True, max_length=30) |
| 883 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 884 | def __unicode__(self): return u'%s' % (self.role) |
| 885 | |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 886 | |
| 887 | class TenantRootPrivilege(PlCoreBase): |
| 888 | user = models.ForeignKey('User', related_name="tenant_root_privileges") |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 889 | tenant_root = models.ForeignKey( |
| 890 | 'TenantRoot', related_name="tenant_root_privileges") |
| 891 | role = models.ForeignKey( |
| 892 | 'TenantRootRole', related_name="tenant_root_privileges") |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 893 | |
| 894 | class Meta: |
| 895 | unique_together = ('user', 'tenant_root', 'role') |
| 896 | |
| Jeremy Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 897 | def __unicode__(self): return u'%s %s %s' % ( |
| 898 | self.tenant_root, self.user, self.role) |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 899 | |
| 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 Baker | 335882a | 2015-07-24 10:15:31 -0700 | [diff] [blame] | 903 | super(TenantRootPrivilege, self).save(*args, **kwds) |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 904 | |
| 905 | def can_update(self, user): |
| Scott Baker | 335882a | 2015-07-24 10:15:31 -0700 | [diff] [blame] | 906 | return user.can_update_tenant_root_privilege(self) |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 907 | |
| Scott Baker | 16573d3 | 2015-07-24 15:36:02 -0700 | [diff] [blame] | 908 | @classmethod |
| 909 | def select_by_user(cls, user): |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 910 | if user.is_admin: |
| Scott Baker | 5116b30 | 2015-07-24 15:48:03 -0700 | [diff] [blame] | 911 | return cls.objects.all() |
| Scott Baker | 4587b82 | 2015-07-01 18:29:08 -0700 | [diff] [blame] | 912 | else: |
| Scott Baker | 5116b30 | 2015-07-24 15:48:03 -0700 | [diff] [blame] | 913 | # 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 Mowery | 11ed899 | 2016-04-17 20:56:54 -0700 | [diff] [blame] | 918 | trp_ids.extend( |
| 919 | [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] | 920 | |
| 921 | return cls.objects.filter(id__in=trp_ids) |
| 922 | |
| Jeremy Mowery | 9563513 | 2016-04-15 17:39:49 -0700 | [diff] [blame] | 923 | |
| Jeremy Mowery | b31bd9e | 2016-03-14 23:59:11 -0700 | [diff] [blame] | 924 | class TenantRole(PlCoreBase): |
| Jeremy Mowery | 98e97d7 | 2016-04-15 00:05:27 -0700 | [diff] [blame] | 925 | """A TenantRole option.""" |
| Jeremy Mowery | 9563513 | 2016-04-15 17:39:49 -0700 | [diff] [blame] | 926 | ROLE_CHOICES = (('admin', 'Admin'), ('access', 'Access')) |
| Jeremy Mowery | b31bd9e | 2016-03-14 23:59:11 -0700 | [diff] [blame] | 927 | role = StrippedCharField(choices=ROLE_CHOICES, unique=True, max_length=30) |
| Jeremy Mowery | 9563513 | 2016-04-15 17:39:49 -0700 | [diff] [blame] | 928 | |
| 929 | def __unicode__(self): return u'%s' % (self.role) |
| 930 | |
| Jeremy Mowery | b31bd9e | 2016-03-14 23:59:11 -0700 | [diff] [blame] | 931 | |
| 932 | class TenantPrivilege(PlCoreBase): |
| Jeremy Mowery | 98e97d7 | 2016-04-15 00:05:27 -0700 | [diff] [blame] | 933 | """"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 Mowery | c86352a | 2016-04-10 23:00:54 -0700 | [diff] [blame] | 941 | id = models.AutoField(primary_key=True) |
| Jeremy Mowery | 5d06a23 | 2016-04-04 22:30:44 -0700 | [diff] [blame] | 942 | user = models.ForeignKey('User', related_name="tenantprivileges") |
| 943 | tenant = models.ForeignKey('Tenant', related_name="tenantprivileges") |
| 944 | role = models.ForeignKey('TenantRole', related_name="tenantprivileges") |
| Jeremy Mowery | b31bd9e | 2016-03-14 23:59:11 -0700 | [diff] [blame] | 945 | |
| Jeremy Mowery | 9563513 | 2016-04-15 17:39:49 -0700 | [diff] [blame] | 946 | def __unicode__(self): return u'%s %s %s' % ( |
| 947 | self.tenant, self.user, self.role) |
| Jeremy Mowery | b31bd9e | 2016-03-14 23:59:11 -0700 | [diff] [blame] | 948 | |
| 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 Mowery | 9563513 | 2016-04-15 17:39:49 -0700 | [diff] [blame] | 967 | trp_ids.extend( |
| 968 | [trp.id for trp in cls.objects.filter(tenant=priv.tenant)]) |
| Jeremy Mowery | b31bd9e | 2016-03-14 23:59:11 -0700 | [diff] [blame] | 969 | |
| 970 | return cls.objects.filter(id__in=trp_ids) |