Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 1 | import os |
| 2 | from django.db import models |
Siobhan Tully | 30fd429 | 2013-05-10 08:59:56 -0400 | [diff] [blame] | 3 | from core.models import PlCoreBase |
| 4 | from core.models import Site |
Tony Mack | c195d72 | 2014-08-20 11:15:17 -0400 | [diff] [blame] | 5 | from core.models.site import SitePrivilege |
Siobhan Tully | 30fd429 | 2013-05-10 08:59:56 -0400 | [diff] [blame] | 6 | from core.models import User |
| 7 | from core.models import Role |
Tony Mack | f3bbe47 | 2014-11-30 15:33:35 -0500 | [diff] [blame] | 8 | from core.models import Controller,ControllerLinkManager,ControllerLinkDeletionManager |
Scott Baker | e8d596f | 2013-05-13 23:17:13 -0700 | [diff] [blame] | 9 | from core.models import ServiceClass |
Scott Baker | cfe0fd9 | 2014-09-12 12:24:24 -0700 | [diff] [blame] | 10 | from core.models.serviceclass import get_default_serviceclass |
Siobhan Tully | de5450d | 2013-06-21 11:35:33 -0400 | [diff] [blame] | 11 | from core.models import Tag |
| 12 | from django.contrib.contenttypes import generic |
Siobhan Tully | ce652d0 | 2013-10-08 21:52:35 -0400 | [diff] [blame] | 13 | from core.models import Service |
Tony Mack | f3bbe47 | 2014-11-30 15:33:35 -0500 | [diff] [blame] | 14 | from core.models import Controller |
Scott Baker | 200181e | 2015-01-12 13:13:05 -0800 | [diff] [blame] | 15 | from core.models import Flavor, Image |
Tony Mack | dac8576 | 2014-09-03 13:19:42 -0400 | [diff] [blame] | 16 | from django.core.exceptions import ValidationError |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 17 | |
| 18 | # Create your models here. |
| 19 | |
| 20 | class Slice(PlCoreBase): |
Tony Mack | 4895203 | 2013-04-12 11:49:34 -0400 | [diff] [blame] | 21 | name = models.CharField(unique=True, help_text="The Name of the Slice", max_length=80) |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 22 | enabled = models.BooleanField(default=True, help_text="Status for this Slice") |
Scott Baker | cfe0fd9 | 2014-09-12 12:24:24 -0700 | [diff] [blame] | 23 | omf_friendly = models.BooleanField(default=False) |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 24 | description=models.TextField(blank=True,help_text="High level description of the slice and expected activities", max_length=1024) |
| 25 | slice_url = models.URLField(blank=True, max_length=512) |
Tony Mack | 360afb8 | 2013-12-17 13:32:45 -0500 | [diff] [blame] | 26 | site = models.ForeignKey(Site, related_name='slices', help_text="The Site this Slice belongs to") |
Scott Baker | cfe0fd9 | 2014-09-12 12:24:24 -0700 | [diff] [blame] | 27 | max_slivers = models.IntegerField(default=10) |
Siobhan Tully | ce652d0 | 2013-10-08 21:52:35 -0400 | [diff] [blame] | 28 | service = models.ForeignKey(Service, related_name='service', null=True, blank=True) |
Scott Baker | 4b0cedd | 2014-04-22 21:01:21 -0700 | [diff] [blame] | 29 | network = models.CharField(default="Private Only",null=True, blank=True, max_length=256) |
Siobhan Tully | de5450d | 2013-06-21 11:35:33 -0400 | [diff] [blame] | 30 | tags = generic.GenericRelation(Tag) |
Scott Baker | cfe0fd9 | 2014-09-12 12:24:24 -0700 | [diff] [blame] | 31 | serviceClass = models.ForeignKey(ServiceClass, related_name = "slices", null=True, default=get_default_serviceclass) |
Tony Mack | 2bd5b41 | 2013-06-11 21:05:06 -0400 | [diff] [blame] | 32 | creator = models.ForeignKey(User, related_name='slices', blank=True, null=True) |
Siobhan Tully | 231f4c8 | 2013-05-02 05:47:24 -0400 | [diff] [blame] | 33 | |
Scott Baker | ee9736d | 2015-01-03 16:26:38 -0800 | [diff] [blame] | 34 | # for tenant view |
Scott Baker | 200181e | 2015-01-12 13:13:05 -0800 | [diff] [blame] | 35 | default_flavor = models.ForeignKey(Flavor, related_name = "slices", null=True, blank=True) |
| 36 | default_image = models.ForeignKey(Image, related_name = "slices", null=True, blank=True); |
Scott Baker | ee9736d | 2015-01-03 16:26:38 -0800 | [diff] [blame] | 37 | mount_data_sets = models.CharField(default="GenBank",null=True, blank=True, max_length=256) |
| 38 | |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 39 | def __unicode__(self): return u'%s' % (self.name) |
| 40 | |
Tony Mack | 2721d6f | 2014-08-11 11:14:58 -0400 | [diff] [blame] | 41 | @property |
| 42 | def slicename(self): |
| 43 | return "%s_%s" % (self.site.login_base, self.name) |
| 44 | |
Tony Mack | 62bc59a | 2013-04-14 23:27:12 -0400 | [diff] [blame] | 45 | def save(self, *args, **kwds): |
Tony Mack | dac8576 | 2014-09-03 13:19:42 -0400 | [diff] [blame] | 46 | site = Site.objects.get(id=self.site.id) |
Tony Mack | ab5d887 | 2014-09-24 12:44:22 -0400 | [diff] [blame] | 47 | # allow preexisting slices to keep their original name for now |
| 48 | if not self.id and not self.name.startswith(site.login_base): |
Tony Mack | dac8576 | 2014-09-03 13:19:42 -0400 | [diff] [blame] | 49 | raise ValidationError('slice name must begin with %s' % site.login_base) |
Scott Baker | 22732bb | 2015-01-29 17:16:10 -0800 | [diff] [blame] | 50 | |
| 51 | if self.name == site.login_base+"_": |
| 52 | raise ValidationError('slice name is too short') |
| 53 | |
| 54 | if " " in self.name: |
| 55 | raise ValidationError('slice name must not contain spaces') |
| 56 | |
Scott Baker | e8d596f | 2013-05-13 23:17:13 -0700 | [diff] [blame] | 57 | if self.serviceClass is None: |
| 58 | # We allowed None=True for serviceClass because Django evolution |
| 59 | # will fail unless it is allowed. But, we we really don't want it to |
| 60 | # ever save None, so fix it up here. |
| 61 | self.serviceClass = ServiceClass.get_default() |
Tony Mack | 2bd5b41 | 2013-06-11 21:05:06 -0400 | [diff] [blame] | 62 | if not self.creator and hasattr(self, 'caller'): |
| 63 | self.creator = self.caller |
Scott Baker | d168ae0 | 2015-01-29 17:53:26 -0800 | [diff] [blame] | 64 | if not self.creator: |
| 65 | raise ValidationError('slice has no creator') |
Tony Mack | 62bc59a | 2013-04-14 23:27:12 -0400 | [diff] [blame] | 66 | super(Slice, self).save(*args, **kwds) |
| 67 | |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 68 | def can_update(self, user): |
| 69 | if user.is_readonly: |
| 70 | return False |
| 71 | if user.is_admin: |
| 72 | return True |
Tony Mack | c195d72 | 2014-08-20 11:15:17 -0400 | [diff] [blame] | 73 | # slice admins can update |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 74 | slice_privs = SlicePrivilege.objects.filter(user=user, slice=self) |
| 75 | for slice_priv in slice_privs: |
Tony Mack | c195d72 | 2014-08-20 11:15:17 -0400 | [diff] [blame] | 76 | if slice_priv.role.role == 'admin': |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 77 | return True |
Tony Mack | c195d72 | 2014-08-20 11:15:17 -0400 | [diff] [blame] | 78 | # site pis can update |
| 79 | site_privs = SitePrivilege.objects.filter(user=user, site=self.site) |
| 80 | for site_priv in site_privs: |
| 81 | if site_priv.role.role == 'pi': |
| 82 | return True |
| 83 | |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 84 | return False |
| 85 | |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 86 | @staticmethod |
| 87 | def select_by_user(user): |
| 88 | if user.is_admin: |
| 89 | qs = Slice.objects.all() |
| 90 | else: |
Tony Mack | 33df82b | 2014-08-20 11:29:40 -0400 | [diff] [blame] | 91 | # users can see slices they belong to |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 92 | slice_ids = [sp.slice.id for sp in SlicePrivilege.objects.filter(user=user)] |
Tony Mack | 33df82b | 2014-08-20 11:29:40 -0400 | [diff] [blame] | 93 | # pis can see slices at their sites |
| 94 | sites = [sp.site for sp in SitePrivilege.objects.filter(user=user)\ |
| 95 | if sp.role.role == 'pi'] |
| 96 | slice_ids.extend([s.id for s in Slice.objects.filter(site__in=sites)]) |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 97 | qs = Slice.objects.filter(id__in=slice_ids) |
| 98 | return qs |
| 99 | |
Tony Mack | c767c98 | 2014-10-22 13:30:41 -0400 | [diff] [blame] | 100 | def delete(self, *args, **kwds): |
| 101 | # delete networks associated with this slice |
| 102 | from core.models.network import Network |
| 103 | nets = Network.objects.filter(slices=self) |
| 104 | nets.delete() |
Tony Mack | f3bbe47 | 2014-11-30 15:33:35 -0500 | [diff] [blame] | 105 | # delete slice controllers |
Tony Mack | a7dbd42 | 2015-01-05 22:48:11 -0500 | [diff] [blame] | 106 | slice_controllers = ControllerSlice.objects.filter(slice=self) |
Tony Mack | f3bbe47 | 2014-11-30 15:33:35 -0500 | [diff] [blame] | 107 | slice_controllers.delete() |
Tony Mack | 9b53a4c | 2014-11-12 10:01:26 -0500 | [diff] [blame] | 108 | # delete slice privilege |
| 109 | slice_privileges = SlicePrivilege.objects.filter(slice=self) |
| 110 | slice_privileges.delete() |
| 111 | # continue with normal delete |
Sapan | e0abef3 | 2014-11-24 16:44:48 -0500 | [diff] [blame] | 112 | super(Slice, self).delete(*args, **kwds) |
Tony Mack | 9b53a4c | 2014-11-12 10:01:26 -0500 | [diff] [blame] | 113 | |
Tony Mack | c767c98 | 2014-10-22 13:30:41 -0400 | [diff] [blame] | 114 | |
Siobhan Tully | bfd11dc | 2013-09-03 12:59:24 -0400 | [diff] [blame] | 115 | class SliceRole(PlCoreBase): |
Siobhan Tully | cf04fb6 | 2014-01-11 11:25:57 -0500 | [diff] [blame] | 116 | ROLE_CHOICES = (('admin','Admin'),('default','Default')) |
Siobhan Tully | bfd11dc | 2013-09-03 12:59:24 -0400 | [diff] [blame] | 117 | |
| 118 | role = models.CharField(choices=ROLE_CHOICES, unique=True, max_length=30) |
| 119 | |
| 120 | def __unicode__(self): return u'%s' % (self.role) |
| 121 | |
| 122 | class SlicePrivilege(PlCoreBase): |
Sapan Bhatia | 13d2db9 | 2014-11-11 21:47:45 -0500 | [diff] [blame] | 123 | user = models.ForeignKey('User', related_name='sliceprivileges') |
| 124 | slice = models.ForeignKey('Slice', related_name='sliceprivileges') |
| 125 | role = models.ForeignKey('SliceRole',related_name='sliceprivileges') |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 126 | |
| 127 | def __unicode__(self): return u'%s %s %s' % (self.slice, self.user, self.role) |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 128 | |
| 129 | def can_update(self, user): |
Tony Mack | b7b4f84 | 2014-02-04 19:50:31 -0500 | [diff] [blame] | 130 | return self.slice.can_update(user) |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 131 | |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 132 | @staticmethod |
| 133 | def select_by_user(user): |
| 134 | if user.is_admin: |
| 135 | qs = SlicePrivilege.objects.all() |
| 136 | else: |
| 137 | sp_ids = [sp.id for sp in SlicePrivilege.objects.filter(user=user)] |
| 138 | qs = SlicePrivilege.objects.filter(id__in=sp_ids) |
| 139 | return qs |
Tony Mack | e4be32f | 2014-03-11 20:45:25 -0400 | [diff] [blame] | 140 | |
Tony Mack | a7dbd42 | 2015-01-05 22:48:11 -0500 | [diff] [blame] | 141 | class ControllerSlice(PlCoreBase): |
Tony Mack | f3bbe47 | 2014-11-30 15:33:35 -0500 | [diff] [blame] | 142 | objects = ControllerLinkManager() |
| 143 | deleted_objects = ControllerLinkDeletionManager() |
Sapan Bhatia | a16be69 | 2014-09-22 14:55:08 -0400 | [diff] [blame] | 144 | |
Tony Mack | f3bbe47 | 2014-11-30 15:33:35 -0500 | [diff] [blame] | 145 | controller = models.ForeignKey(Controller, related_name='controllerslices') |
| 146 | slice = models.ForeignKey(Slice, related_name='controllerslices') |
Tony Mack | c218b59 | 2014-03-19 15:17:53 -0400 | [diff] [blame] | 147 | tenant_id = models.CharField(null=True, blank=True, max_length=200, help_text="Keystone tenant id") |
Tony Mack | e4be32f | 2014-03-11 20:45:25 -0400 | [diff] [blame] | 148 | |
Tony Mack | f3bbe47 | 2014-11-30 15:33:35 -0500 | [diff] [blame] | 149 | def __unicode__(self): return u'%s %s' % (self.slice, self.controller) |
Tony Mack | e4be32f | 2014-03-11 20:45:25 -0400 | [diff] [blame] | 150 | |
| 151 | @staticmethod |
| 152 | def select_by_user(user): |
| 153 | if user.is_admin: |
Tony Mack | a7dbd42 | 2015-01-05 22:48:11 -0500 | [diff] [blame] | 154 | qs = ControllerSlice.objects.all() |
Tony Mack | e4be32f | 2014-03-11 20:45:25 -0400 | [diff] [blame] | 155 | else: |
| 156 | slices = Slice.select_by_user(user) |
Tony Mack | a7dbd42 | 2015-01-05 22:48:11 -0500 | [diff] [blame] | 157 | qs = ControllerSlice.objects.filter(slice__in=slices) |
Tony Mack | e4be32f | 2014-03-11 20:45:25 -0400 | [diff] [blame] | 158 | return qs |
Sapan Bhatia | 4d7cdf2 | 2014-12-16 01:09:04 -0500 | [diff] [blame] | 159 | |
| 160 | def get_cpu_stats(self): |
| 161 | filter = 'project_id=%s'%self.tenant_id |
| 162 | return monitor.get_meter('cpu',filter,None) |
| 163 | |
| 164 | def get_bw_stats(self): |
| 165 | filter = 'project_id=%s'%self.tenant_id |
| 166 | return monitor.get_meter('network.outgoing.bytes',filter,None) |
| 167 | |
| 168 | def get_node_stats(self): |
| 169 | return len(self.slice.slivers) |