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