planetstack/observer/steps/sync_slivers.py

   1 import os
   2 import base64
   3 from django.db.models import F, Q
   4 from planetstack.config import Config
   5 from observer.openstacksyncstep import OpenStackSyncStep
   6 from core.models.sliver import Sliver
   7 from core.models.slice import SlicePrivilege, SliceDeployments
   8 from util.logger import Logger, logging
   9
  10 logger = Logger(level=logging.INFO)
  11 m util.logger import Logger, logging
  12
  13 logger = Logger(level=logging.INFO)
  14
  15 class SyncSlivers(OpenStackSyncStep):
  16     provides=[Sliver]
  17     requested_interval=0
  18
  19     def fetch_pending(self):
  20         return Sliver.objects.filter(Q(enacted__lt=F('updated')) | Q(enacted=None))
  21
  22     def get_requested_networks(self, slice):
  23         network_ids = [x.network_id for x in slice.networks.all()]
  24
  25         if slice.network_id is not None:
  26             network_ids.append(slice.network_id)
  27
  28         networks = []
  29         for network_id in network_ids:
  30             networks.append({"net-id": network_id})
  31
  32         return networks
  33
  34     def sync_record(self, sliver):
  35         logger.info("sync'ing sliver %s" % sliver)
  36         metadata_update = {}
  37         if ("numberCores" in sliver.changed_fields):
  38             metadata_update["cpu_cores"] = str(sliver.numberCores)
  39
  40         for tag in sliver.slice.tags.all():
  41             if tag.name.startswith("sysctl-"):
  42                 metadata_update[tag.name] = tag.value
  43
  44         if not sliver.instance_id:
  45             nics = self.get_requested_networks(sliver.slice)
  46             file("/tmp/scott-manager","a").write("slice: %s\nreq: %s\n" % (str(sliver.slice.name), str(nics)))
  47             slice_memberships = SlicePrivilege.objects.filter(slice=sliver.slice)
  48             pubkeys = [sm.user.public_key for sm in slice_memberships if sm.user.public_key is not None]
  49             if sliver.creator.public_key:
  50                 pubkeys.append(sliver.creator.public_key)
  51             driver = self.driver.client_driver(caller=sliver.creator, tenant=sliver.slice.name, deployment=sliver.deploymentNetwork.name)
  52             # look up image id
  53             deployment_driver = self.driver.admin_driver(deployment=sliver.deploymentNetwork.name)
  54             image_id = None
  55             images = deployment_driver.shell.glance.get_images()
  56             for image in images:
  57                 if image['name'] == sliver.image.name:
  58                     image_id = image['id']
  59
  60             # look up key name at the deployment
  61             keyname = None
  62             slice_deployments = SliceDeployments.objects.filter(slice = sliver.slice,
  63                                                                deployment = sliver.deploymentNetwork)
  64             for slice_deployment in slice_deployments:
  65                 if slice_deployment.keyname:
  66                     keyname = slice_deployment.keyname
  67                     break
  68
  69             instance = driver.spawn_instance(name=sliver.name,
  70                                 key_name = keyname,
  71                                 image_id = image_id,
  72                                 hostname = sliver.node.name,
  73                                 pubkeys = pubkeys,
  74                                 nics = nics )
  75             sliver.instance_id = instance.id
  76             sliver.instance_name = getattr(instance, 'OS-EXT-SRV-ATTR:instance_name')
  77
  78         if sliver.instance_id and metadata_update:
  79             driver.update_instance_metadata(sliver.instance_id, metadata_update)
  80
  81         sliver.save()