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
|
#!/usr/bin/python
# -*- coding: utf-8 -*-
"""
This script generates a bundle config for the haproxy managing public apis
Parameters:
-l, --lab : lab config file
"""
from jinja2 import Environment, FileSystemLoader
from keystoneauth1.identity import v2
from keystoneauth1 import session
from keystoneclient.v2_0 import client
from optparse import OptionParser
import os
import yaml
#
# Parse parameters
#
parser = OptionParser()
parser.add_option("-l", "--lab", dest="lab", help="lab config file")
(options, args) = parser.parse_args()
labconfig_file = options.lab
#
# Set Path and configs path
#
# Capture our current directory
TPL_DIR = os.path.dirname(os.path.abspath(__file__))+'/config_tpl'
#
# Local Functions
#
def load_yaml(filepath):
"""Load YAML file"""
with open(filepath, 'r') as stream:
try:
return yaml.load(stream)
except yaml.YAMLError as exc:
print(exc)
#
# Config import
#
# Load scenario Config
config = load_yaml(labconfig_file)
# Add public api ip to config
if 'public_api_ip' in config['lab']['racks'][0]:
config['public_api_ip'] = config['lab']['racks'][0]['public_api_ip']
else:
first_public_ip = config['lab']['racks'][0][
'floating-ip-range'].split(',')[0]
# managing ipv6 and ipv4 format
sep = ':' if ':' in first_public_ip else '.'
api_ip = first_public_ip.split(sep)
api_ip[-1] = str(int(api_ip[-1])-1)
config['public_api_ip'] = sep.join(api_ip)
# get endpoint list from keystone
username = os.environ['OS_USERNAME']
password = os.environ['OS_PASSWORD']
tenant_name = os.environ['OS_TENANT_NAME']
auth_url = os.environ['OS_AUTH_URL']
auth = v2.Password(username=username,
password=password,
tenant_name=tenant_name,
auth_url=auth_url)
sess = session.Session(auth=auth)
keystone = client.Client(session=sess)
services = keystone.services.list()
endpoints = keystone.endpoints.list()
srv = dict()
for service in services:
if service.name != 'cinderv2':
srv[service.id] = {'name': service.name}
for endpoint in endpoints:
if endpoint.service_id in srv.keys():
internal = endpoint.internalurl.split('/')[2].split(':')
srv[endpoint.service_id]['ip'] = ':'.join(internal[:-1])
srv[endpoint.service_id]['port'] = internal[-1]
config['public_api_services'] = srv
#
# Transform template to deployconfig.yaml according to config
#
# Create the jinja2 environment.
env = Environment(loader=FileSystemLoader(TPL_DIR),
trim_blocks=True)
template = env.get_template('public-api-proxy.yaml')
# Render the template
output = template.render(**config)
# Check output syntax
try:
yaml.load(output)
except yaml.YAMLError as exc:
print(exc)
# print output
print(output)
|
