planetstack/model_policies/model_policy_Slice.py

   1
   2 def handle(slice):
   3         import pdb
   4         pdb.set_trace()
   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)
  11
  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)
  16
  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)
  23                         sd.save()
  24
  25         # make sure slice has at least 1 public and 1 private networkd
  26         public_nets = []
  27         private_net = None
  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':
  35                         private_net = network
  36         if not public_nets:
  37                 # ensure there is at least one public network, and default it to dedicated
  38                 nat_net = Network(
  39                     name = slice.name+'-nat',
  40                     template = NetworkTemplate.objects.get(name='Public shared IPv4'),
  41                     owner = slice
  42                     )
  43                 nat_net.save()
  44                 public_nets.append(nat_net)
  45
  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
  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()
  68
  69
  70
  71