a9936bd6312f041444265b55d33bff9db80824b9
[plstackapi.git] / planetstack / model_policies / model_policy_Slice.py
1
2 def handle(slice):
3     from core.models import Controller, ControllerSlice, SiteDeployment, Network, NetworkSlice,NetworkTemplate, Slice
4     from collections import defaultdict
5
6     # slice = Slice.get(slice_id)
7
8     controller_slices = ControllerSlice.objects.filter(slice=slice)
9     existing_controllers = [cs.controller for cs in controller_slices] 
10         
11     all_controllers = Controller.objects.all() 
12     for controller in all_controllers:
13         if controller not in existing_controllers:
14             sd = ControllerSlice(slice=slice, controller=controller)
15             sd.save()
16
17     # make sure slice has at least 1 public and 1 private networkd
18     public_nets = []
19     private_net = None
20     networks = Network.objects.filter(owner=slice)
21     for network in networks:
22         if network.template.name == 'Public dedicated IPv4':
23             public_nets.append(network)
24         elif network.template.name == 'Public shared IPv4':
25             public_nets.append(network)
26         elif network.template.name == 'Private':
27             private_net = network
28     if not public_nets:
29                 # ensure there is at least one public network, and default it to dedicated
30         nat_net = Network(
31                 name = slice.name+'-nat',
32                     template = NetworkTemplate.objects.get(name='Public shared IPv4'),
33                 owner = slice
34                 )
35         nat_net.save()
36         public_nets.append(nat_net)
37
38     if not private_net:
39         private_net = Network(
40         name = slice.name+'-private',
41         template = NetworkTemplate.objects.get(name='Private'),
42         owner = slice
43         )
44         private_net.save()
45     # create slice networks
46     public_net_slice = None
47     private_net_slice = None
48     net_slices = NetworkSlice.objects.filter(slice=slice, network__in=[private_net]+public_nets)
49     for net_slice in net_slices:
50         if net_slice.network in public_nets:
51             public_net_slice = net_slice
52         elif net_slice.network == private_net:
53             private_net_slice = net_slice
54     if not public_net_slice:
55         public_net_slice = NetworkSlice(slice=slice, network=public_nets[0])
56         public_net_slice.save()
57     if not private_net_slice:
58         private_net_slice = NetworkSlice(slice=slice, network=private_net)
59         private_net_slice.save()
60
61
62
63