X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=planetstack%2Fopenstack%2Fdriver.py;h=6fb81a5e3bb91ee1c5f7ec120a815798ebf1d474;hb=9dbe365846ccc1c1a5a95958a8bbd32db15277f8;hp=b6995b8a37d2794346d1ea623b3ae87c1d62967d;hpb=387a73f79ab3fb6b9c2168f9c8975956281d4bfc;p=plstackapi.git diff --git a/planetstack/openstack/driver.py b/planetstack/openstack/driver.py index b6995b8..6fb81a5 100644 --- a/planetstack/openstack/driver.py +++ b/planetstack/openstack/driver.py @@ -1,9 +1,10 @@ import commands +import hashlib from planetstack.config import Config +from core.models import Deployment try: from openstack.client import OpenStackClient - from openstack.driver import OpenStackDriver has_openstack = True except: has_openstack = False @@ -12,41 +13,51 @@ manager_enabled = Config().api_nova_enabled class OpenStackDriver: - def __init__(self, config = None, client=None): + def __init__(self, config = None, client=None): if config: self.config = Config(config) else: - self.config = Config() - - self.admin_client = OpenStackClient() - self.admin_user = self.admin_client.keystone.users.find(name=self.admin_client.keystone.username) + self.config = Config() if client: self.shell = client - else: - self.shell = OpenStackClient() - def client_driver(self, caller=None, tenant=None): + self.enabled = manager_enabled + self.has_openstack = has_openstack + self.deployment = None + self.admin_user = None + + def client_driver(self, caller=None, tenant=None, deployment=None): if caller: auth = {'username': caller.email, 'password': hashlib.md5(caller.password).hexdigest()[:6], 'tenant': tenant} - client = OpenStackClient(**auth) + client = OpenStackClient(deployment=deployment, **auth) else: - client = OpenStackClient(tenant=tenant) + admin_driver = self.admin_driver(tenant=tenant, deployment=deployment) + client = OpenStackClient(tenant=tenant, deployment=admin_driver.deployment) + driver = OpenStackDriver(client=client) + #driver.admin_user = admin_driver.admin_user + #driver.deployment = admin_driver.deployment return driver - def admin_driver(self, tenant=None): - client = OpenStackClient(tenant=tenant) - driver = OpenStackDriver(client=client) + def admin_driver(self, tenant=None, deployment=None): + deployment = Deployment.objects.get(name=deployment) + client = OpenStackClient(tenant=tenant, deployment=deployment) + driver = OpenStackDriver(client=client) + driver.admin_user = client.keystone.users.find(name=deployment.admin_user) + driver.deployment = deployment + return driver def create_role(self, name): roles = self.shell.keystone.roles.findall(name=name) - if not roles: + roles_title = self.shell.keystone.roles.findall(name=name.title()) + roles_found = roles + roles_title + if not roles_found: role = self.shell.keystone.roles.create(name) else: - role = roles[0] + role = roles_found[0] return role def delete_role(self, filter): @@ -106,12 +117,27 @@ class OpenStackDriver: for key in keys: self.shell.nova.keypairs.delete(key) self.shell.keystone.users.delete(user) - return 1 + return 1 + + def get_admin_role(self): + role = None + for admin_role_name in ['admin', 'Admin']: + roles = self.shell.keystone.roles.findall(name=admin_role_name) + if roles: + role = roles[0] + break + return role def add_user_role(self, kuser_id, tenant_id, role_name): user = self.shell.keystone.users.find(id=kuser_id) tenant = self.shell.keystone.tenants.find(id=tenant_id) - role = self.shell.keystone.roles.find(name=role_name) + # admin role can be lowercase or title. Look for both + role = None + if role_name.lower() == 'admin': + role = self.get_admin_role() + else: + # look up non admin role or force exception when admin role isnt found + role = self.shell.keystone.roles.find(name=role_name) role_found = False user_roles = user.list_roles(tenant.id) @@ -126,7 +152,13 @@ class OpenStackDriver: def delete_user_role(self, kuser_id, tenant_id, role_name): user = self.shell.keystone.users.find(id=kuser_id) tenant = self.shell.keystone.tenants.find(id=tenant_id) - role = self.shell.keystone.roles.find(name=role_name) + # admin role can be lowercase or title. Look for both + role = None + if role_name.lower() == 'admin': + role = self.get_admin_role() + else: + # look up non admin role or force exception when admin role isnt found + role = self.shell.keystone.roles.find(name=role_name) role_found = False user_roles = user.list_roles(tenant.id) @@ -235,17 +267,21 @@ class OpenStackDriver: subnet = snet if not subnet: + # HACK: Add metadata route -- Neutron does not reliably supply this + metadata_ip = cidr_ip.replace("0/24", "3") + allocation_pools = [{'start': start, 'end': end}] subnet = {'subnet': {'name': name, 'network_id': network_id, 'ip_version': ip_version, 'cidr': cidr_ip, - 'dns_nameservers': ['8.8.8.8', '8.8.4.4'], + #'dns_nameservers': ['8.8.8.8', '8.8.4.4'], + 'host_routes': [{'destination':'169.254.169.254/32','nexthop':metadata_ip}], + 'gateway_ip': None, 'allocation_pools': allocation_pools}} subnet = self.shell.quantum.create_subnet(subnet)['subnet'] - self.add_external_route(subnet) - # TODO: Add route to external network - # e.g. # route add -net 10.0.3.0/24 dev br-ex gw 10.100.0.5 + # self.add_external_route(subnet) + return subnet def update_subnet(self, id, fields): @@ -381,33 +417,42 @@ class OpenStackDriver: return (subnet_id, subnet) - def spawn_instance(self, name, key_name=None, hostname=None, image_id=None, security_group=None, pubkeys=[], nics=None, metadata=None): - flavor_name = self.config.nova_default_flavor + def spawn_instance(self, name, key_name=None, availability_zone=None, hostname=None, image_id=None, security_group=None, pubkeys=[], nics=None, metadata=None, userdata=None, flavor_name=None): + if not flavor_name: + flavor_name = self.config.nova_default_flavor + flavor = self.shell.nova.flavors.find(name=flavor_name) - #if not image: - # image = self.config.nova_default_imave + if not security_group: security_group = self.config.nova_default_security_group files = {} - if pubkeys: - files['/root/.ssh/authorized_keys'] = "\n".join(pubkeys) - + #if pubkeys: + # files["/root/.ssh/authorized_keys"] = "\n".join(pubkeys).encode('base64') hints = {} - availability_zone = None + + # determine availability zone and compute host + availability_zone_filter = None + if not availability_zone: + availability_zone_filter = 'nova' + else: + availability_zone_filter = availability_zone if hostname: - availability_zone = 'nova:%s' % hostname + availability_zone_filter += ':%s' % hostname.split('.')[0] + server = self.shell.nova.servers.create( name=name, key_name = key_name, flavor=flavor.id, image=image_id, security_group = security_group, - files=files, + #files = files, scheduler_hints=hints, - availability_zone=availability_zone, + availability_zone=availability_zone_filter, nics=nics, - meta=metadata) + networks=nics, + meta=metadata, + userdata=userdata) return server def destroy_instance(self, id):