blob: df340f637b0d0bbb037d6c8c63c01b32ef5947ed [file] [log] [blame]
Tony Mackf8a1a612014-05-06 23:42:40 -04001import urlparse
Siobhan Tully30fd4292013-05-10 08:59:56 -04002try:
3 from keystoneclient.v2_0 import client as keystone_client
Sapan Bhatia14356b72014-11-05 10:32:41 -05004 #from glance import client as glance_client
Tony Mackf8a1a612014-05-06 23:42:40 -04005 import glanceclient
Siobhan Tully30fd4292013-05-10 08:59:56 -04006 from novaclient.v1_1 import client as nova_client
Sapan Bhatia14356b72014-11-05 10:32:41 -05007 from neutronclient.v2_0 import client as quantum_client
Siobhan Tully30fd4292013-05-10 08:59:56 -04008 has_openstack = True
9except:
10 has_openstack = False
11
12from planetstack.config import Config
13
14def require_enabled(callable):
15 def wrapper(*args, **kwds):
16 if has_openstack:
17 return callable(*args, **kwds)
18 else:
19 return None
20 return wrapper
21
22def parse_novarc(filename):
23 opts = {}
24 f = open(filename, 'r')
25 for line in f:
26 try:
27 line = line.replace('export', '').strip()
28 parts = line.split('=')
29 if len(parts) > 1:
30 value = parts[1].replace("\'", "")
31 value = value.replace('\"', '')
32 opts[parts[0]] = value
33 except:
34 pass
35 f.close()
36 return opts
37
38class Client:
Sapan Bhatiaa554a012014-12-22 01:38:33 -050039 def __init__(self, username=None, password=None, tenant=None, url=None, token=None, endpoint=None, controller=None, admin=True, *args, **kwds):
Tony Mack94466922014-06-19 20:09:40 -040040
Siobhan Tully30fd4292013-05-10 08:59:56 -040041 self.has_openstack = has_openstack
Tony Mack4cc17882014-12-26 13:57:08 -050042 self.url = controller.auth_url
Tony Mack49e839c2014-04-07 19:49:01 -040043 if admin:
Tony Mack4cc17882014-12-26 13:57:08 -050044 self.username = controller.admin_user
45 self.password = controller.admin_password
46 self.tenant = controller.admin_tenant
Tony Mack49e839c2014-04-07 19:49:01 -040047 else:
48 self.username = None
49 self.password = None
50 self.tenant = None
Siobhan Tully30fd4292013-05-10 08:59:56 -040051
52 if username:
53 self.username = username
54 if password:
55 self.password = password
56 if tenant:
57 self.tenant = tenant
58 if url:
59 self.url = url
Tony Macke4be32f2014-03-11 20:45:25 -040060 if token:
61 self.token = token
62 if endpoint:
63 self.endpoint = endpoint
Siobhan Tully30fd4292013-05-10 08:59:56 -040064
Tony Mack976d7742014-03-18 22:00:52 -040065 #if '@' in self.username:
66 # self.username = self.username[:self.username.index('@')]
Siobhan Tully30fd4292013-05-10 08:59:56 -040067
68class KeystoneClient(Client):
69 def __init__(self, *args, **kwds):
70 Client.__init__(self, *args, **kwds)
71 if has_openstack:
72 self.client = keystone_client.Client(username=self.username,
73 password=self.password,
74 tenant_name=self.tenant,
Tony Mackdd240952014-06-03 23:02:00 -040075 auth_url=self.url
Tony Macke4be32f2014-03-11 20:45:25 -040076 )
Siobhan Tully30fd4292013-05-10 08:59:56 -040077
78 @require_enabled
79 def connect(self, *args, **kwds):
80 self.__init__(*args, **kwds)
81
82 @require_enabled
83 def __getattr__(self, name):
84 return getattr(self.client, name)
85
86
87class GlanceClient(Client):
88 def __init__(self, *args, **kwds):
89 Client.__init__(self, *args, **kwds)
90 if has_openstack:
Sapan Bhatia14356b72014-11-05 10:32:41 -050091 self.client = glanceclient.get_client(host='0.0.0.0',
Siobhan Tully30fd4292013-05-10 08:59:56 -040092 username=self.username,
93 password=self.password,
94 tenant=self.tenant,
95 auth_url=self.url)
96 @require_enabled
97 def __getattr__(self, name):
98 return getattr(self.client, name)
99
Tony Mackf8a1a612014-05-06 23:42:40 -0400100class GlanceClientNew(Client):
101 def __init__(self, version, endpoint, token, *args, **kwds):
102 Client.__init__(self, *args, **kwds)
103 if has_openstack:
104 self.client = glanceclient.Client(version, endpoint=endpoint, token=token)
105
106 @require_enabled
107 def __getattr__(self, name):
108 return getattr(self.client, name)
109
Siobhan Tully30fd4292013-05-10 08:59:56 -0400110class NovaClient(Client):
111 def __init__(self, *args, **kwds):
112 Client.__init__(self, *args, **kwds)
113 if has_openstack:
114 self.client = nova_client.Client(username=self.username,
115 api_key=self.password,
116 project_id=self.tenant,
117 auth_url=self.url,
118 region_name='',
119 extensions=[],
120 service_type='compute',
121 service_name='',
122 )
123
124 @require_enabled
125 def connect(self, *args, **kwds):
126 self.__init__(*args, **kwds)
127
128 @require_enabled
129 def __getattr__(self, name):
130 return getattr(self.client, name)
131
Tony Mackb0d97422013-06-10 09:57:45 -0400132class NovaDB(Client):
133 def __init__(self, *args, **kwds):
134 Client.__init__(self, *args, **kwds)
135 if has_openstack:
136 self.ctx = get_admin_context()
Tony Mack7b0dad02013-06-10 13:42:21 -0400137 nova_db_api.FLAGS(default_config_files=['/etc/nova/nova.conf'])
Tony Mackb0d97422013-06-10 09:57:45 -0400138 self.client = nova_db_api
139
140
141 @require_enabled
142 def connect(self, *args, **kwds):
143 self.__init__(*args, **kwds)
144
145 @require_enabled
146 def __getattr__(self, name):
147 return getattr(self.client, name)
148
Siobhan Tully30fd4292013-05-10 08:59:56 -0400149class QuantumClient(Client):
150 def __init__(self, *args, **kwds):
151 Client.__init__(self, *args, **kwds)
152 if has_openstack:
153 self.client = quantum_client.Client(username=self.username,
154 password=self.password,
155 tenant_name=self.tenant,
156 auth_url=self.url)
157 @require_enabled
158 def connect(self, *args, **kwds):
159 self.__init__(*args, **kwds)
160
161 @require_enabled
162 def __getattr__(self, name):
163 return getattr(self.client, name)
164
165class OpenStackClient:
166 """
167 A simple native shell to the openstack backend services.
168 This class can receive all nova calls to the underlying testbed
169 """
170
171 def __init__ ( self, *args, **kwds) :
172 # instantiate managers
173 self.keystone = KeystoneClient(*args, **kwds)
Tony Mackf8a1a612014-05-06 23:42:40 -0400174 url_parsed = urlparse.urlparse(self.keystone.url)
175 hostname = url_parsed.netloc.split(':')[0]
176 token = self.keystone.client.tokens.authenticate(username=self.keystone.username, password=self.keystone.password, tenant_name=self.keystone.tenant)
Sapan Bhatia14356b72014-11-05 10:32:41 -0500177 #self.glance = GlanceClient(*args, **kwds)
Tony Mackf8a1a612014-05-06 23:42:40 -0400178
Tony Mack94466922014-06-19 20:09:40 -0400179 self.glanceclient = GlanceClientNew('1', endpoint='http://%s:9292' % hostname, token=token.id, **kwds)
Siobhan Tully30fd4292013-05-10 08:59:56 -0400180 self.nova = NovaClient(*args, **kwds)
Sapan Bhatia14356b72014-11-05 10:32:41 -0500181 # self.nova_db = NovaDB(*args, **kwds)
Siobhan Tully30fd4292013-05-10 08:59:56 -0400182 self.quantum = QuantumClient(*args, **kwds)
Tony Mackf8a1a612014-05-06 23:42:40 -0400183
Siobhan Tully30fd4292013-05-10 08:59:56 -0400184
185 @require_enabled
186 def connect(self, *args, **kwds):
187 self.__init__(*args, **kwds)
188
189 @require_enabled
190 def authenticate(self):
191 return self.keystone.authenticate()