planetstack/model_policies/model_policy_Network.py

   1 from core.models import *
   2
   3 def handle(network):
   4         from core.models import ControllerSlice,ControllerNetwork, Network
   5         from collections import defaultdict
   6
   7         # network = Network.get(network_id)
   8         # network controllers are not visible to users. We must ensure
   9         # networks are deployed at all deploymets available to their slices.
  10         slice_controllers = ControllerSlice.objects.all()
  11         slice_deploy_lookup = defaultdict(list)
  12         for slice_controller in slice_controllers:
  13                 slice_deploy_lookup[slice_controller.slice].append(slice_controller.controller)
  14
  15         network_controllers = ControllerNetwork.objects.all()
  16         network_deploy_lookup = defaultdict(list)
  17         for network_controller in network_controllers:
  18                 network_deploy_lookup[network_controller.network].append(network_controller.controller)
  19
  20         expected_controllers = slice_deploy_lookup[network.owner]
  21         for expected_controller in expected_controllers:
  22                 if network not in network_deploy_lookup or \
  23                   expected_controller not in network_deploy_lookup[network]:
  24                         nd = ControllerNetwork(network=network, controller=expected_controller)
  25                         nd.save()