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