Scott Baker | c7f363b | 2015-01-14 00:34:45 -0800 | [diff] [blame] | 1 | from core.models import Slice, SlicePrivilege, SliceRole, Sliver, Site, Node, User |
Scott Baker | e791dc6 | 2014-08-28 14:02:54 -0700 | [diff] [blame] | 2 | from plus import PlusObjectMixin |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 3 | from operator import itemgetter, attrgetter |
Scott Baker | 5f4770d | 2014-07-13 11:17:58 -0700 | [diff] [blame] | 4 | |
Scott Baker | e791dc6 | 2014-08-28 14:02:54 -0700 | [diff] [blame] | 5 | class SlicePlus(Slice, PlusObjectMixin): |
Scott Baker | 88e3437 | 2014-07-13 11:46:36 -0700 | [diff] [blame] | 6 | class Meta: |
| 7 | proxy = True |
| 8 | |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 9 | def __init__(self, *args, **kwargs): |
| 10 | super(SlicePlus, self).__init__(*args, **kwargs) |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 11 | self._update_users = None |
Scott Baker | a31bb24 | 2015-01-16 19:07:36 -0800 | [diff] [blame] | 12 | self._sliceInfo = None |
Scott Baker | cd369ea | 2015-01-18 16:07:58 -0800 | [diff] [blame] | 13 | self.getSliceInfo() |
| 14 | self._site_allocation = self._sliceInfo["sitesUsed"] |
| 15 | self._initial_site_allocation = self._site_allocation |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 16 | |
Scott Baker | 5f4770d | 2014-07-13 11:17:58 -0700 | [diff] [blame] | 17 | def getSliceInfo(self, user=None): |
Scott Baker | a31bb24 | 2015-01-16 19:07:36 -0800 | [diff] [blame] | 18 | if not self._sliceInfo: |
| 19 | used_sites = {} |
| 20 | used_deployments = {} |
| 21 | sliverCount = 0 |
| 22 | for sliver in self.slivers.all(): |
| 23 | site = sliver.node.site_deployment.site |
| 24 | deployment = sliver.node.site_deployment.deployment |
| 25 | used_sites[site.name] = used_sites.get(site.name, 0) + 1 |
| 26 | used_deployments[deployment.name] = used_deployments.get(deployment.name, 0) + 1 |
| 27 | sliverCount = sliverCount + 1 |
Scott Baker | 5f4770d | 2014-07-13 11:17:58 -0700 | [diff] [blame] | 28 | |
Scott Baker | a31bb24 | 2015-01-16 19:07:36 -0800 | [diff] [blame] | 29 | users = {} |
| 30 | for priv in SlicePrivilege.objects.filter(slice=self): |
| 31 | if not (priv.user.id in users.keys()): |
| 32 | users[priv.user.id] = {"name": priv.user.email, "id": priv.user.id, "roles": []} |
| 33 | users[priv.user.id]["roles"].append(priv.role.role) |
Scott Baker | 5f4770d | 2014-07-13 11:17:58 -0700 | [diff] [blame] | 34 | |
Scott Baker | a31bb24 | 2015-01-16 19:07:36 -0800 | [diff] [blame] | 35 | self._sliceInfo= {"sitesUsed": used_sites, |
| 36 | "deploymentsUsed": used_deployments, |
| 37 | "sliverCount": sliverCount, |
| 38 | "siteCount": len(used_sites.keys()), |
| 39 | "users": users, |
| 40 | "roles": []} |
| 41 | |
| 42 | if user: |
| 43 | auser = self._sliceInfo["users"].get(user.id, None) |
| 44 | if (auser): |
| 45 | self._sliceInfo["roles"] = auser["roles"] |
| 46 | |
| 47 | return self._sliceInfo |
Scott Baker | 88e3437 | 2014-07-13 11:46:36 -0700 | [diff] [blame] | 48 | |
Scott Baker | 9062b0c | 2015-01-08 22:34:51 -0800 | [diff] [blame] | 49 | @property |
Scott Baker | 03696ee | 2015-01-11 13:45:19 -0800 | [diff] [blame] | 50 | def site_allocation(self): |
Scott Baker | cd369ea | 2015-01-18 16:07:58 -0800 | [diff] [blame] | 51 | return self._site_allocation |
Scott Baker | 03696ee | 2015-01-11 13:45:19 -0800 | [diff] [blame] | 52 | |
| 53 | @site_allocation.setter |
| 54 | def site_allocation(self, value): |
Scott Baker | cd369ea | 2015-01-18 16:07:58 -0800 | [diff] [blame] | 55 | self._site_allocation = value |
Scott Baker | 03696ee | 2015-01-11 13:45:19 -0800 | [diff] [blame] | 56 | |
| 57 | @property |
Scott Baker | a31bb24 | 2015-01-16 19:07:36 -0800 | [diff] [blame] | 58 | def user_names(self): |
| 59 | return [user["name"] for user in self.getSliceInfo()["users"].values()] |
| 60 | |
Scott Baker | a410100 | 2015-01-16 19:26:54 -0800 | [diff] [blame] | 61 | @user_names.setter |
| 62 | def user_names(self, value): |
| 63 | pass # it's read-only |
| 64 | |
Scott Baker | a31bb24 | 2015-01-16 19:07:36 -0800 | [diff] [blame] | 65 | @property |
Scott Baker | fcf655e | 2015-01-12 19:45:40 -0800 | [diff] [blame] | 66 | def users(self): |
Scott Baker | a31bb24 | 2015-01-16 19:07:36 -0800 | [diff] [blame] | 67 | return [user["id"] for user in self.getSliceInfo()["users"].values()] |
Scott Baker | fcf655e | 2015-01-12 19:45:40 -0800 | [diff] [blame] | 68 | |
| 69 | @users.setter |
| 70 | def users(self, value): |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 71 | self._update_users = value |
| 72 | #print "XXX set users to", value |
Scott Baker | fcf655e | 2015-01-12 19:45:40 -0800 | [diff] [blame] | 73 | |
| 74 | @property |
Scott Baker | 697512d | 2015-01-08 22:37:34 -0800 | [diff] [blame] | 75 | def network_ports(self): |
Scott Baker | 9062b0c | 2015-01-08 22:34:51 -0800 | [diff] [blame] | 76 | # XXX this assumes there is only one network that can have ports bound |
| 77 | # to it for a given slice. This is intended for the tenant view, which |
| 78 | # will obey this field. |
| 79 | networkPorts = "" |
| 80 | for networkSlice in self.networkslices.all(): |
| 81 | network = networkSlice.network |
| 82 | if network.ports: |
| 83 | networkPorts = network.ports |
| 84 | |
| 85 | return networkPorts |
| 86 | |
Scott Baker | 697512d | 2015-01-08 22:37:34 -0800 | [diff] [blame] | 87 | @network_ports.setter |
| 88 | def network_ports(self, value): |
Scott Baker | 9062b0c | 2015-01-08 22:34:51 -0800 | [diff] [blame] | 89 | print "XXX set networkPorts to", value |
| 90 | |
Scott Baker | 88e3437 | 2014-07-13 11:46:36 -0700 | [diff] [blame] | 91 | @staticmethod |
| 92 | def select_by_user(user): |
Scott Baker | 88e3437 | 2014-07-13 11:46:36 -0700 | [diff] [blame] | 93 | if user.is_admin: |
| 94 | qs = SlicePlus.objects.all() |
| 95 | else: |
| 96 | slice_ids = [sp.slice.id for sp in SlicePrivilege.objects.filter(user=user)] |
| 97 | qs = SlicePlus.objects.filter(id__in=slice_ids) |
Scott Baker | 88e3437 | 2014-07-13 11:46:36 -0700 | [diff] [blame] | 98 | return qs |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 99 | |
Scott Baker | c7f363b | 2015-01-14 00:34:45 -0800 | [diff] [blame] | 100 | def get_node_allocation(self, siteList): |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 101 | siteIDList = [site.id for site in siteList] |
| 102 | nodeList = [] |
| 103 | for node in Node.objects.all(): |
| 104 | if (node.site_deployment.site.id in siteIDList): |
| 105 | node.sliverCount = 0 |
| 106 | for sliver in node.slivers.all(): |
| 107 | if sliver.slice.id == self.id: |
| 108 | node.sliverCount = node.sliverCount + 1 |
| 109 | nodeList.append(node) |
| 110 | return nodeList |
| 111 | |
| 112 | def save(self, *args, **kwargs): |
Scott Baker | cd369ea | 2015-01-18 16:07:58 -0800 | [diff] [blame] | 113 | updated_image = self.has_field_changed("default_image") |
| 114 | updated_flavor = self.has_field_changed("default_flavor") |
| 115 | |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 116 | super(SlicePlus, self).save(*args, **kwargs) |
| 117 | |
Scott Baker | cd369ea | 2015-01-18 16:07:58 -0800 | [diff] [blame] | 118 | updated_sites = (self._site_allocation != self._initial_site_allocation) or updated_image or updated_flavor |
| 119 | if updated_sites: |
| 120 | self.save_site_allocation(noAct=True, reset=(updated_image or updated_flavor)) |
Scott Baker | c7f363b | 2015-01-14 00:34:45 -0800 | [diff] [blame] | 121 | |
| 122 | if self._update_users: |
| 123 | self.save_users(noAct=True) |
| 124 | |
Scott Baker | cd369ea | 2015-01-18 16:07:58 -0800 | [diff] [blame] | 125 | if updated_sites: |
| 126 | self.save_site_allocation(reset=(updated_image or updated_flavor)) |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 127 | |
Scott Baker | c7f363b | 2015-01-14 00:34:45 -0800 | [diff] [blame] | 128 | if self._update_users: |
| 129 | self.save_users() |
| 130 | |
Scott Baker | cd369ea | 2015-01-18 16:07:58 -0800 | [diff] [blame] | 131 | def save_site_allocation(self, noAct = False, reset=False): |
| 132 | print "save_site_allocation, reset=",reset |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 133 | |
Scott Baker | c7f363b | 2015-01-14 00:34:45 -0800 | [diff] [blame] | 134 | all_slice_slivers = self.slivers.all() |
Scott Baker | cd369ea | 2015-01-18 16:07:58 -0800 | [diff] [blame] | 135 | for site_name in self._site_allocation.keys(): |
| 136 | desired_allocation = self._site_allocation[site_name] |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 137 | |
| 138 | # make a list of the slivers for this site |
| 139 | slivers = [] |
| 140 | for sliver in all_slice_slivers: |
| 141 | if sliver.node.site_deployment.site.name == site_name: |
| 142 | slivers.append(sliver) |
| 143 | |
| 144 | # delete extra slivers |
Scott Baker | cd369ea | 2015-01-18 16:07:58 -0800 | [diff] [blame] | 145 | while (reset and len(slivers)>0) or (len(slivers) > desired_allocation): |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 146 | sliver = slivers.pop() |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 147 | if (not noAct): |
Scott Baker | cd369ea | 2015-01-18 16:07:58 -0800 | [diff] [blame] | 148 | print "deleting sliver", sliver |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 149 | sliver.delete() |
Scott Baker | cd369ea | 2015-01-18 16:07:58 -0800 | [diff] [blame] | 150 | else: |
| 151 | print "would delete sliver", sliver |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 152 | |
| 153 | # add more slivers |
| 154 | if (len(slivers) < desired_allocation): |
| 155 | site = Site.objects.get(name = site_name) |
Scott Baker | c7f363b | 2015-01-14 00:34:45 -0800 | [diff] [blame] | 156 | nodes = self.get_node_allocation([site]) |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 157 | |
| 158 | if (not nodes): |
Scott Baker | 79be5c0 | 2015-01-13 16:28:05 -0800 | [diff] [blame] | 159 | raise ValueError("no nodes in site %s" % site_name) |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 160 | |
| 161 | while (len(slivers) < desired_allocation): |
| 162 | # pick the least allocated node |
| 163 | nodes = sorted(nodes, key=attrgetter("sliverCount")) |
| 164 | node = nodes[0] |
| 165 | |
| 166 | sliver = Sliver(name=node.name, |
| 167 | slice=self, |
| 168 | node=node, |
| 169 | image = self.default_image, |
| 170 | flavor = self.default_flavor, |
| 171 | creator = self.creator, |
| 172 | deployment = node.site_deployment.deployment) |
| 173 | slivers.append(sliver) |
| 174 | if (not noAct): |
Scott Baker | cd369ea | 2015-01-18 16:07:58 -0800 | [diff] [blame] | 175 | print "added sliver", sliver |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 176 | sliver.save() |
Scott Baker | cd369ea | 2015-01-18 16:07:58 -0800 | [diff] [blame] | 177 | else: |
| 178 | print "would add sliver", sliver |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 179 | |
| 180 | node.sliverCount = node.sliverCount + 1 |
| 181 | |
Scott Baker | c7f363b | 2015-01-14 00:34:45 -0800 | [diff] [blame] | 182 | def save_users(self, noAct = False): |
| 183 | new_users = self._update_users |
| 184 | |
| 185 | default_role = SliceRole.objects.get(role="default") |
| 186 | |
| 187 | slice_privs = self.sliceprivileges.all() |
| 188 | slice_user_ids = [priv.user.id for priv in slice_privs] |
| 189 | |
| 190 | for user_id in new_users: |
| 191 | if (user_id not in slice_user_ids): |
Scott Baker | c7f363b | 2015-01-14 00:34:45 -0800 | [diff] [blame] | 192 | priv = SlicePrivilege(slice=self, user=User.objects.get(id=user_id), role=default_role) |
| 193 | if (not noAct): |
| 194 | priv.save() |
| 195 | |
| 196 | print "added user id", user_id |
| 197 | |
| 198 | for priv in slice_privs: |
| 199 | if (priv.role.id != default_role.id): |
| 200 | # only mess with 'default' users; don't kill an admin |
| 201 | continue |
| 202 | |
| 203 | if (priv.user.id not in new_users): |
| 204 | if (not noAct): |
| 205 | priv.delete() |
| 206 | |
| 207 | print "deleted user id", user_id |
| 208 | |
| 209 | |
| 210 | |
| 211 | |
Scott Baker | 2fbf7dc | 2015-01-13 16:22:57 -0800 | [diff] [blame] | 212 | |