1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
import errno
import os
import re
import subprocess
import yaml
class NoSuchEnvironment(Exception):
"""Raised when a specified environment does not exist."""
def get_selected_environment(selected, allow_jenv=True):
if selected is None:
selected = default_env()
if allow_jenv:
jenv_config = get_jenv_config(get_juju_home(), selected)
if jenv_config is not None:
return jenv_config, selected
environments = get_environments()
env = environments.get(selected)
if env is None:
raise NoSuchEnvironment(
'Environment "{}" does not exist.'.format(selected))
return get_environments()[selected], selected
def call_with_nova_env(args, environment):
new_environ = dict(os.environ)
new_environ.update(translate_to_env(environment))
subprocess.call(args, env=new_environ)
def get_stateserver_ips(environment, name):
new_environ = dict(os.environ)
new_environ.update(translate_to_env(environment))
listing = subprocess.check_output(['nova', 'list'], env=new_environ)
match = re.search('juju-%s-machine-0[^=]*=([^|]*)' % name, listing)
ips = match.group(1).strip().split(', ')
return ips
def get_juju_home():
home = os.environ.get('JUJU_HOME')
if home is None:
home = os.path.join(os.environ.get('HOME'), '.juju')
return home
def get_environments_path(juju_home):
return os.path.join(juju_home, 'environments.yaml')
def get_jenv_path(juju_home, name):
return os.path.join(juju_home, 'environments', '%s.jenv' % name)
def get_jenv_config(home, environment):
single_name = get_jenv_path(home, environment)
try:
with open(single_name) as env:
return yaml.safe_load(env)['bootstrap-config']
except IOError as e:
if e.errno != errno.ENOENT:
raise
def get_environments():
"""Return the environments for juju."""
home = get_juju_home()
with open(get_environments_path(home)) as env:
return yaml.safe_load(env)['environments']
def default_env():
"""Determine Juju's default environment."""
output = subprocess.check_output(['juju', 'switch'])
match = re.search('\"(.*)\"', output)
if match is None:
return output.rstrip('\n')
return match.group(1)
def translate_to_env(current_env):
"""Translate openstack settings to environment variables."""
if current_env['type'] not in ('openstack', 'rackspace'):
raise Exception('Not an openstack environment. (type: %s)' %
current_env['type'])
# Region doesn't follow the mapping for other vars.
new_environ = {'OS_REGION_NAME': current_env['region']}
for key in ['username', 'password', 'tenant-name', 'auth-url']:
new_environ['OS_' + key.upper().replace('-', '_')] = current_env[key]
return new_environ
def get_euca_env(current_env):
"""Translate openstack settings to environment variables."""
# Region doesn't follow the mapping for other vars.
new_environ = {
'EC2_URL': 'https://%s.ec2.amazonaws.com' % current_env['region']}
for key in ['access-key', 'secret-key']:
env_key = key.upper().replace('-', '_')
new_environ['EC2_' + env_key] = current_env[key]
new_environ['AWS_' + env_key] = current_env[key]
return new_environ
def get_awscli_env(current_env):
"""Translate openstack settings to environment variables."""
# Region doesn't follow the mapping for other vars.
new_environ = {
'AWS_ACCESS_KEY_ID': 'access-key',
'AWS_SECRET_ACCESS_KEY': 'secret-key',
'AWS_DEFAULT_REGION': 'region'
}
for key, value in new_environ.items():
if value not in current_env:
del new_environ[key]
new_environ[key] = current_env[value]
return new_environ
def describe_substrate(config):
if config['type'] == 'local':
return {
'kvm': 'KVM (local)',
'lxc': 'LXC (local)'
}[config.get('container', 'lxc')]
elif config['type'] == 'openstack':
if config['auth-url'] == (
'https://keystone.canonistack.canonical.com:443/v2.0/'):
return 'Canonistack'
else:
return 'Openstack'
try:
return {
'ec2': 'AWS',
'rackspace': 'Rackspace',
'joyent': 'Joyent',
'azure': 'Azure',
'maas': 'MAAS',
}[config['type']]
except KeyError:
return config['type']
|