#!/usr/bin/env python ''' Example custom dynamic inventory script for Ansible, in Python. ''' from argparse import ArgumentParser from json import dumps, JSONEncoder from os import remove from os.path import isfile from asyncio import run from risotto.machine import load from risotto.image import load_config from risotto.utils import SERVERS from tiramisu.error import PropertiesOptionError from rougail.utils import normalize_family from rougail import RougailSystemdTemplate from rougail.template.base import RougailLeader, RougailExtra TIRAMISU_CACHE = 'tiramisu_cache.py' VALUES_CACHE = 'values_cache.py' class RougailEncoder(JSONEncoder): def default(self, obj): if isinstance(obj, RougailLeader): return obj._follower if isinstance(obj, RougailExtra): return obj._suboption if isinstance(obj, PropertiesOptionError): return 'PropertiesOptionError' return JSONEncoder.default(self, obj) class RisottoInventory(object): def __init__(self): parser = ArgumentParser() parser.add_argument('--list', action='store_true') parser.add_argument('--host', action='store') self.args = parser.parse_args() async def run(self): if self.args.list: if isfile(TIRAMISU_CACHE): remove(TIRAMISU_CACHE) if isfile(VALUES_CACHE): remove(VALUES_CACHE) return await self.do_inventory() elif self.args.host: return await self.get_vars(self.args.host) raise Exception('pfff') async def do_inventory(self): module_infos = load_config(True, True, True, ) servers = [] for server_name, server in SERVERS.items(): module_name = server['module'] if module_name != 'host': continue servers.append(server_name) return dumps({ 'group': { 'hosts': servers, 'vars': { # FIXME 'ansible_ssh_host': '192.168.56.156', 'ansible_ssh_user': 'root', 'ansible_python_interpreter': '/usr/bin/python3' } } }) async def get_vars(self, host_name: str, ) -> dict: try: module_infos, rougailconfig, config = await load(TIRAMISU_CACHE, VALUES_CACHE, ) except Exception as err: # import traceback # traceback.print_exc() print(err) exit(1) ret = {} modules = set() for server_name, server in SERVERS.items(): if server['module'] == 'host' and server_name != host_name: continue modules.add(server['module']) subconfig = config.option(normalize_family(server_name)) engine = RougailSystemdTemplate(subconfig, rougailconfig) await engine.load_variables() if server['module'] != 'host' and engine.rougail_variables_dict['general']['host'] != host_name: continue ret[server_name] = engine.rougail_variables_dict ret['modules'] = {module_name: module_info['infos'].depends for module_name, module_info in module_infos.items() if module_name in modules} ret['delete_old_image'] = False ret['configure_host'] = True ret['only_machine'] = None return dumps(ret, cls=RougailEncoder) # Get the inventory. async def main(): inv = RisottoInventory() values = await inv.run() print(values) run(main())