Tony Mack | 01942f5 | 2013-03-26 14:26:40 -0400 | [diff] [blame] | 1 | from plstackapi.planetstack.config import Config |
Tony Mack | ad8b093 | 2013-04-07 19:32:07 -0400 | [diff] [blame] | 2 | from plstackapi.openstack.client import OpenStackClient |
Tony Mack | 01942f5 | 2013-03-26 14:26:40 -0400 | [diff] [blame] | 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 | |
Tony Mack | ad8b093 | 2013-04-07 19:32:07 -0400 | [diff] [blame] | 6 | def __init__(self, config = None, client=None): |
Tony Mack | 01942f5 | 2013-03-26 14:26:40 -0400 | [diff] [blame] | 7 | if config: |
| 8 | self.config = Config(config) |
| 9 | else: |
| 10 | self.config = Config() |
Tony Mack | ad8b093 | 2013-04-07 19:32:07 -0400 | [diff] [blame] | 11 | |
| 12 | if client: |
| 13 | self.shell = client |
| 14 | else: |
| 15 | self.shell = OpenStackClient() |
Tony Mack | 01942f5 | 2013-03-26 14:26:40 -0400 | [diff] [blame] | 16 | |
Tony Mack | b5f339f | 2013-04-08 16:29:38 -0400 | [diff] [blame] | 17 | def create_role(self, name): |
| 18 | roles = self.shell.keystone.roles.findall(name=name) |
| 19 | if not roles: |
| 20 | role = self.shell.keystone.roles.create(name) |
| 21 | else: |
| 22 | role = roles[0] |
| 23 | return role |
| 24 | |
Tony Mack | a704053 | 2013-04-08 21:45:43 -0400 | [diff] [blame] | 25 | def delete_role(self, filter): |
| 26 | roles = self.shell.keystone.roles.findall(**filter) |
Tony Mack | b5f339f | 2013-04-08 16:29:38 -0400 | [diff] [blame] | 27 | for role in roles: |
| 28 | self.shell.keystone.roles.delete(role) |
| 29 | return 1 |
| 30 | |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 31 | def create_tenant(self, tenant_name, enabled, description): |
Tony Mack | 5ee2190 | 2013-03-29 20:36:42 -0400 | [diff] [blame] | 32 | """Create keystone tenant. Suggested fields: name, description, enabled""" |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 33 | tenants = self.shell.keystone.tenants.findall(name=tenant_name) |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 34 | if not tenants: |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 35 | fields = {'tenant_name': tenant_name, 'enabled': enabled, |
Tony Mack | 488950a | 2013-04-07 20:35:50 -0400 | [diff] [blame] | 36 | 'description': description} |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 37 | tenant = self.shell.keystone.tenants.create(**fields) |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 38 | else: |
| 39 | tenant = tenants[0] |
| 40 | return tenant |
| 41 | |
| 42 | def update_tenant(self, id, **kwds): |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 43 | return self.shell.keystone.tenants.update(id, **kwds) |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 44 | |
| 45 | def delete_tenant(self, id): |
Tony Mack | fd24d0d | 2013-04-14 00:59:17 -0400 | [diff] [blame] | 46 | tenants = self.shell.keystone.tenants.findall(id=id) |
| 47 | for tenant in tenants: |
| 48 | self.shell.keystone.tenants.delete(tenant) |
| 49 | return 1 |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 50 | |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 51 | def create_user(self, name, email, password, enabled): |
| 52 | users = self.shell.keystone.users.findall(email=email) |
| 53 | if not users: |
| 54 | fields = {'name': name, 'email': email, 'password': password, |
| 55 | 'enabled': enabled} |
Tony Mack | fdcbf83 | 2013-04-09 11:47:11 -0400 | [diff] [blame] | 56 | user = self.shell.keystone.users.create(**fields) |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 57 | else: |
| 58 | user = users[0] |
| 59 | return user |
| 60 | |
Tony Mack | fd24d0d | 2013-04-14 00:59:17 -0400 | [diff] [blame] | 61 | def delete_user(self, id): |
| 62 | users = self.shell.keystone.users.findall(id=id) |
| 63 | for user in users: |
| 64 | self.shell.keystone.users.delete(user) |
| 65 | return 1 |
| 66 | |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 67 | def add_user_role(self, user_id, tenant_id, role_name): |
| 68 | user = self.shell.keystone.users.find(id=user_id) |
| 69 | tenant = self.shell.keystone.tenants.find(id=tenant_id) |
Tony Mack | 54304c6 | 2013-04-28 11:27:07 -0400 | [diff] [blame] | 70 | role = self.shell.keystone.roles.find(name=role_name) |
| 71 | |
| 72 | role_found = False |
| 73 | user_roles = user.list_roles(tenant.id) |
| 74 | for user_role in user_roles: |
| 75 | if user_role.name == role.name: |
| 76 | role_found = True |
| 77 | if not role_found: |
| 78 | tenant.add_user(user, role) |
| 79 | |
| 80 | return 1 |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 81 | |
| 82 | def delete_user_role(self, user_id, tenant_id, role_name): |
| 83 | user = self.shell.keystone.users.find(id=user_id) |
| 84 | tenant = self.shell.keystone.tenants.find(id=tenant_id) |
Tony Mack | 54304c6 | 2013-04-28 11:27:07 -0400 | [diff] [blame] | 85 | role = self.shell.keystone.roles.find(name=role_name) |
| 86 | |
| 87 | role_found = False |
| 88 | user_roles = user.list_roles(tenant.id) |
| 89 | for user_role in user_roles: |
| 90 | if user_role.name == role.name: |
| 91 | role_found = True |
| 92 | if role_found: |
| 93 | tenant.remove_user(user, role) |
| 94 | |
| 95 | return 1 |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 96 | |
| 97 | def update_user(self, id, **kwds): |
| 98 | return self.shell.keystone.users.update(id, **kwds) |
| 99 | |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame] | 100 | def create_router(self, name, set_gateway=True): |
Tony Mack | 28be756 | 2013-04-10 08:45:55 -0400 | [diff] [blame] | 101 | routers = self.shell.quantum.list_routers(name=name)['routers'] |
| 102 | if routers: |
| 103 | router = routers[0] |
| 104 | else: |
Tony Mack | 62bc59a | 2013-04-14 23:27:12 -0400 | [diff] [blame] | 105 | router = self.shell.quantum.create_router({'router': {'name': name}})['router'] |
Tony Mack | c1665b1 | 2013-04-29 16:45:14 -0400 | [diff] [blame^] | 106 | # add router to external network |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame] | 107 | if set_gateway: |
Tony Mack | 257bedc | 2013-04-10 08:58:09 -0400 | [diff] [blame] | 108 | nets = self.shell.quantum.list_networks()['networks'] |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame] | 109 | for net in nets: |
| 110 | if net['router:external'] == True: |
Tony Mack | 9dcc050 | 2013-04-10 09:07:45 -0400 | [diff] [blame] | 111 | self.shell.quantum.add_gateway_router(router['id'], |
| 112 | {'network_id': net['id']}) |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame] | 113 | |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 114 | return router |
| 115 | |
Tony Mack | 62bc59a | 2013-04-14 23:27:12 -0400 | [diff] [blame] | 116 | def delete_router(self, id): |
| 117 | routers = self.shell.quantum.list_routers(id=id)['routers'] |
Tony Mack | 28be756 | 2013-04-10 08:45:55 -0400 | [diff] [blame] | 118 | for router in routers: |
| 119 | self.shell.quantum.delete_router(router['id']) |
Tony Mack | c1665b1 | 2013-04-29 16:45:14 -0400 | [diff] [blame^] | 120 | |
| 121 | # remove router form external network |
| 122 | nets = self.shell.quantum.list_networks()['networks'] |
| 123 | for net in nets: |
| 124 | if net['router:external'] == True: |
| 125 | self.shell.quantum.add_gateway_router(router['id'], |
| 126 | {'network_id': net['id']}) |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame] | 127 | |
| 128 | def add_router_interface(self, router_id, subnet_id): |
Tony Mack | 4895203 | 2013-04-12 11:49:34 -0400 | [diff] [blame] | 129 | router = self.shell.quantum.show_router(router_id)['router'] |
| 130 | subnet = self.shell.quantum.show_subnet(subnet_id)['subnet'] |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame] | 131 | if router and subnet: |
Tony Mack | 4895203 | 2013-04-12 11:49:34 -0400 | [diff] [blame] | 132 | self.shell.quantum.add_interface_router(router_id, {'subnet_id': subnet_id}) |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame] | 133 | |
| 134 | def delete_router_interface(self, router_id, subnet_id): |
Tony Mack | 4895203 | 2013-04-12 11:49:34 -0400 | [diff] [blame] | 135 | router = self.shell.quantum.show_router(router_id) |
| 136 | subnet = self.shell.quantum.show_subnet(subnet_id) |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame] | 137 | if router and subnet: |
Tony Mack | 4895203 | 2013-04-12 11:49:34 -0400 | [diff] [blame] | 138 | self.shell.quantum.remove_interface_router(router_id, {'subnet_id': subnet_id}) |
Tony Mack | 240903f | 2013-04-06 01:31:37 -0400 | [diff] [blame] | 139 | |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 140 | def create_network(self, name): |
Tony Mack | 28be756 | 2013-04-10 08:45:55 -0400 | [diff] [blame] | 141 | nets = self.shell.quantum.list_networks(name=name)['networks'] |
| 142 | if nets: |
Tony Mack | 53c6334 | 2013-04-10 08:11:45 -0400 | [diff] [blame] | 143 | net = nets[0] |
Tony Mack | 82a688d | 2013-04-10 08:12:58 -0400 | [diff] [blame] | 144 | else: |
Tony Mack | 62bc59a | 2013-04-14 23:27:12 -0400 | [diff] [blame] | 145 | net = self.shell.quantum.create_network({'network': {'name': name}})['network'] |
Tony Mack | 53c6334 | 2013-04-10 08:11:45 -0400 | [diff] [blame] | 146 | return net |
| 147 | |
Tony Mack | 62bc59a | 2013-04-14 23:27:12 -0400 | [diff] [blame] | 148 | def delete_network(self, id): |
| 149 | nets = self.shell.quantum.list_networks()['networks'] |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 150 | for net in nets: |
Tony Mack | 62bc59a | 2013-04-14 23:27:12 -0400 | [diff] [blame] | 151 | if net['id'] == id: |
Tony Mack | 94c6ea0 | 2013-04-16 21:41:50 -0400 | [diff] [blame] | 152 | # delete_all ports |
| 153 | self.delete_network_ports(net['id']) |
Tony Mack | 62bc59a | 2013-04-14 23:27:12 -0400 | [diff] [blame] | 154 | # delete all subnets: |
| 155 | for subnet_id in net['subnets']: |
| 156 | self.delete_subnet(subnet_id) |
| 157 | self.shell.quantum.delete_network(net['id']) |
Tony Mack | fd24d0d | 2013-04-14 00:59:17 -0400 | [diff] [blame] | 158 | return 1 |
Tony Mack | 94c6ea0 | 2013-04-16 21:41:50 -0400 | [diff] [blame] | 159 | |
| 160 | def delete_network_ports(self, network_id): |
| 161 | ports = self.shell.quantum.list_ports()['ports'] |
| 162 | for port in ports: |
Tony Mack | c1665b1 | 2013-04-29 16:45:14 -0400 | [diff] [blame^] | 163 | if port['network_id'] == network_id: |
Tony Mack | 94c6ea0 | 2013-04-16 21:41:50 -0400 | [diff] [blame] | 164 | self.shell.quantum.delete_port(port['id']) |
| 165 | return 1 |
| 166 | |
| 167 | def delete_subnet_ports(self, subnet_id): |
| 168 | ports = self.shell.quantum.list_ports()['ports'] |
| 169 | for port in ports: |
| 170 | delete = False |
| 171 | for fixed_ip in port['fixed_ips']: |
| 172 | if fixed_ip['subnet_id'] == subnet_id: |
| 173 | delete=True |
| 174 | break |
| 175 | if delete: |
| 176 | self.shell.quantum.delete_port(port['id']) |
| 177 | return 1 |
| 178 | |
Tony Mack | 4895203 | 2013-04-12 11:49:34 -0400 | [diff] [blame] | 179 | def create_subnet(self, name, network_id, cidr_ip, ip_version, start, end): |
| 180 | #nets = self.shell.quantum.list_networks(name=network_name)['networks'] |
| 181 | #if not nets: |
| 182 | # raise Exception, "No such network: %s" % network_name |
| 183 | #net = nets[0] |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 184 | |
Tony Mack | 4895203 | 2013-04-12 11:49:34 -0400 | [diff] [blame] | 185 | subnet = None |
| 186 | subnets = self.shell.quantum.list_subnets()['subnets'] |
| 187 | for snet in subnets: |
| 188 | if snet['cidr'] == cidr_ip and snet['network_id'] == network_id: |
| 189 | subnet = snet |
| 190 | |
| 191 | if not subnet: |
| 192 | allocation_pools = [{'start': start, 'end': end}] |
| 193 | subnet = {'subnet': {'name': name, |
| 194 | 'network_id': network_id, |
| 195 | 'ip_version': ip_version, |
| 196 | 'cidr': cidr_ip, |
| 197 | 'dns_nameservers': ['8.8.8.8', '8.8.8.4'], |
| 198 | 'allocation_pools': allocation_pools}} |
| 199 | subnet = self.shell.quantum.create_subnet(subnet)['subnet'] |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 200 | |
| 201 | # TODO: Add route to external network |
| 202 | # e.g. # route add -net 10.0.3.0/24 dev br-ex gw 10.100.0.5 |
| 203 | return subnet |
| 204 | |
Tony Mack | 4895203 | 2013-04-12 11:49:34 -0400 | [diff] [blame] | 205 | def update_subnet(self, id, fields): |
| 206 | return self.shell.quantum.update_subnet(id, fields) |
| 207 | |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 208 | def delete_subnet(self, id): |
Tony Mack | fd24d0d | 2013-04-14 00:59:17 -0400 | [diff] [blame] | 209 | #return self.shell.quantum.delete_subnet(id=id) |
| 210 | # inefficient but fault tolerant |
| 211 | subnets = self.shell.quantum.list_subnets()['subnets'] |
| 212 | for subnet in subnets: |
| 213 | if subnet['id'] == id: |
Tony Mack | 94c6ea0 | 2013-04-16 21:41:50 -0400 | [diff] [blame] | 214 | self.delete_subnet_ports(subnet['id']) |
Tony Mack | 31d3413 | 2013-04-15 00:02:06 -0400 | [diff] [blame] | 215 | self.shell.quantum.delete_subnet(id) |
Tony Mack | fd24d0d | 2013-04-14 00:59:17 -0400 | [diff] [blame] | 216 | return |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 217 | |
| 218 | def create_keypair(self, name, key): |
Tony Mack | c2a1c7c | 2013-04-09 13:06:34 -0400 | [diff] [blame] | 219 | keys = self.shell.nova.keypairs.findall(name=name) |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 220 | if keys: |
Tony Mack | 52cb930 | 2013-04-09 12:25:35 -0400 | [diff] [blame] | 221 | key = keys[0] |
| 222 | else: |
Tony Mack | c2a1c7c | 2013-04-09 13:06:34 -0400 | [diff] [blame] | 223 | key = self.shell.nova.keypairs.create(name=name, public_key=key) |
Tony Mack | 52cb930 | 2013-04-09 12:25:35 -0400 | [diff] [blame] | 224 | return key |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 225 | |
Tony Mack | 759b57a | 2013-04-14 21:03:31 -0400 | [diff] [blame] | 226 | def delete_keypair(self, id): |
| 227 | keys = self.shell.nova.keypairs.findall(id=id) |
Tony Mack | 0211f1b | 2013-04-04 23:56:24 -0400 | [diff] [blame] | 228 | for key in keys: |
Tony Mack | c2a1c7c | 2013-04-09 13:06:34 -0400 | [diff] [blame] | 229 | self.shell.nova.keypairs.delete(key) |
Tony Mack | fd24d0d | 2013-04-14 00:59:17 -0400 | [diff] [blame] | 230 | return 1 |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 231 | |
Tony Mack | 221957d | 2013-04-13 19:08:34 -0400 | [diff] [blame] | 232 | def spawn_instance(self, name, key_name=None, hostname=None, flavor_id=None, image_id=None, security_group=None, pubkeys=[]): |
| 233 | #if not flavor_id: |
| 234 | # flavor = self.config.nova_default_flavor |
| 235 | #if not image: |
| 236 | # image = self.config.nova_default_imave |
Tony Mack | 01942f5 | 2013-03-26 14:26:40 -0400 | [diff] [blame] | 237 | if not security_group: |
| 238 | security_group = self.config.nova_default_security_group |
| 239 | |
| 240 | authorized_keys = "\n".join(pubkeys) |
| 241 | files = {'/root/.ssh/authorized_keys': authorized_keys} |
| 242 | |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 243 | hints = {} |
Tony Mack | 79b28ca | 2013-04-15 21:47:45 -0400 | [diff] [blame] | 244 | availability_zone = None |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 245 | if hostname: |
Tony Mack | 79b28ca | 2013-04-15 21:47:45 -0400 | [diff] [blame] | 246 | availability_zone = 'nova:%s' % hostname |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 247 | server = self.shell.nova.servers.create( |
| 248 | name=name, |
| 249 | key_name = key_name, |
| 250 | flavor=flavor_id, |
| 251 | image=image_id, |
| 252 | security_group = security_group, |
| 253 | files=files, |
Tony Mack | 79b28ca | 2013-04-15 21:47:45 -0400 | [diff] [blame] | 254 | scheduler_hints=hints, |
| 255 | availability_zone=availability_zone) |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 256 | return server |
Tony Mack | 01942f5 | 2013-03-26 14:26:40 -0400 | [diff] [blame] | 257 | |
Tony Mack | cdec090 | 2013-04-15 00:38:49 -0400 | [diff] [blame] | 258 | def destroy_instance(self, id): |
| 259 | servers = self.shell.nova.servers.findall(id=id) |
Tony Mack | 01942f5 | 2013-03-26 14:26:40 -0400 | [diff] [blame] | 260 | for server in servers: |
Tony Mack | cdec090 | 2013-04-15 00:38:49 -0400 | [diff] [blame] | 261 | self.shell.nova.servers.delete(server) |