Renamed SliceDeployment back to SliceDeployments

[plstackapi.git] / planetstack / model_policies / model_policy_Slice.py

diff --git a/planetstack/model_policies/model_policy_Slice.py b/planetstack/model_policies/model_policy_Slice.py
index 493975f..56e2f53 100644 (file)
--- a/planetstack/model_policies/model_policy_Slice.py
+++ b/planetstack/model_policies/model_policy_Slice.py
@@ -1,13 +1,13 @@
 
 def handle(slice):
-       from core.models import SiteDeployments,SliceDeployments,Deployment,Network,NetworkSlice,NetworkTemplate
+       from core.models import SiteDeployments,SliceDeployment,Deployment,Network,NetworkSlice,NetworkTemplate
        from collections import defaultdict
        site_deployments = SiteDeployments.objects.all()
        site_deploy_lookup = defaultdict(list)
        for site_deployment in site_deployments:
                site_deploy_lookup[site_deployment.site].append(site_deployment.deployment)
        
-       slice_deployments = SliceDeployments.objects.all()
+       slice_deployments = SliceDeployment.objects.all()
        slice_deploy_lookup = defaultdict(list)
        for slice_deployment in slice_deployments:
                slice_deploy_lookup[slice_deployment.slice].append(slice_deployment.deployment)
@@ -19,26 +19,30 @@ def handle(slice):
        for expected_deployment in expected_deployments:
                if slice not in slice_deploy_lookup or \
                   expected_deployment not in slice_deploy_lookup[slice]:
-                       sd = SliceDeployments(slice=slice, deployment=expected_deployment)
+                       sd = SliceDeployment(slice=slice, deployment=expected_deployment)
                        sd.save()
 
        # make sure slice has at least 1 public and 1 private networkd
-       public_net = None
+       public_nets = []
        private_net = None
        networks = Network.objects.filter(owner=slice)
        for network in networks:
                if network.template.name == 'Public dedicated IPv4':
-                       public_net = network
+                       public_nets.append(network)
+               elif network.template.name == 'Public shared IPv4':
+                       public_nets.append(network)
                elif network.template.name == 'Private':
-                       private_net = network 
-       if not public_net:
-               public_net = Network(
-               name = slice.name+'-public',
-               template = NetworkTemplate.objects.get(name='Public dedicated IPv4'),
-               owner = slice
-               )
-               public_net.save()
-        
+                       private_net = network
+       if not public_nets:
+                # ensure there is at least one public network, and default it to dedicated
+               nat_net = Network(
+                   name = slice.name+'-nat',
+                   template = NetworkTemplate.objects.get(name='Public shared IPv4'),
+                   owner = slice
+                   )
+               nat_net.save()
+                public_nets.append(nat_net)
+
        if not private_net:
                private_net = Network(
                name = slice.name+'-private',
@@ -49,19 +53,19 @@ def handle(slice):
        # create slice networks
        public_net_slice = None
        private_net_slice = None
-       net_slices = NetworkSlice.objects.filter(slice=slice, network__in=[public_net,private_net])
+       net_slices = NetworkSlice.objects.filter(slice=slice, network__in=[private_net]+public_nets)
        for net_slice in net_slices:
-               if net_slice.network == public_net:
-                       public_net_slice = net_slice 
+               if net_slice.network in public_nets:
+                       public_net_slice = net_slice
                elif net_slice.network == private_net:
-                       private_net_slice = net_slice 
+                       private_net_slice = net_slice
        if not public_net_slice:
-               public_net_slice = NetworkSlice(slice=slice, network=public_net)
+               public_net_slice = NetworkSlice(slice=slice, network=public_nets[0])
                public_net_slice.save()
        if not private_net_slice:
                private_net_slice = NetworkSlice(slice=slice, network=private_net)
-               private_net_slice.save()                
-                      
-             
-        
-        
+               private_net_slice.save()
+
+
+
+