blob: c76004ba25057715f547ab92b9b6b2be822759ea [file] [log] [blame]
Scott Baker58a9c7a2013-07-29 15:43:07 -07001import os
2import socket
3from django.db import models
Tony Macke9b08692014-04-07 19:38:28 -04004from core.models import PlCoreBase, Site, Slice, Sliver, Deployment
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):
Tony Mack31683c82014-04-02 15:39:32 -040054 return self.owner.can_update(user)
Tony Mack5b061472014-02-04 07:57:10 -050055
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:
Tony Mack5efa1332014-04-02 15:45:48 -040061 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)
Tony Mack5b061472014-02-04 07:57:10 -050064 return qs
65
Tony Macke9b08692014-04-07 19:38:28 -040066class NetworkDeployments(PlCoreBase):
67 # Stores the openstack ids at various deployments
68 network = models.ForeignKey(Network)
69 deployment = models.ForeignKey(Deployment)
70 network_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
Scott Baker87191e72013-08-06 08:55:07 -070088class 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
Tony Mack5b061472014-02-04 07:57:10 -0500108 def can_update(self, user):
109 return self.slice.can_update(user)
110
Tony Mack5b061472014-02-04 07:57:10 -0500111 @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
Scott Baker58a9c7a2013-07-29 15:43:07 -0700120class NetworkSliver(PlCoreBase):
121 network = models.ForeignKey(Network)
122 sliver = models.ForeignKey(Sliver)
Scott Baker026bfe72013-07-29 16:03:50 -0700123 ip = models.GenericIPAddressField(help_text="Sliver ip address", blank=True, null=True)
Scott Bakerf4df9522013-08-19 17:56:45 -0700124 port_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum port id")
Scott Baker58a9c7a2013-07-29 15:43:07 -0700125
126 def save(self, *args, **kwds):
Scott Baker87191e72013-08-06 08:55:07 -0700127 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
Scott Baker58a9c7a2013-07-29 15:43:07 -0700135 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
Tony Mack5b061472014-02-04 07:57:10 -0500143 def can_update(self, user):
144 return self.sliver.can_update(user)
145
Tony Mack5b061472014-02-04 07:57:10 -0500146 @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
Scott Baker58a9c7a2013-07-29 15:43:07 -0700155class 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
163class 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
169class 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