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
   8 class SyncSlivers(OpenStackSyncStep):
   9     provides=[Sliver]
  10     requested_interval=0
  11
  12     def fetch_pending(self):
  13         return Sliver.objects.filter(Q(enacted__lt=F('updated')) | Q(enacted=None))
  14
  15     def get_requested_networks(self, slice):
  16         network_ids = [x.network_id for x in slice.networks.all()]
  17
  18         if slice.network_id is not None:
  19             network_ids.append(slice.network_id)
  20
  21         networks = []
  22         for network_id in network_ids:
  23             networks.append({"net-id": network_id})
  24
  25         return networks
  26
  27     def sync_record(self, slice):
  28         metadata_update = {}
  29         if ("numberCores" in sliver.changed_fields):
  30             metadata_update["cpu_cores"] = str(sliver.numberCores)
  31
  32         for tag in sliver.slice.tags.all():
  33             if tag.name.startswith("sysctl-"):
  34                 metadata_update[tag.name] = tag.value
  35
  36         if not sliver.instance_id:
  37             nics = self.get_requested_networks(sliver.slice)
  38             file("/tmp/scott-manager","a").write("slice: %s\nreq: %s\n" % (str(sliver.slice.name), str(nics)))
  39             slice_memberships = SliceMembership.objects.filter(slice=sliver.slice)
  40             pubkeys = [sm.user.public_key for sm in slice_memberships if sm.user.public_key]
  41             pubkeys.append(sliver.creator.public_key)
  42             instance = self.driver.spawn_instance(name=sliver.name,
  43                                 key_name = sliver.creator.keyname,
  44                                 image_id = sliver.image.image_id,
  45                                 hostname = sliver.node.name,
  46                                 pubkeys = pubkeys,
  47                                 nics = nics )
  48             sliver.instance_id = instance.id
  49             sliver.instance_name = getattr(instance, 'OS-EXT-SRV-ATTR:instance_name')
  50
  51         if sliver.instance_id and metadata_update:
  52             self.driver.update_instance_metadata(sliver.instance_id, metadata_update)
  53
  54         sliver.save()