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
8 # If true, then IP addresses will be allocated by the model. If false, then
9 # we will assume the observer handles it.
12 class NetworkTemplate(PlCoreBase):
13 VISIBILITY_CHOICES = (('public', 'public'), ('private', 'private'))
15 name = models.CharField(max_length=32)
16 description = models.CharField(max_length=1024, blank=True, null=True)
17 guaranteedBandwidth = models.IntegerField(default=0)
18 visibility = models.CharField(max_length=30, choices=VISIBILITY_CHOICES, default="private")
20 def __unicode__(self): return u'%s' % (self.name)
22 class Network(PlCoreBase):
23 name = models.CharField(max_length=32)
24 template = models.ForeignKey(NetworkTemplate)
25 subnet = models.CharField(max_length=32, blank=True)
26 ports = models.CharField(max_length=1024, blank=True, null=True)
27 labels = models.CharField(max_length=1024, blank=True, null=True)
28 owner = models.ForeignKey(Slice, related_name="ownedNetworks")
30 guaranteedBandwidth = models.IntegerField(default=0)
31 permitAllSlices = models.BooleanField(default=False)
32 permittedSlices = models.ManyToManyField(Slice, blank=True, related_name="availableNetworks")
33 slices = models.ManyToManyField(Slice, blank=True, related_name="networks", through="NetworkSlice")
34 slivers = models.ManyToManyField(Sliver, blank=True, related_name="networks", through="NetworkSliver")
36 def __unicode__(self): return u'%s' % (self.name)
38 def save(self, *args, **kwds):
39 if (not self.subnet) and (NO_OBSERVER):
40 from util.network_subnet_allocator import find_unused_subnet
41 self.subnet = find_unused_subnet(existing_subnets=[x.subnet for x in Network.objects.all()])
42 super(Network, self).save(*args, **kwds)
44 class NetworkSlice(PlCoreBase):
45 # This object exists solely so we can implement the permission check when
46 # adding slices to networks. It adds no additional fields to the relation.
48 network = models.ForeignKey(Network)
49 slice = models.ForeignKey(Slice)
51 def save(self, *args, **kwds):
53 if (slice not in self.network.permittedSlices.all()) and (slice != self.network.owner) and (not self.network.permitAllSlices):
54 # to add a sliver to the network, then one of the following must be true:
55 # 1) sliver's slice is in network's permittedSlices list,
56 # 2) sliver's slice is network's owner, or
57 # 3) network's permitAllSlices is true
58 raise ValueError("Slice %s is not allowed to connect to network %s" % (str(slice), str(self.network)))
60 super(NetworkSlice, self).save(*args, **kwds)
62 def __unicode__(self): return u'%s-%s' % (self.network.name, self.slice.name)
64 class NetworkSliver(PlCoreBase):
65 network = models.ForeignKey(Network)
66 sliver = models.ForeignKey(Sliver)
67 ip = models.GenericIPAddressField(help_text="Sliver ip address", blank=True, null=True)
69 def save(self, *args, **kwds):
70 slice = self.sliver.slice
71 if (slice not in self.network.permittedSlices.all()) and (slice != self.network.owner) and (not self.network.permitAllSlices):
72 # to add a sliver to the network, then one of the following must be true:
73 # 1) sliver's slice is in network's permittedSlices list,
74 # 2) sliver's slice is network's owner, or
75 # 3) network's permitAllSlices is true
76 raise ValueError("Slice %s is not allowed to connect to network %s" % (str(slice), str(self.network)))
78 if (not self.ip) and (NO_OBSERVER):
79 from util.network_subnet_allocator import find_unused_address
80 self.ip = find_unused_address(self.network.subnet,
81 [x.ip for x in self.network.networksliver_set.all()])
82 super(NetworkSliver, self).save(*args, **kwds)
84 def __unicode__(self): return u'%s-%s' % (self.network.name, self.sliver.instance_name)
86 class Router(PlCoreBase):
87 name = models.CharField(max_length=32)
88 owner = models.ForeignKey(Slice, related_name="routers")
89 permittedNetworks = models.ManyToManyField(Network, blank=True, related_name="availableRouters")
90 networks = models.ManyToManyField(Network, blank=True, related_name="routers")
92 def __unicode__(self): return u'%s' % (self.name)
94 class NetworkParameterType(PlCoreBase):
95 name = models.SlugField(help_text="The name of this parameter", max_length=128)
96 description = models.CharField(max_length=1024)
98 def __unicode__(self): return u'%s' % (self.name)
100 class NetworkParameter(PlCoreBase):
101 parameter = models.ForeignKey(NetworkParameterType, related_name="parameters", help_text="The type of the parameter")
102 value = models.CharField(help_text="The value of this parameter", max_length=1024)
104 # The required fields to do a ObjectType lookup, and object_id assignment
105 content_type = models.ForeignKey(ContentType)
106 object_id = models.PositiveIntegerField()
107 content_object = generic.GenericForeignKey('content_type', 'object_id')
109 def __unicode__(self):
110 return self.parameter.name