Sapan Bhatia | b3aeb1b | 2014-09-03 15:39:13 -0400 | [diff] [blame] | 1 | |
| 2 | def handle(slice): |
Tony Mack | d2046f3 | 2014-10-08 22:54:02 -0400 | [diff] [blame] | 3 | from core.models import SiteDeployments,SliceDeployments,Deployment,Network,NetworkSlice,NetworkTemplate |
Sapan Bhatia | 728acce | 2014-09-15 03:48:20 -0400 | [diff] [blame] | 4 | from collections import defaultdict |
Sapan Bhatia | b3aeb1b | 2014-09-03 15:39:13 -0400 | [diff] [blame] | 5 | site_deployments = SiteDeployments.objects.all() |
| 6 | site_deploy_lookup = defaultdict(list) |
| 7 | for site_deployment in site_deployments: |
| 8 | site_deploy_lookup[site_deployment.site].append(site_deployment.deployment) |
| 9 | |
| 10 | slice_deployments = SliceDeployments.objects.all() |
| 11 | slice_deploy_lookup = defaultdict(list) |
| 12 | for slice_deployment in slice_deployments: |
| 13 | slice_deploy_lookup[slice_deployment.slice].append(slice_deployment.deployment) |
| 14 | |
| 15 | all_deployments = Deployment.objects.all() |
| 16 | # slices are added to all deployments for now |
| 17 | expected_deployments = all_deployments |
| 18 | #expected_deployments = site_deploy_lookup[slice.site] |
| 19 | for expected_deployment in expected_deployments: |
| 20 | if slice not in slice_deploy_lookup or \ |
| 21 | expected_deployment not in slice_deploy_lookup[slice]: |
| 22 | sd = SliceDeployments(slice=slice, deployment=expected_deployment) |
| 23 | sd.save() |
| 24 | |
Tony Mack | d2046f3 | 2014-10-08 22:54:02 -0400 | [diff] [blame] | 25 | # make sure slice has at least 1 public and 1 private networkd |
Scott Baker | cb7649c | 2014-10-16 00:43:54 -0700 | [diff] [blame^] | 26 | public_nets = [] |
Tony Mack | d2046f3 | 2014-10-08 22:54:02 -0400 | [diff] [blame] | 27 | private_net = None |
| 28 | networks = Network.objects.filter(owner=slice) |
| 29 | for network in networks: |
| 30 | if network.template.name == 'Public dedicated IPv4': |
Scott Baker | cb7649c | 2014-10-16 00:43:54 -0700 | [diff] [blame^] | 31 | public_nets.append(network) |
| 32 | elif network.template.name == 'Public shared IPv4': |
| 33 | public_nets.append(network) |
Tony Mack | d2046f3 | 2014-10-08 22:54:02 -0400 | [diff] [blame] | 34 | elif network.template.name == 'Private': |
Scott Baker | cb7649c | 2014-10-16 00:43:54 -0700 | [diff] [blame^] | 35 | private_net = network |
| 36 | if not public_nets: |
| 37 | # ensure there is at least one public network, and default it to dedicated |
| 38 | dedicated_public_net = Network( |
| 39 | name = slice.name+'-public', |
| 40 | template = NetworkTemplate.objects.get(name='Public dedicated IPv4'), |
| 41 | owner = slice |
| 42 | ) |
| 43 | dedicated_public_net.save() |
| 44 | public_nets.append(dedicated_public_net) |
| 45 | |
Tony Mack | d2046f3 | 2014-10-08 22:54:02 -0400 | [diff] [blame] | 46 | if not private_net: |
| 47 | private_net = Network( |
| 48 | name = slice.name+'-private', |
| 49 | template = NetworkTemplate.objects.get(name='Private'), |
| 50 | owner = slice |
| 51 | ) |
| 52 | private_net.save() |
| 53 | # create slice networks |
| 54 | public_net_slice = None |
| 55 | private_net_slice = None |
Scott Baker | cb7649c | 2014-10-16 00:43:54 -0700 | [diff] [blame^] | 56 | net_slices = NetworkSlice.objects.filter(slice=slice, network__in=[private_net]+public_nets) |
Tony Mack | d2046f3 | 2014-10-08 22:54:02 -0400 | [diff] [blame] | 57 | for net_slice in net_slices: |
Scott Baker | cb7649c | 2014-10-16 00:43:54 -0700 | [diff] [blame^] | 58 | if net_slice.network in public_nets: |
| 59 | public_net_slice = net_slice |
Tony Mack | d2046f3 | 2014-10-08 22:54:02 -0400 | [diff] [blame] | 60 | elif net_slice.network == private_net: |
Scott Baker | cb7649c | 2014-10-16 00:43:54 -0700 | [diff] [blame^] | 61 | private_net_slice = net_slice |
Tony Mack | d2046f3 | 2014-10-08 22:54:02 -0400 | [diff] [blame] | 62 | if not public_net_slice: |
Scott Baker | cb7649c | 2014-10-16 00:43:54 -0700 | [diff] [blame^] | 63 | public_net_slice = NetworkSlice(slice=slice, network=public_nets[0]) |
Tony Mack | d2046f3 | 2014-10-08 22:54:02 -0400 | [diff] [blame] | 64 | public_net_slice.save() |
| 65 | if not private_net_slice: |
| 66 | private_net_slice = NetworkSlice(slice=slice, network=private_net) |
Scott Baker | cb7649c | 2014-10-16 00:43:54 -0700 | [diff] [blame^] | 67 | private_net_slice.save() |
| 68 | |
| 69 | |
| 70 | |
| 71 | |