blob: 23ef6eaa6b79531de00b50cb18562422b3cae42c [file] [log] [blame]
Scott Baker58a9c7a2013-07-29 15:43:07 -07001import os
2import socket
3from django.db import models
Siobhan Tullycf04fb62014-01-11 11:25:57 -05004from core.models import PlCoreBase, Site, Slice, Sliver
Scott Baker58a9c7a2013-07-29 15:43:07 -07005from django.contrib.contenttypes.models import ContentType
6from 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 Baker026bfe72013-07-29 16:03:50 -070010NO_OBSERVER=False
Scott Baker58a9c7a2013-07-29 15:43:07 -070011
12class NetworkTemplate(PlCoreBase):
13 VISIBILITY_CHOICES = (('public', 'public'), ('private', 'private'))
Scott Baker87e5e092013-08-07 18:58:10 -070014 TRANSLATION_CHOICES = (('none', 'none'), ('NAT', 'NAT'))
Scott Baker58a9c7a2013-07-29 15:43:07 -070015
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 Baker87e5e092013-08-07 18:58:10 -070020 translation = models.CharField(max_length=30, choices=TRANSLATION_CHOICES, default="none")
Scott Baker5f814b52013-08-09 14:51:21 -070021 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 Baker58a9c7a2013-07-29 15:43:07 -070023
24 def __unicode__(self): return u'%s' % (self.name)
25
26class 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)
Siobhan Tullyce652d02013-10-08 21:52:35 -040032 owner = models.ForeignKey(Slice, related_name="ownedNetworks", help_text="Slice that owns control of this Network")
Scott Baker58a9c7a2013-07-29 15:43:07 -070033
34 guaranteedBandwidth = models.IntegerField(default=0)
35 permitAllSlices = models.BooleanField(default=False)
36 permittedSlices = models.ManyToManyField(Slice, blank=True, related_name="availableNetworks")
Scott Baker87191e72013-08-06 08:55:07 -070037 slices = models.ManyToManyField(Slice, blank=True, related_name="networks", through="NetworkSlice")
Scott Baker58a9c7a2013-07-29 15:43:07 -070038 slivers = models.ManyToManyField(Sliver, blank=True, related_name="networks", through="NetworkSliver")
39
Scott Baker87191e72013-08-06 08:55:07 -070040 # 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 Baker58a9c7a2013-07-29 15:43:07 -070045 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
Tony Mack5b061472014-02-04 07:57:10 -050053 def can_update(self, user):
54 return self.slice.can_update(user)
55
Tony Mack5b061472014-02-04 07:57:10 -050056 @staticmethod
57 def select_by_user(user):
58 if user.is_admin:
59 qs = Network.objects.all()
60 else:
61 slice_ids = [s.id for s in Slice.select_by_user(user)]
62 qs = Network.objects.filter(id__in=slice_ids)
63 return qs
64
Scott Baker87191e72013-08-06 08:55:07 -070065class NetworkSlice(PlCoreBase):
66 # This object exists solely so we can implement the permission check when
67 # adding slices to networks. It adds no additional fields to the relation.
68
69 network = models.ForeignKey(Network)
70 slice = models.ForeignKey(Slice)
71
72 def save(self, *args, **kwds):
73 slice = self.slice
74 if (slice not in self.network.permittedSlices.all()) and (slice != self.network.owner) and (not self.network.permitAllSlices):
75 # to add a sliver to the network, then one of the following must be true:
76 # 1) sliver's slice is in network's permittedSlices list,
77 # 2) sliver's slice is network's owner, or
78 # 3) network's permitAllSlices is true
79 raise ValueError("Slice %s is not allowed to connect to network %s" % (str(slice), str(self.network)))
80
81 super(NetworkSlice, self).save(*args, **kwds)
82
83 def __unicode__(self): return u'%s-%s' % (self.network.name, self.slice.name)
84
Tony Mack5b061472014-02-04 07:57:10 -050085 def can_update(self, user):
86 return self.slice.can_update(user)
87
Tony Mack5b061472014-02-04 07:57:10 -050088 @staticmethod
89 def select_by_user(user):
90 if user.is_admin:
91 qs = NetworkSlice.objects.all()
92 else:
93 slice_ids = [s.id for s in Slice.select_by_user(user)]
94 qs = NetworkSlice.objects.filter(id__in=slice_ids)
95 return qs
96
Scott Baker58a9c7a2013-07-29 15:43:07 -070097class NetworkSliver(PlCoreBase):
98 network = models.ForeignKey(Network)
99 sliver = models.ForeignKey(Sliver)
Scott Baker026bfe72013-07-29 16:03:50 -0700100 ip = models.GenericIPAddressField(help_text="Sliver ip address", blank=True, null=True)
Scott Bakerf4df9522013-08-19 17:56:45 -0700101 port_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum port id")
Scott Baker58a9c7a2013-07-29 15:43:07 -0700102
103 def save(self, *args, **kwds):
Scott Baker87191e72013-08-06 08:55:07 -0700104 slice = self.sliver.slice
105 if (slice not in self.network.permittedSlices.all()) and (slice != self.network.owner) and (not self.network.permitAllSlices):
106 # to add a sliver to the network, then one of the following must be true:
107 # 1) sliver's slice is in network's permittedSlices list,
108 # 2) sliver's slice is network's owner, or
109 # 3) network's permitAllSlices is true
110 raise ValueError("Slice %s is not allowed to connect to network %s" % (str(slice), str(self.network)))
111
Scott Baker58a9c7a2013-07-29 15:43:07 -0700112 if (not self.ip) and (NO_OBSERVER):
113 from util.network_subnet_allocator import find_unused_address
114 self.ip = find_unused_address(self.network.subnet,
115 [x.ip for x in self.network.networksliver_set.all()])
116 super(NetworkSliver, self).save(*args, **kwds)
117
118 def __unicode__(self): return u'%s-%s' % (self.network.name, self.sliver.instance_name)
119
Tony Mack5b061472014-02-04 07:57:10 -0500120 def can_update(self, user):
121 return self.sliver.can_update(user)
122
Tony Mack5b061472014-02-04 07:57:10 -0500123 @staticmethod
124 def select_by_user(user):
125 if user.is_admin:
126 qs = NetworkSliver.objects.all()
127 else:
128 sliver_ids = [s.id for s in NetworkSliver.select_by_user(user)]
129 qs = NetworkSliver.objects.filter(id__in=sliver_ids)
130 return qs
131
Scott Baker58a9c7a2013-07-29 15:43:07 -0700132class Router(PlCoreBase):
133 name = models.CharField(max_length=32)
134 owner = models.ForeignKey(Slice, related_name="routers")
135 permittedNetworks = models.ManyToManyField(Network, blank=True, related_name="availableRouters")
136 networks = models.ManyToManyField(Network, blank=True, related_name="routers")
137
138 def __unicode__(self): return u'%s' % (self.name)
139
140class NetworkParameterType(PlCoreBase):
141 name = models.SlugField(help_text="The name of this parameter", max_length=128)
142 description = models.CharField(max_length=1024)
143
144 def __unicode__(self): return u'%s' % (self.name)
145
146class NetworkParameter(PlCoreBase):
147 parameter = models.ForeignKey(NetworkParameterType, related_name="parameters", help_text="The type of the parameter")
148 value = models.CharField(help_text="The value of this parameter", max_length=1024)
149
150 # The required fields to do a ObjectType lookup, and object_id assignment
151 content_type = models.ForeignKey(ContentType)
152 object_id = models.PositiveIntegerField()
153 content_object = generic.GenericForeignKey('content_type', 'object_id')
154
155 def __unicode__(self):
156 return self.parameter.name
157
158