dcedd1d01b795684c03f7adc11bc49f2998a8e8c
[plstackapi.git] / planetstack / openstack_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 Slice, SlicePrivilege, SliceDeployments
8 from core.models.network import Network, NetworkSlice, 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 get_userdata(self, sliver):
18         userdata = 'opencloud:\n   slicename: "%s"\n   hostname: "%s"\n' % (sliver.slice.name, sliver.node.name)
19         return userdata
20
21     def sync_record(self, sliver):
22         logger.info("sync'ing sliver:%s deployment:%s " % (sliver, sliver.node.deployment))
23         metadata_update = {}
24         if ("numberCores" in sliver.changed_fields):
25             metadata_update["cpu_cores"] = str(sliver.numberCores)
26
27         for tag in sliver.slice.tags.all():
28             if tag.name.startswith("sysctl-"):
29                 metadata_update[tag.name] = tag.value
30
31         if not sliver.instance_id:
32             driver = self.driver.client_driver(caller=sliver.creator, tenant=sliver.slice.name, deployment=sliver.deploymentNetwork.name)
33             # public keys
34             slice_memberships = SlicePrivilege.objects.filter(slice=sliver.slice)
35             pubkeys = [sm.user.public_key for sm in slice_memberships if sm.user.public_key]
36             if sliver.creator.public_key:
37                 pubkeys.append(sliver.creator.public_key)
38             if sliver.slice.creator.public_key:
39                 pubkeys.append(sliver.slice.creator.public_key) 
40             # netowrks
41             # include all networks available to the slice and/or associated network templates
42             nics = []
43             networks = [ns.network for ns in NetworkSlice.objects.filter(slice=sliver.slice)]   
44             network_deployments = NetworkDeployments.objects.filter(network__in=networks, 
45                                                                     deployment=sliver.node.deployment)
46             # Gather private networks first. This includes networks with a template that has
47             # visibility = private and translation = none
48             for network_deployment in network_deployments:
49                 if network_deployment.network.template.visibility == 'private' and \
50                    network_deployment.network.template.translation == 'none': 
51                     nics.append({'net-id': network_deployment.net_id})
52     
53             # now include network template
54             network_templates = [network.template.sharedNetworkName for network in networks \
55                                  if network.template.sharedNetworkName]
56             #logger.info("%s %s %s %s" % (driver.shell.quantum.username, driver.shell.quantum.password, driver.shell.quantum.tenant, driver.shell.quantum.url))
57             for net in driver.shell.quantum.list_networks()['networks']:
58                 if net['name'] in network_templates: 
59                     nics.append({'net-id': net['id']}) 
60
61             # look up image id
62             deployment_driver = self.driver.admin_driver(deployment=sliver.deploymentNetwork.name)
63             image_id = None
64             images = deployment_driver.shell.glance.get_images()
65             for image in images:
66                 if image['name'] == sliver.image.name:
67                     image_id = image['id']
68                     
69             # look up key name at the deployment
70             # create/fetch keypair
71             keyname = None
72             if sliver.creator.public_key:
73                 keyname = sliver.creator.email.lower().replace('@', 'AT').replace('.', '') +\
74                           sliver.slice.name
75                 key_fields =  {'name': keyname,
76                                'public_key': sliver.creator.public_key}
77                 driver.create_keypair(**key_fields)
78
79             userData = self.get_userdata(sliver)
80             if sliver.userData:
81                 userData = sliver.userData
82
83             instance = driver.spawn_instance(name=sliver.name,
84                                 key_name = keyname,
85                                 image_id = image_id,
86                                 hostname = sliver.node.name,
87                                 pubkeys = pubkeys,
88                                 nics = nics,
89                                 userdata = userData,
90                                 flavor_name = sliver.flavor.flavor )
91             sliver.instance_id = instance.id
92             sliver.instance_name = getattr(instance, 'OS-EXT-SRV-ATTR:instance_name')
93             sliver.save()    
94
95         if sliver.instance_id and metadata_update:
96             driver.update_instance_metadata(sliver.instance_id, metadata_update)
97
98     def delete_record(self, sliver):
99         if sliver.instance_id:
100             driver = self.driver.client_driver(caller=sliver.creator, 
101                                                tenant=sliver.slice.name,
102                                                deployment=sliver.deploymentNetwork.name)
103             driver.destroy_instance(sliver.instance_id)