def take_action(self, parsed_args):
column_headers = ('id',
'uuid',
'username',
'user_id',
'user_uuid',
'key',
'is_leader',
'provider',
'provider_id',
'provider_uuid',
'usage',
'quota_cpu',
'quota_memory',
'quota_storage',
'quota_floating_ip_count',
'quota_instance_count',
'quota_port_count',
'quota_snapshot_count',
'quota_storage_count')
api = AtmosphereAPI(self.app_args.auth_token, base_url=self.app_args.base_url, timeout=self.app_args.api_server_timeout, verify=self.app_args.verify_cert)
data = api.get_identity(parsed_args.id)
identity = ()
if data.ok:
message = data.message
identity = (
message['id'],
message['uuid'],
message['user']['username'],
message['user']['id'],
message['user']['uuid'],
message['key'],
message['is_leader'],
message['provider']['name'],
message['provider']['id'],
message['provider']['uuid'],
message['usage'],
message['quota']['cpu'],
message['quota']['memory'],
message['quota']['storage'],
message['quota']['floating_ip_count'],
message['quota']['instance_count'],
message['quota']['port_count'],
message['quota']['snapshot_count'],
message['quota']['storage_count']
)
return (column_headers, identity)
def take_action(self, parsed_args):
api = AtmosphereAPI(self.app_args.auth_token,
base_url=self.app_args.base_url,
timeout=self.app_args.api_server_timeout,
verify=self.app_args.verify_cert)
payload = {
"name": parsed_args.name,
"identity": parsed_args.identity,
"size_alias": parsed_args.size_alias,
"project": parsed_args.project,
"allocation_source_id": parsed_args.allocation_source_id,
"deploy": parsed_args.deploy,
"scripts": [],
"extra": {}
}
if parsed_args.image:
# get the provider from the identity
payload['source_alias'] = None
id_data = api.get_identity(parsed_args.identity)
if id_data.ok:
provider_id = id_data.message['provider']['id']
self.log.debug(
'Found provider id {} for identity uuid {}'.format(
provider_id, parsed_args.identity))
# get the provider machine from the latest image version
v_data = api.get_image_versions(parsed_args.image)
if v_data.ok:
# get latest image version
latest_version = v_data.message['results'][-1]
machines = latest_version['machines']
machine_id = next((m['uuid'] for m in machines
if m['provider']['id'] == provider_id),
None)
self.log.debug(
'Found machine id {} for latest image version {} and provider id {}'
.format(machine_id, latest_version['name'],
provider_id))
payload['source_alias'] = machine_id
else:
self.app.stdout.write(
'Error, instance not created. Could not lookup latest image version from image uuid supplied: {}'
.format(parsed_args.image))
else:
self.app.stdout.write(
'Error, instance not created. Could not lookup provider from identity uuid supplied: {}'
.format(parsed_args.identity))
else:
payload['source_alias'] = parsed_args.source_alias
self.log.debug('INPUT: {}'.format(json.dumps(payload)))
instance = ()
column_headers = ('id', 'uuid', 'name', 'username',
'allocation_source', 'image_id', 'image_version',
'launched', 'image_size', 'provider')
if payload['source_alias']:
data = api.create_instance(json.dumps(payload))
if data.ok:
message = data.message
launched = ts_to_isodate(message['start_date'])
instance = (message['id'], message['uuid'], message['name'],
message['user']['username'],
message['allocation_source']['name'],
message['image']['id'], message['version']['name'],
launched, message['size']['name'],
message['provider']['name'])
else:
self.app.stdout.write(
'Error, instance not created! Make sure to supply name, identity, project, size, source (or image), and allocation_source.'
)
return (column_headers, instance)
def test_getting_identity_when_response_is_ok(self):
api = AtmosphereAPI('token', base_url=self.mock_users_base_url)
response = api.get_identity("32e3354c-03cc-40e1-8c33-02bc7f6be299")
assert response.ok
assert response.message['id'] == 1403 and response.message['quota'][
'cpu'] == 132
def test_getting_identity_when_response_is_not_ok(self):
api = AtmosphereAPI('token', base_url=self.mock_users_bad_base_url)
response = api.get_identity("32e3354c-03cc-40e1-8c33-02bc7f6be299")
assert not response.ok