59 lines
998 B
Python
Executable File
59 lines
998 B
Python
Executable File
#!/usr/bin/python
|
|
|
|
import sys
|
|
|
|
from charmhelpers.core.hookenv import (
|
|
Hooks,
|
|
UnregisteredHookError,
|
|
config,
|
|
log,
|
|
relation_set,
|
|
)
|
|
|
|
from pg_dir_utils import (
|
|
register_configs,
|
|
ensure_files,
|
|
restart_pg,
|
|
stop_pg,
|
|
)
|
|
|
|
hooks = Hooks()
|
|
CONFIGS = register_configs()
|
|
|
|
|
|
@hooks.hook()
|
|
def install():
|
|
ensure_files()
|
|
|
|
|
|
@hooks.hook('plumgrid-plugin-relation-joined')
|
|
def plumgrid_dir():
|
|
ensure_files()
|
|
CONFIGS.write_all()
|
|
restart_pg()
|
|
|
|
|
|
@hooks.hook('plumgrid-relation-joined')
|
|
def plumgrid_joined(relation_id=None):
|
|
#We can pass information to the edge and gateway from there.
|
|
relation_data = {
|
|
'pg_virtual_ip': config('plumgrid-virtual-ip'),
|
|
}
|
|
relation_set(relation_id=relation_id, **relation_data)
|
|
|
|
|
|
@hooks.hook('stop')
|
|
def stop():
|
|
stop_pg()
|
|
|
|
|
|
def main():
|
|
try:
|
|
hooks.execute(sys.argv)
|
|
except UnregisteredHookError as e:
|
|
log('Unknown hook {} - skipping.'.format(e))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|