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