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 router = self.shell.quantum.create_router(name=name)
81 nets = self.shell.quantum.list_networks()
83 if net['router:external'] == True:
84 self.shell.quantum.add_gateway_router(router, net)
88 def delete_router(self, name):
89 return self.shell.quantum.delete_router(name=name)
91 def add_router_interface(self, router_id, subnet_id):
94 for r in self.shell.quantum.list_routers():
95 if r['id'] == router_id:
98 for s in self.shell.quantum.list_subnets():
99 if s['id'] == subnet_id:
103 if router and subnet:
104 self.shell.quantum.router_add_interface(router, subnet)
106 def delete_router_interface(self, router_id, subnet_id):
109 for r in self.shell.quantum.list_routers():
110 if r['id'] == router_id:
113 for s in self.shell.quantum.list_subnets():
114 if s['id'] == subnet_id:
118 if router and subnet:
119 self.shell.quantum.router_remove_interface(router, subnet)
121 def create_network(self, name):
122 return self.shell.quantum.create_network(name=name, admin_state_up=True)
124 def delete_network(self, name):
125 nets = self.shell.quantum.list_networks(name=name)
127 # delete all subnets:
128 #subnets = self.api.client_shell.quantum.list_subnets(network_id=net['network_id'])['subnets']
129 for subnet_id in net['subnets']:
130 self.delete_subnet(subnet_id)
131 self.shell.quantum.delete_network(net['id'])
133 def create_subnet(self, network_name, cidr_ip, ip_version, start, end):
134 nets = self.shell.quantum.list_networks(name=network_name)
136 raise Exception, "No such network: %s" % network_name
139 subnets = self.shell.quantum.list_subnets(name=self.name)
140 allocation_pools = [{'start': start, 'end': end}]
141 subnet = self.shell.quantum.create_subnet(network_id=net['id'],
142 ip_version=ip_version,
144 dns_nameservers=['8.8.8.8', '8.8.8.4'],
145 allocation_pools=allocation_pools)
147 # TODO: Add route to external network
148 # e.g. # route add -net 10.0.3.0/24 dev br-ex gw 10.100.0.5
151 def delete_subnet(self, id):
152 return self.shell.quantum.delete_subnet(id=id)
155 def create_keypair(self, name, key):
156 keys = self.shell.nova.keypairs.findall(name=name)
160 key = self.shell.nova.keypairs.create(name=name, public_key=key)
163 def delete_keypair(self, name):
164 keys = self.shell.nova.keypairs.findall(name=name)
166 self.shell.nova.keypairs.delete(key)
168 def spawn_instance(self, name, key_name=None, hostname=None, flavor=None, image=None, security_group=None, pubkeys=[]):
170 flavor = self.config.nova_default_flavor
172 image = self.config.nova_default_imave
173 if not security_group:
174 security_group = self.config.nova_default_security_group
176 authorized_keys = "\n".join(pubkeys)
177 files = {'/root/.ssh/authorized_keys': authorized_keys}
179 flavor_id = self.shell.nova.flavors.find(name=flavor)
180 images = self.shell.glance.get_images(name=image)
182 raise Exception, "Image not found: " + image
183 image_id = images[0]['id']
186 hints['force_hosts']= hostname
187 server = self.shell.nova.servers.create(
192 security_group = security_group,
194 scheduler_hints=hints)
197 def destroy_instance(self, name, id=None):
198 args = {'name': name}
201 servers = self.shell.nova.servers.findall(**args)
202 for server in servers:
203 if name == server.name:
204 if not id or id == server.id:
205 self.shell.nova.servers.delete(server)