Tony Mack | 01942f5 | 2013-03-26 14:26:40 -0400 | [diff] [blame] | 1 | from plstackapi.planetstack.config import Config |
| 2 | from plstackapi.openstack.shell import OpenStackShell |
| 3 | |
Tony Mack | a033ced | 2013-03-29 17:57:06 -0400 | [diff] [blame] | 4 | class OpenStackDriver: |
Tony Mack | 01942f5 | 2013-03-26 14:26:40 -0400 | [diff] [blame] | 5 | |
| 6 | def __init__(self, config = None): |
| 7 | if config: |
| 8 | self.config = Config(config) |
| 9 | else: |
| 10 | self.config = Config() |
| 11 | self.shell = OpenStackShell() |
| 12 | |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 13 | def create_tenant(self, tenant_name, enabled, description): |
Tony Mack | 5ee2190 | 2013-03-29 20:36:42 -0400 | [diff] [blame] | 14 | """Create keystone tenant. Suggested fields: name, description, enabled""" |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 15 | tenants = self.shell.keystone.tenants.findall(name=tenant_name) |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 16 | if not tenants: |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 17 | fields = {'tenant_name': tenant_name, 'enabled': enabled, |
| 18 | 'description', description} |
| 19 | tenant = self.shell.keystone.tenants.create(**fields) |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 20 | else: |
| 21 | tenant = tenants[0] |
| 22 | return tenant |
| 23 | |
| 24 | def update_tenant(self, id, **kwds): |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 25 | return self.shell.keystone.tenants.update(id, **kwds) |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 26 | |
| 27 | def delete_tenant(self, id): |
| 28 | tenant = self.shell.keystone.tenants.find(id=id) |
| 29 | return self.shell.keystone.tenants.delete(tenant) |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 30 | |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 31 | def create_user(self, name, email, password, enabled): |
| 32 | users = self.shell.keystone.users.findall(email=email) |
| 33 | if not users: |
| 34 | fields = {'name': name, 'email': email, 'password': password, |
| 35 | 'enabled': enabled} |
| 36 | user = self.shell.keystone.create(**fields) |
| 37 | else: |
| 38 | user = users[0] |
| 39 | return user |
| 40 | |
| 41 | def add_user_role(self, user_id, tenant_id, role_name): |
| 42 | user = self.shell.keystone.users.find(id=user_id) |
| 43 | tenant = self.shell.keystone.tenants.find(id=tenant_id) |
| 44 | role = self.shell.keystone.roles.find(role_name) |
| 45 | return tenant.add_user(user, role) |
| 46 | |
| 47 | def delete_user_role(self, user_id, tenant_id, role_name): |
| 48 | user = self.shell.keystone.users.find(id=user_id) |
| 49 | tenant = self.shell.keystone.tenants.find(id=tenant_id) |
| 50 | role = self.shell.keystone.roles.find(role_name) |
| 51 | return tenant.delete_user(user, role) |
| 52 | |
| 53 | def update_user(self, id, **kwds): |
| 54 | return self.shell.keystone.users.update(id, **kwds) |
| 55 | |
| 56 | def delete_user(self, id): |
| 57 | user = self.shell.keystone.users.find(id=id) |
| 58 | return self.shell.keystone.users.delete(user) |
| 59 | |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame^] | 60 | def create_router(self, name, set_gateway=True): |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 61 | router = self.shell.quantum.create_router(name=name) |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame^] | 62 | if set_gateway: |
| 63 | nets = self.shell.quantum.list_networks() |
| 64 | for net in nets: |
| 65 | if net['router:external'] == True: |
| 66 | self.shell.quantum.add_gateway_router(router, net) |
| 67 | |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 68 | return router |
| 69 | |
| 70 | def delete_router(self, name): |
| 71 | return self.shell.quantum.delete_router(name=name) |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame^] | 72 | |
| 73 | def add_router_interface(self, router_id, subnet_id): |
| 74 | router = None |
| 75 | subnet = None |
| 76 | for r in self.shell.quantum.list_routers(): |
| 77 | if r['id'] == router_id: |
| 78 | router = r |
| 79 | break |
| 80 | for s in self.shell.quantum.list_subnets(): |
| 81 | if s['id'] == subnet_id: |
| 82 | subnet = s |
| 83 | break |
| 84 | |
| 85 | if router and subnet: |
| 86 | self.shell.quantum.router_add_interface(router, subnet) |
| 87 | |
| 88 | def delete_router_interface(self, router_id, subnet_id): |
| 89 | router = None |
| 90 | subnet = None |
| 91 | for r in self.shell.quantum.list_routers(): |
| 92 | if r['id'] == router_id: |
| 93 | router = r |
| 94 | break |
| 95 | for s in self.shell.quantum.list_subnets(): |
| 96 | if s['id'] == subnet_id: |
| 97 | subnet = s |
| 98 | break |
| 99 | |
| 100 | if router and subnet: |
| 101 | self.shell.quantum.router_remove_interface(router, subnet) |
| 102 | |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 103 | def create_network(self, name): |
| 104 | return self.shell.quantum.create_network(name=name, admin_state_up=True) |
| 105 | |
| 106 | def delete_network(self, name): |
| 107 | nets = self.shell.quantum.list_networks(name=name) |
| 108 | for net in nets: |
| 109 | # delete all subnets: |
| 110 | #subnets = self.api.client_shell.quantum.list_subnets(network_id=net['network_id'])['subnets'] |
| 111 | for subnet_id in net['subnets']: |
| 112 | self.delete_subnet(subnet_id) |
| 113 | self.shell.quantum.delete_network(net['id']) |
| 114 | |
| 115 | def create_subnet(self, network_name, cidr_ip, ip_version, start, end): |
| 116 | nets = self.shell.quantum.list_networks(name=network_name) |
| 117 | if not nets: |
| 118 | raise Exception, "No such network: %s" % network_name |
| 119 | nets = nets[0] |
| 120 | |
| 121 | subnets = self.shell.quantum.list_subnets(name=self.name) |
| 122 | allocation_pools = [{'start': start, 'end': end}] |
| 123 | subnet = self.shell.quantum.create_subnet(network_id=net['id'], |
| 124 | ip_version=ip_version, |
| 125 | cidr=cidr_ip, |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame^] | 126 | dns_nameservers=['8.8.8.8', '8.8.8.4'], |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 127 | allocation_pools=allocation_pools) |
| 128 | |
| 129 | # TODO: Add route to external network |
| 130 | # e.g. # route add -net 10.0.3.0/24 dev br-ex gw 10.100.0.5 |
| 131 | return subnet |
| 132 | |
| 133 | def delete_subnet(self, id): |
| 134 | return self.client.quantum.delete_subnet(id=id) |
| 135 | |
| 136 | |
| 137 | def create_keypair(self, name, key): |
| 138 | keys = self.client.nova.keypairs.findall(name=name) |
| 139 | if keys: |
| 140 | raise Exception, "Key name already exists: %s" % name |
| 141 | return self.client.nova.keypairs.create(name=name, public_key=key) |
| 142 | |
| 143 | def delete_keypair(self, name): |
| 144 | keys = self.client.nova.keypairs.findall(name=name) |
| 145 | for key in keys: |
| 146 | self.client.nova.keypairs.delete(key) |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 147 | |
| 148 | def spawn_instance(self, name, key_name=None, hostname=None, flavor=None, image=None, security_group=None, pubkeys=[]): |
Tony Mack | 01942f5 | 2013-03-26 14:26:40 -0400 | [diff] [blame] | 149 | if not flavor: |
| 150 | flavor = self.config.nova_default_flavor |
| 151 | if not image: |
| 152 | image = self.config.nova_default_imave |
| 153 | if not security_group: |
| 154 | security_group = self.config.nova_default_security_group |
| 155 | |
| 156 | authorized_keys = "\n".join(pubkeys) |
| 157 | files = {'/root/.ssh/authorized_keys': authorized_keys} |
| 158 | |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 159 | flavor_id = self.shell.nova.flavors.find(name=flavor) |
| 160 | images = self.shell.glance.get_images(name=image) |
| 161 | if not images: |
| 162 | raise Exception, "Image not found: " + image |
| 163 | image_id = images[0]['id'] |
| 164 | hints = {} |
| 165 | if hostname: |
| 166 | hints['force_hosts']= hostname |
| 167 | server = self.shell.nova.servers.create( |
| 168 | name=name, |
| 169 | key_name = key_name, |
| 170 | flavor=flavor_id, |
| 171 | image=image_id, |
| 172 | security_group = security_group, |
| 173 | files=files, |
| 174 | scheduler_hints=hints) |
| 175 | return server |
Tony Mack | 01942f5 | 2013-03-26 14:26:40 -0400 | [diff] [blame] | 176 | |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 177 | def destroy_instance(self, name, id=None): |
| 178 | args = {'name': name} |
| 179 | if id: |
| 180 | args['id'] = id |
| 181 | servers = self.shell.nova.servers.findall(**args) |
Tony Mack | 01942f5 | 2013-03-26 14:26:40 -0400 | [diff] [blame] | 182 | for server in servers: |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 183 | if name == server.name: |
Tony Mack | 07ece3b | 2013-03-29 20:51:07 -0400 | [diff] [blame] | 184 | if not id or id == server.id: |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 185 | self.shell.nova.servers.delete(server) |