charm-plumgrid-gateway/hooks/pg_gw_context.py

78 lines
2.0 KiB
Python

# Copyright (c) 2015, PLUMgrid Inc, http://plumgrid.com
# This file contains the class that generates context for PLUMgrid template files.
from charmhelpers.core.hookenv import (
relation_ids,
related_units,
relation_get,
config,
)
from charmhelpers.contrib.openstack import context
from socket import gethostname as get_unit_hostname
def _pg_dir_settings():
'''
Inspects relation with PLUMgrid director.
'''
pg_settings = {
'pg_dir_ip': '192.168.100.201',
}
for rid in relation_ids('plumgrid'):
for unit in related_units(rid):
rdata = relation_get(rid=rid, unit=unit)
pg_settings = {
'pg_dir_ip': rdata['private-address'],
}
return pg_settings
class PGGwContext(context.NeutronContext):
@property
def plugin(self):
'''
Over-riding function in NeutronContext Class to return 'plumgrid'
as the neutron plugin.
'''
return 'plumgrid'
@property
def network_manager(self):
'''
Over-riding function in NeutronContext Class to return 'neutron'
as the network manager.
'''
return 'neutron'
def _save_flag_file(self):
'''
Over-riding function in NeutronContext Class.
Function only needed for OVS.
'''
pass
def pg_ctxt(self):
'''
Generated Config for all PLUMgrid templates inside the templates folder.
'''
pg_ctxt = super(PGGwContext, self).pg_ctxt()
if not pg_ctxt:
return {}
conf = config()
pg_dir_settings = _pg_dir_settings()
pg_ctxt['local_ip'] = pg_dir_settings['pg_dir_ip']
unit_hostname = get_unit_hostname()
pg_ctxt['pg_hostname'] = unit_hostname
pg_ctxt['interface'] = "juju-br0"
pg_ctxt['label'] = unit_hostname
pg_ctxt['fabric_mode'] = 'host'
pg_ctxt['ext_interface'] = conf['external-interface']
return pg_ctxt