89 lines
2.8 KiB
Python
89 lines
2.8 KiB
Python
#!/usr/bin/env python
|
|
import os
|
|
import yaml
|
|
|
|
|
|
AUTH_URL = os.environ['auth_url']
|
|
ADMIN_PASSWORD = os.environ['admin_password']
|
|
CLOUD_NAME = os.environ['cloud_name']
|
|
HOME_DIR = os.environ['home_dir']
|
|
IDENTITY_API_VERSION = os.environ['identity_api_version']
|
|
PROJECT_NAME = os.environ['project_name']
|
|
PROJECT_DOMAIN_NAME = os.environ['project_domain_name']
|
|
REGION_NAME = os.environ['region_name']
|
|
USER_NAME = os.environ['user_name']
|
|
USER_DOMAIN_NAME = os.environ['user_domain_name']
|
|
|
|
CONFIG_DIR = os.path.join(HOME_DIR, '.config')
|
|
OS_DIR = os.path.join(CONFIG_DIR, 'openstack')
|
|
USER_CLOUDS_YAML = os.path.join(OS_DIR, 'clouds.yaml')
|
|
GLOBAL_OS_DIR = os.path.join('/etc', 'openstack')
|
|
GLOBAL_CLOUDS_YAML = os.path.join(GLOBAL_OS_DIR, 'clouds.yaml')
|
|
|
|
CLOUD = {CLOUD_NAME: {'auth': {'auth_url': AUTH_URL,
|
|
'project_name': PROJECT_NAME,
|
|
'project_domain_name': PROJECT_DOMAIN_NAME,
|
|
'username': USER_NAME,
|
|
'user_domain_name': USER_DOMAIN_NAME,
|
|
'password': ADMIN_PASSWORD},
|
|
'region_name': REGION_NAME,
|
|
'identity_api_version': IDENTITY_API_VERSION}
|
|
}
|
|
|
|
|
|
def _create_clouds_yaml(clouds_yaml):
|
|
with open(clouds_yaml, 'w') as f:
|
|
yaml.dump({'clouds': {}}, f, default_flow_style=False)
|
|
os.chmod(clouds_yaml, 0o600)
|
|
|
|
|
|
def _read_clouds_yaml(clouds_yaml):
|
|
with open(clouds_yaml, 'r') as f:
|
|
clouds = yaml.safe_load(f)
|
|
if 'clouds' not in clouds:
|
|
clouds.update({'clouds': {}})
|
|
|
|
return clouds
|
|
|
|
|
|
def _write_clouds_yaml(clouds_yaml, clouds):
|
|
with open(clouds_yaml, 'w') as f:
|
|
yaml.dump(clouds, f, default_flow_style=False)
|
|
|
|
|
|
try:
|
|
# Get the uid and gid for the homedir
|
|
user_id = os.stat(HOME_DIR).st_uid
|
|
group_id = os.stat(HOME_DIR).st_gid
|
|
|
|
if not os.path.isdir(CONFIG_DIR):
|
|
os.makedirs(CONFIG_DIR)
|
|
os.chown(CONFIG_DIR, user_id, group_id)
|
|
|
|
if not os.path.isdir(OS_DIR):
|
|
os.makedirs(OS_DIR)
|
|
os.chown(OS_DIR, user_id, group_id)
|
|
|
|
if not os.path.isdir(GLOBAL_OS_DIR):
|
|
os.makedirs(GLOBAL_OS_DIR)
|
|
|
|
if not os.path.isfile(USER_CLOUDS_YAML):
|
|
_create_clouds_yaml(USER_CLOUDS_YAML)
|
|
|
|
if not os.path.isfile(GLOBAL_CLOUDS_YAML):
|
|
_create_clouds_yaml(GLOBAL_CLOUDS_YAML)
|
|
|
|
user_clouds = _read_clouds_yaml(USER_CLOUDS_YAML)
|
|
global_clouds = _read_clouds_yaml(GLOBAL_CLOUDS_YAML)
|
|
|
|
user_clouds['clouds'].update(CLOUD)
|
|
global_clouds['clouds'].update(CLOUD)
|
|
|
|
_write_clouds_yaml(USER_CLOUDS_YAML, user_clouds)
|
|
_write_clouds_yaml(GLOBAL_CLOUDS_YAML, global_clouds)
|
|
|
|
os.chown(USER_CLOUDS_YAML, user_id, group_id)
|
|
except Exception:
|
|
print('ERROR: Create clouds.yaml failed.')
|
|
raise
|