dcc3b1e2562cb5effd25cdb107f88fd969f20abd
[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 from observer.ansible import *
11
12 logger = Logger(level=logging.INFO)
13
14 class SyncSlivers(OpenStackSyncStep):
15     provides=[Sliver]
16     requested_interval=0
17
18     def get_userdata(self, sliver):
19         userdata = 'opencloud:\n   slicename: "%s"\n   hostname: "%s"\n' % (sliver.slice.name, sliver.node.name)
20         return userdata
21
22     def sync_record(self, sliver):
23         logger.info("sync'ing sliver:%s deployment:%s " % (sliver, sliver.node.deployment))
24         metadata_update = {}
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         # public keys
32         slice_memberships = SlicePrivilege.objects.filter(slice=sliver.slice)
33         pubkeys = [sm.user.public_key for sm in slice_memberships if sm.user.public_key]
34         if sliver.creator.public_key:
35             pubkeys.append(sliver.creator.public_key)
36
37         if sliver.slice.creator.public_key:
38             pubkeys.append(sliver.slice.creator.public_key) 
39
40         nics = []
41         networks = [ns.network for ns in NetworkSlice.objects.filter(slice=sliver.slice)]   
42         network_deployments = NetworkDeployments.objects.filter(network__in=networks, 
43                                                                 deployment=sliver.node.deployment)
44
45         for network_deployment in network_deployments:
46             if network_deployment.network.template.visibility == 'private' and \
47                network_deployment.network.template.translation == 'none': 
48                 nics.append({'net-id': network_deployment.net_id})
49
50         # now include network template
51         network_templates = [network.template.sharedNetworkName for network in networks \
52                              if network.template.sharedNetworkName]
53
54         driver = self.driver.client_driver(caller=sliver.creator, tenant=sliver.slice.name, deployment=sliver.deploymentNetwork.name)
55         for net in driver.shell.quantum.list_networks()['networks']:
56             if net['name'] in network_templates: 
57                 nics.append({'net-id': net['id']}) 
58
59         # look up image id
60         deployment_driver = self.driver.admin_driver(deployment=sliver.deploymentNetwork.name)
61         image_id = None
62         images = deployment_driver.shell.glance.get_images()
63         for image in images:
64             if image['name'] == sliver.image.name:
65                 image_id = image['id']
66                 
67         # look up key name at the deployment
68         # create/fetch keypair
69         keyname = None
70         keyname = sliver.creator.email.lower().replace('@', 'AT').replace('.', '') +\
71                   sliver.slice.name
72         key_fields =  {'name': keyname,
73                        'public_key': sliver.creator.public_key}
74             
75
76         userData = self.get_userdata(sliver)
77         if sliver.userData:
78             userData = sliver.userData
79             
80         tenant_fields = {'endpoint':slice_deployment.deployment.auth_url,
81                      'admin_user': slice_deployment.deployment.admin_user,
82                      'admin_password': slice_deployment.deployment.admin_password,
83                      'admin_tenant': 'admin',
84                      'tenant': slice_deployment.slice.name,
85                      'tenant_description': slice_deployment.slice.description,
86                      'roles':roles,
87                      'max_instances':max_instances,
88                      'name':sliver.name,
89                      'image_id':image_id,
90                      'key_name':keyname,
91                      'flavor_id':sliver.flavor.flavor,
92                      'nics':nics,
93                      'meta':metadata_update,
94                      'key':key_fields,
95                      'user_data':userData}
96
97         res = run_template('sync_slivers.yaml', tenant_fields)
98         if (len(res)!=2):
99             raise Exception('Could not sync tenants for slice %s'%slice_deployment.slice.name)
100         else:
101             sliver_id = res[1]['id'] # 0 is for the key
102
103             sliver.instance_id = instance.id
104             sliver.instance_name = getattr(instance, 'OS-EXT-SRV-ATTR:instance_name')
105             sliver.save()    
106
107     def delete_record(self, sliver):
108         if sliver.instance_id:
109             driver = self.driver.client_driver(caller=sliver.creator, 
110                                                tenant=sliver.slice.name,
111                                                deployment=sliver.deploymentNetwork.name)
112             driver.destroy_instance(sliver.instance_id)