5 from core.models import Controller, ControllerSiteDeployments, ControllerSlices,Controller,Network,NetworkSlice,NetworkTemplate
6 from collections import defaultdict
7 ctrl_site_deployments = ControllerSiteDeployments.objects.all()
8 site_deploy_lookup = defaultdict(list)
9 for ctrl_site_deployment in ctrl_site_deployments:
10 site_deploy_lookup[ctrl_site_deployment.site_deployment].append(ctrl_site_deployment)
12 slice_controllers = ControllerSlices.objects.all()
13 slice_deploy_lookup = defaultdict(list)
14 for slice_controller in slice_controllers:
15 slice_deploy_lookup[slice_controller.slice].append(slice_controller)
17 #expected_controllers = site_deploy_lookup[slice.site]
18 all_controllers = Controller.objects.all()
19 for expected_controller in all_controllers:
20 if slice not in slice_deploy_lookup or \
21 expected_controller not in slice_deploy_lookup[slice]:
22 sd = ControllerSlices(slice=slice, controller=expected_controller)
25 # make sure slice has at least 1 public and 1 private networkd
28 networks = Network.objects.filter(owner=slice)
29 for network in networks:
30 if network.template.name == 'Public dedicated IPv4':
31 public_nets.append(network)
32 elif network.template.name == 'Public shared IPv4':
33 public_nets.append(network)
34 elif network.template.name == 'Private':
37 # ensure there is at least one public network, and default it to dedicated
39 name = slice.name+'-nat',
40 template = NetworkTemplate.objects.get(name='Public shared IPv4'),
44 public_nets.append(nat_net)
47 private_net = Network(
48 name = slice.name+'-private',
49 template = NetworkTemplate.objects.get(name='Private'),
53 # create slice networks
54 public_net_slice = None
55 private_net_slice = None
56 net_slices = NetworkSlice.objects.filter(slice=slice, network__in=[private_net]+public_nets)
57 for net_slice in net_slices:
58 if net_slice.network in public_nets:
59 public_net_slice = net_slice
60 elif net_slice.network == private_net:
61 private_net_slice = net_slice
62 if not public_net_slice:
63 public_net_slice = NetworkSlice(slice=slice, network=public_nets[0])
64 public_net_slice.save()
65 if not private_net_slice:
66 private_net_slice = NetworkSlice(slice=slice, network=private_net)
67 private_net_slice.save()