X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=planetstack%2Fopenstack%2Fdriver.py;h=093ab5a9d59ba41b8576ce110aead4ad6f656052;hb=ffae8c3132d6a1ae7ad7213eafdc3fa6621fac1d;hp=a6addba2b24392a06d189e5ea2b9d599deb090c7;hpb=0d5ea5c7e775695379b698e6b7357ae32d5cd182;p=plstackapi.git diff --git a/planetstack/openstack/driver.py b/planetstack/openstack/driver.py index a6addba..093ab5a 100644 --- a/planetstack/openstack/driver.py +++ b/planetstack/openstack/driver.py @@ -1,29 +1,63 @@ import commands +import hashlib from planetstack.config import Config -from openstack.client import OpenStackClient +from core.models import Deployment + +try: + from openstack.client import OpenStackClient + has_openstack = True +except: + has_openstack = False + +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 + + 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): + admin_driver = self.admin_driver(tenant=tenant, deployment=deployment) + if caller: + auth = {'username': caller.email, + 'password': hashlib.md5(caller.password).hexdigest()[:6], + 'tenant': tenant} + client = OpenStackClient(deployment=admin_driver.deployment, **auth) else: - self.shell = OpenStackClient() + 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, 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): @@ -83,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) @@ -103,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) @@ -160,12 +215,12 @@ class OpenStackDriver: if router and subnet: self.shell.quantum.remove_interface_router(router_id, {'subnet_id': subnet_id}) - def create_network(self, name): + def create_network(self, name, shared=False): nets = self.shell.quantum.list_networks(name=name)['networks'] if nets: net = nets[0] else: - net = self.shell.quantum.create_network({'network': {'name': name}})['network'] + net = self.shell.quantum.create_network({'network': {'name': name, 'shared': shared}})['network'] return net def delete_network(self, id): @@ -212,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): @@ -358,38 +417,55 @@ class OpenStackDriver: return (subnet_id, subnet) - def spawn_instance(self, name, key_name=None, hostname=None, image_id=None, security_group=None, pubkeys=[], networks=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 + 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, - networks=networks) + availability_zone=availability_zone_filter, + nics=nics, + networks=nics, + meta=metadata, + userdata=userdata) return server - + def destroy_instance(self, id): - servers = self.shell.nova.servers.findall(id=id) + if (self.shell.nova.tenant=="admin"): + # findall() is implemented as a list() followed by a python search of the + # list. Since findall() doesn't accept "all_tenants", we do this using + # list() ourselves. This allows us to delete an instance as admin. + servers = self.shell.nova.servers.list(search_opts={"all_tenants": True}) + else: + servers = self.shell.nova.servers.list() for server in servers: - self.shell.nova.servers.delete(server) + if server.id == id: + result=self.shell.nova.servers.delete(server) def update_instance_metadata(self, id, metadata): servers = self.shell.nova.servers.findall(id=id)