Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 1 | import os |
| 2 | import socket |
| 3 | from django.db import models |
| 4 | from core.models import PlCoreBase, Site, Slice, Sliver |
| 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. |
Scott Baker | 026bfe7 | 2013-07-29 16:03:50 -0700 | [diff] [blame] | 10 | NO_OBSERVER=False |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 11 | |
| 12 | class NetworkTemplate(PlCoreBase): |
| 13 | VISIBILITY_CHOICES = (('public', 'public'), ('private', 'private')) |
Scott Baker | 87e5e09 | 2013-08-07 18:58:10 -0700 | [diff] [blame] | 14 | TRANSLATION_CHOICES = (('none', 'none'), ('NAT', 'NAT')) |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 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") |
Scott Baker | 87e5e09 | 2013-08-07 18:58:10 -0700 | [diff] [blame] | 20 | translation = models.CharField(max_length=30, choices=TRANSLATION_CHOICES, default="none") |
Scott Baker | 5f814b5 | 2013-08-09 14:51:21 -0700 | [diff] [blame] | 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") |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 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") |
| 33 | |
| 34 | guaranteedBandwidth = models.IntegerField(default=0) |
| 35 | permitAllSlices = models.BooleanField(default=False) |
| 36 | permittedSlices = models.ManyToManyField(Slice, blank=True, related_name="availableNetworks") |
Scott Baker | 87191e7 | 2013-08-06 08:55:07 -0700 | [diff] [blame] | 37 | slices = models.ManyToManyField(Slice, blank=True, related_name="networks", through="NetworkSlice") |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 38 | slivers = models.ManyToManyField(Sliver, blank=True, related_name="networks", through="NetworkSliver") |
| 39 | |
Scott Baker | 87191e7 | 2013-08-06 08:55:07 -0700 | [diff] [blame] | 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 | |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 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 | |
Scott Baker | 87191e7 | 2013-08-06 08:55:07 -0700 | [diff] [blame] | 53 | class NetworkSlice(PlCoreBase): |
| 54 | # This object exists solely so we can implement the permission check when |
| 55 | # adding slices to networks. It adds no additional fields to the relation. |
| 56 | |
| 57 | network = models.ForeignKey(Network) |
| 58 | slice = models.ForeignKey(Slice) |
| 59 | |
| 60 | def save(self, *args, **kwds): |
| 61 | slice = self.slice |
| 62 | if (slice not in self.network.permittedSlices.all()) and (slice != self.network.owner) and (not self.network.permitAllSlices): |
| 63 | # to add a sliver to the network, then one of the following must be true: |
| 64 | # 1) sliver's slice is in network's permittedSlices list, |
| 65 | # 2) sliver's slice is network's owner, or |
| 66 | # 3) network's permitAllSlices is true |
| 67 | raise ValueError("Slice %s is not allowed to connect to network %s" % (str(slice), str(self.network))) |
| 68 | |
| 69 | super(NetworkSlice, self).save(*args, **kwds) |
| 70 | |
| 71 | def __unicode__(self): return u'%s-%s' % (self.network.name, self.slice.name) |
| 72 | |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 73 | class NetworkSliver(PlCoreBase): |
| 74 | network = models.ForeignKey(Network) |
| 75 | sliver = models.ForeignKey(Sliver) |
Scott Baker | 026bfe7 | 2013-07-29 16:03:50 -0700 | [diff] [blame] | 76 | ip = models.GenericIPAddressField(help_text="Sliver ip address", blank=True, null=True) |
Scott Baker | f4df952 | 2013-08-19 17:56:45 -0700 | [diff] [blame] | 77 | port_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum port id") |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 78 | |
| 79 | def save(self, *args, **kwds): |
Scott Baker | 87191e7 | 2013-08-06 08:55:07 -0700 | [diff] [blame] | 80 | slice = self.sliver.slice |
| 81 | if (slice not in self.network.permittedSlices.all()) and (slice != self.network.owner) and (not self.network.permitAllSlices): |
| 82 | # to add a sliver to the network, then one of the following must be true: |
| 83 | # 1) sliver's slice is in network's permittedSlices list, |
| 84 | # 2) sliver's slice is network's owner, or |
| 85 | # 3) network's permitAllSlices is true |
| 86 | raise ValueError("Slice %s is not allowed to connect to network %s" % (str(slice), str(self.network))) |
| 87 | |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 88 | if (not self.ip) and (NO_OBSERVER): |
| 89 | from util.network_subnet_allocator import find_unused_address |
| 90 | self.ip = find_unused_address(self.network.subnet, |
| 91 | [x.ip for x in self.network.networksliver_set.all()]) |
| 92 | super(NetworkSliver, self).save(*args, **kwds) |
| 93 | |
| 94 | def __unicode__(self): return u'%s-%s' % (self.network.name, self.sliver.instance_name) |
| 95 | |
| 96 | class Router(PlCoreBase): |
| 97 | name = models.CharField(max_length=32) |
| 98 | owner = models.ForeignKey(Slice, related_name="routers") |
| 99 | permittedNetworks = models.ManyToManyField(Network, blank=True, related_name="availableRouters") |
| 100 | networks = models.ManyToManyField(Network, blank=True, related_name="routers") |
| 101 | |
| 102 | def __unicode__(self): return u'%s' % (self.name) |
| 103 | |
| 104 | class NetworkParameterType(PlCoreBase): |
| 105 | name = models.SlugField(help_text="The name of this parameter", max_length=128) |
| 106 | description = models.CharField(max_length=1024) |
| 107 | |
| 108 | def __unicode__(self): return u'%s' % (self.name) |
| 109 | |
| 110 | class NetworkParameter(PlCoreBase): |
| 111 | parameter = models.ForeignKey(NetworkParameterType, related_name="parameters", help_text="The type of the parameter") |
| 112 | value = models.CharField(help_text="The value of this parameter", max_length=1024) |
| 113 | |
| 114 | # The required fields to do a ObjectType lookup, and object_id assignment |
| 115 | content_type = models.ForeignKey(ContentType) |
| 116 | object_id = models.PositiveIntegerField() |
| 117 | content_object = generic.GenericForeignKey('content_type', 'object_id') |
| 118 | |
| 119 | def __unicode__(self): |
| 120 | return self.parameter.name |
| 121 | |
| 122 | |