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
   8
   9 class SyncSlivers(OpenStackSyncStep):
  10     provides=[Sliver]
  11     requested_interval=0
  12
  13     def fetch_pending(self):
  14         return Sliver.objects.filter(Q(enacted__lt=F('updated')) | Q(enacted=None))
  15
  16     def get_requested_networks(self, slice):
  17         network_ids = [x.network_id for x in slice.networks.all()]
  18
  19         if slice.network_id is not None:
  20             network_ids.append(slice.network_id)
  21
  22         networks = []
  23         for network_id in network_ids:
  24             networks.append({"net-id": network_id})
  25
  26         return networks
  27
  28     def sync_record(self, sliver):
  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)
  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             pubkeys.append(sliver.creator.public_key)
  43             instance = self.driver.spawn_instance(name=sliver.name,
  44                                 key_name = sliver.creator.keyname,
  45                                 image_id = sliver.image.image_id,
  46                                 hostname = sliver.node.name,
  47                                 pubkeys = pubkeys,
  48                                 nics = nics )
  49             sliver.instance_id = instance.id
  50             sliver.instance_name = getattr(instance, 'OS-EXT-SRV-ATTR:instance_name')
  51
  52         if sliver.instance_id and metadata_update:
  53             self.driver.update_instance_metadata(sliver.instance_id, metadata_update)
  54
  55         sliver.save()