planetstack/core/models/network.py

   1 import os
   2 import socket
   3 from django.db import models
   4 from core.models import PlCoreBase, Site, Slice, Sliver, Deployment
   5 from django.contrib.contenttypes.models import ContentType
   6 from django.contrib.contenttypes import generic
   7
   8 # If true, then IP addresses will be allocated by the model. If false, then
   9 # we will assume the observer handles it.
  10 NO_OBSERVER=False
  11
  12 class NetworkTemplate(PlCoreBase):
  13     VISIBILITY_CHOICES = (('public', 'public'), ('private', 'private'))
  14     TRANSLATION_CHOICES = (('none', 'none'), ('NAT', 'NAT'))
  15
  16     name = models.CharField(max_length=32)
  17     description = models.CharField(max_length=1024, blank=True, null=True)
  18     guaranteedBandwidth = models.IntegerField(default=0)
  19     visibility = models.CharField(max_length=30, choices=VISIBILITY_CHOICES, default="private")
  20     translation = models.CharField(max_length=30, choices=TRANSLATION_CHOICES, default="none")
  21     sharedNetworkName = models.CharField(max_length=30, blank=True, null=True)
  22     sharedNetworkId = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum network")
  23
  24     def __unicode__(self):  return u'%s' % (self.name)
  25
  26 class Network(PlCoreBase):
  27     name = models.CharField(max_length=32)
  28     template = models.ForeignKey(NetworkTemplate)
  29     subnet = models.CharField(max_length=32, blank=True)
  30     ports = models.CharField(max_length=1024, blank=True, null=True)
  31     labels = models.CharField(max_length=1024, blank=True, null=True)
  32     owner = models.ForeignKey(Slice, related_name="ownedNetworks", help_text="Slice that owns control of this Network")
  33
  34     guaranteedBandwidth = models.IntegerField(default=0)
  35     permitAllSlices = models.BooleanField(default=False)
  36     permittedSlices = models.ManyToManyField(Slice, blank=True, related_name="availableNetworks")
  37     slices = models.ManyToManyField(Slice, blank=True, related_name="networks", through="NetworkSlice")
  38     slivers = models.ManyToManyField(Sliver, blank=True, related_name="networks", through="NetworkSliver")
  39
  40     # for observer/manager
  41     network_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum network")
  42     router_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum router id")
  43     subnet_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum subnet id")
  44
  45     def __unicode__(self):  return u'%s' % (self.name)
  46
  47     def save(self, *args, **kwds):
  48         if (not self.subnet) and (NO_OBSERVER):
  49             from util.network_subnet_allocator import find_unused_subnet
  50             self.subnet = find_unused_subnet(existing_subnets=[x.subnet for x in Network.objects.all()])
  51         super(Network, self).save(*args, **kwds)
  52
  53     def can_update(self, user):
  54         return self.owner.can_update(user)
  55
  56     @staticmethod
  57     def select_by_user(user):
  58         if user.is_admin:
  59             qs = Network.objects.all()
  60         else:
  61             slices = Slice.select_by_user(user)
  62             #slice_ids = [s.id for s in Slice.select_by_user(user)]
  63             qs = Network.objects.filter(owner__in=slices)
  64         return qs
  65
  66 class NetworkDeployments(PlCoreBase):
  67     # Stores the openstack ids at various deployments
  68     network = models.ForeignKey(Network)
  69     deployment = models.ForeignKey(Deployment)
  70     net_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum network")
  71     router_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum router id")
  72     subnet_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum subnet id")
  73     subnet = models.CharField(max_length=32, blank=True)
  74
  75     def can_update(self, user):
  76         return user.is_admin
  77
  78     @staticmethod
  79     def select_by_user(user):
  80         if user.is_admin:
  81             qs = NetworkDeployments.objects.all()
  82         else:
  83             slices = Slice.select_by_user(user)
  84             networks = Network.objects.filter(owner__in=slices)
  85             qs = NetworkDeployments.objects.filter(network__in=networks)
  86         return qs
  87
  88 class NetworkSlice(PlCoreBase):
  89     # This object exists solely so we can implement the permission check when
  90     # adding slices to networks. It adds no additional fields to the relation.
  91
  92     network = models.ForeignKey(Network)
  93     slice = models.ForeignKey(Slice)
  94
  95     def save(self, *args, **kwds):
  96         slice = self.slice
  97         if (slice not in self.network.permittedSlices.all()) and (slice != self.network.owner) and (not self.network.permitAllSlices):
  98             # to add a sliver to the network, then one of the following must be true:
  99             #   1) sliver's slice is in network's permittedSlices list,
 100             #   2) sliver's slice is network's owner, or
 101             #   3) network's permitAllSlices is true
 102             raise ValueError("Slice %s is not allowed to connect to network %s" % (str(slice), str(self.network)))
 103
 104         super(NetworkSlice, self).save(*args, **kwds)
 105
 106     def __unicode__(self):  return u'%s-%s' % (self.network.name, self.slice.name)
 107
 108     def can_update(self, user):
 109         return self.slice.can_update(user)
 110
 111     @staticmethod
 112     def select_by_user(user):
 113         if user.is_admin:
 114             qs = NetworkSlice.objects.all()
 115         else:
 116             slice_ids = [s.id for s in Slice.select_by_user(user)]
 117             qs = NetworkSlice.objects.filter(id__in=slice_ids)
 118         return qs
 119
 120 class NetworkSliver(PlCoreBase):
 121     network = models.ForeignKey(Network)
 122     sliver = models.ForeignKey(Sliver)
 123     ip = models.GenericIPAddressField(help_text="Sliver ip address", blank=True, null=True)
 124     port_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum port id")
 125
 126     def save(self, *args, **kwds):
 127         slice = self.sliver.slice
 128         if (slice not in self.network.permittedSlices.all()) and (slice != self.network.owner) and (not self.network.permitAllSlices):
 129             # to add a sliver to the network, then one of the following must be true:
 130             #   1) sliver's slice is in network's permittedSlices list,
 131             #   2) sliver's slice is network's owner, or
 132             #   3) network's permitAllSlices is true
 133             raise ValueError("Slice %s is not allowed to connect to network %s" % (str(slice), str(self.network)))
 134
 135         if (not self.ip) and (NO_OBSERVER):
 136             from util.network_subnet_allocator import find_unused_address
 137             self.ip = find_unused_address(self.network.subnet,
 138                                           [x.ip for x in self.network.networksliver_set.all()])
 139         super(NetworkSliver, self).save(*args, **kwds)
 140
 141     def __unicode__(self):  return u'%s-%s' % (self.network.name, self.sliver.instance_name)
 142
 143     def can_update(self, user):
 144         return self.sliver.can_update(user)
 145
 146     @staticmethod
 147     def select_by_user(user):
 148         if user.is_admin:
 149             qs = NetworkSliver.objects.all()
 150         else:
 151             sliver_ids = [s.id for s in NetworkSliver.select_by_user(user)]
 152             qs = NetworkSliver.objects.filter(id__in=sliver_ids)
 153         return qs
 154
 155 class Router(PlCoreBase):
 156     name = models.CharField(max_length=32)
 157     owner = models.ForeignKey(Slice, related_name="routers")
 158     permittedNetworks = models.ManyToManyField(Network, blank=True, related_name="availableRouters")
 159     networks = models.ManyToManyField(Network, blank=True, related_name="routers")
 160
 161     def __unicode__(self):  return u'%s' % (self.name)
 162
 163 class NetworkParameterType(PlCoreBase):
 164     name = models.SlugField(help_text="The name of this parameter", max_length=128)
 165     description = models.CharField(max_length=1024)
 166
 167     def __unicode__(self):  return u'%s' % (self.name)
 168
 169 class NetworkParameter(PlCoreBase):
 170     parameter = models.ForeignKey(NetworkParameterType, related_name="parameters", help_text="The type of the parameter")
 171     value = models.CharField(help_text="The value of this parameter", max_length=1024)
 172
 173     # The required fields to do a ObjectType lookup, and object_id assignment
 174     content_type = models.ForeignKey(ContentType)
 175     object_id = models.PositiveIntegerField()
 176     content_object = generic.GenericForeignKey('content_type', 'object_id')
 177
 178     def __unicode__(self):
 179         return self.parameter.name
 180
 181