1 from plstackapi.planetstack.config import Config
2 from plstackapi.openstack.client import OpenStackClient
6 def __init__(self, config = None, client=None):
8 self.config = Config(config)
10 self.config = Config()
15 self.shell = OpenStackClient()
17 def create_role(self, name):
18 roles = self.shell.keystone.roles.findall(name=name)
20 role = self.shell.keystone.roles.create(name)
25 def delete_role(self, filter):
26 roles = self.shell.keystone.roles.findall(**filter)
28 self.shell.keystone.roles.delete(role)
31 def create_tenant(self, tenant_name, enabled, description):
32 """Create keystone tenant. Suggested fields: name, description, enabled"""
33 tenants = self.shell.keystone.tenants.findall(name=tenant_name)
35 fields = {'tenant_name': tenant_name, 'enabled': enabled,
36 'description': description}
37 tenant = self.shell.keystone.tenants.create(**fields)
42 def update_tenant(self, id, **kwds):
43 return self.shell.keystone.tenants.update(id, **kwds)
45 def delete_tenant(self, id):
46 tenant = self.shell.keystone.tenants.find(id=id)
47 return self.shell.keystone.tenants.delete(tenant)
49 def create_user(self, name, email, password, enabled):
50 users = self.shell.keystone.users.findall(email=email)
52 fields = {'name': name, 'email': email, 'password': password,
54 user = self.shell.keystone.users.create(**fields)
59 def add_user_role(self, user_id, tenant_id, role_name):
60 user = self.shell.keystone.users.find(id=user_id)
61 tenant = self.shell.keystone.tenants.find(id=tenant_id)
62 role = self.shell.keystone.roles.find(role_name)
63 return tenant.add_user(user, role)
65 def delete_user_role(self, user_id, tenant_id, role_name):
66 user = self.shell.keystone.users.find(id=user_id)
67 tenant = self.shell.keystone.tenants.find(id=tenant_id)
68 role = self.shell.keystone.roles.find(role_name)
69 return tenant.delete_user(user, role)
71 def update_user(self, id, **kwds):
72 return self.shell.keystone.users.update(id, **kwds)
74 def delete_user(self, id):
75 user = self.shell.keystone.users.find(id=id)
76 return self.shell.keystone.users.delete(user)
78 def create_router(self, name, set_gateway=True):
79 routers = self.shell.quantum.list_routers(name=name)['routers']
83 router = self.shell.quantum.create_router({'router': {'name': name}})
85 nets = self.shell.quantum.list_networks()['networks']
87 if net['router:external'] == True:
88 self.shell.quantum.add_gateway_router(router, net)
92 def delete_router(self, name):
93 routers = self.shell.quantum.list_routers(name=name)['routers']
94 for router in routers:
95 self.shell.quantum.delete_router(router['id'])
97 def add_router_interface(self, router_id, subnet_id):
100 for r in self.shell.quantum.list_routers():
101 if r['id'] == router_id:
104 for s in self.shell.quantum.list_subnets():
105 if s['id'] == subnet_id:
109 if router and subnet:
110 self.shell.quantum.router_add_interface(router, subnet)
112 def delete_router_interface(self, router_id, subnet_id):
115 for r in self.shell.quantum.list_routers():
116 if r['id'] == router_id:
119 for s in self.shell.quantum.list_subnets():
120 if s['id'] == subnet_id:
124 if router and subnet:
125 self.shell.quantum.router_remove_interface(router, subnet)
127 def create_network(self, name):
128 nets = self.shell.quantum.list_networks(name=name)['networks']
132 net = self.shell.quantum.create_network({'network': {'name': name}})
135 def delete_network(self, name):
136 nets = self.shell.quantum.list_networks(name=name)['networks']
138 # delete all subnets:
139 #subnets = self.api.client_shell.quantum.list_subnets(network_id=net['network_id'])['subnets']
140 for subnet_id in net['subnets']:
141 self.delete_subnet(subnet_id)
142 self.shell.quantum.delete_network(net['id'])
144 def create_subnet(self, network_name, cidr_ip, ip_version, start, end):
145 nets = self.shell.quantum.list_networks(name=network_name)
147 raise Exception, "No such network: %s" % network_name
150 subnets = self.shell.quantum.list_subnets(name=self.name)
151 allocation_pools = [{'start': start, 'end': end}]
152 subnet = self.shell.quantum.create_subnet(network_id=net['id'],
153 ip_version=ip_version,
155 dns_nameservers=['8.8.8.8', '8.8.8.4'],
156 allocation_pools=allocation_pools)
158 # TODO: Add route to external network
159 # e.g. # route add -net 10.0.3.0/24 dev br-ex gw 10.100.0.5
162 def delete_subnet(self, id):
163 return self.shell.quantum.delete_subnet(id=id)
166 def create_keypair(self, name, key):
167 keys = self.shell.nova.keypairs.findall(name=name)
171 key = self.shell.nova.keypairs.create(name=name, public_key=key)
174 def delete_keypair(self, name):
175 keys = self.shell.nova.keypairs.findall(name=name)
177 self.shell.nova.keypairs.delete(key)
179 def spawn_instance(self, name, key_name=None, hostname=None, flavor=None, image=None, security_group=None, pubkeys=[]):
181 flavor = self.config.nova_default_flavor
183 image = self.config.nova_default_imave
184 if not security_group:
185 security_group = self.config.nova_default_security_group
187 authorized_keys = "\n".join(pubkeys)
188 files = {'/root/.ssh/authorized_keys': authorized_keys}
190 flavor_id = self.shell.nova.flavors.find(name=flavor)
191 images = self.shell.glance.get_images(name=image)
193 raise Exception, "Image not found: " + image
194 image_id = images[0]['id']
197 hints['force_hosts']= hostname
198 server = self.shell.nova.servers.create(
203 security_group = security_group,
205 scheduler_hints=hints)
208 def destroy_instance(self, name, id=None):
209 args = {'name': name}
212 servers = self.shell.nova.servers.findall(**args)
213 for server in servers:
214 if name == server.name:
215 if not id or id == server.id:
216 self.shell.nova.servers.delete(server)