ini2yaml

#! /usr/bin/env python3
'''
Executable to convert ansible ini inventory file into yaml file
'''
import sys
import re
import configparser

import six
import yaml


class LiteralUniCode(six.text_type):
    """literal unicode for representer"""


def literal_unicode_representer(dumper, data):
    '''unicode representer for yaml representation'''
    return dumper.represent_scalar('tag:yaml.org,2002:str', data, style='|')


yaml.add_representer(LiteralUniCode, literal_unicode_representer)

config = configparser.RawConfigParser(allow_no_value=True)
config.optionxform = str
config.read_file(sys.stdin)

inventory = {}

varRegex = re.compile("[\t ]*([a-zA-Z][a-zA-Z0-9_]+)=('[^']+'|\"[^\"]+\"|[^ ]+)")
noQuotesNeededRegex = re.compile("^([-.0-9a-zA-Z]+|'[^']+'|\"[^\"]+\")$")


# Parse host variable and return corresponding YAML object
def parse_value(host_vars):
    '''Parses the values under host segment'''
    # Integers, booleans and quoted strings strings must not be quoted
    if noQuotesNeededRegex.match(host_vars):
        result = yaml.load('value: ' + host_vars, Loader=yaml.FullLoader)['value']
    else:
        result = yaml.load('value: "' + host_vars + '"', Loader=yaml.FullLoader)['value']
    if isinstance(result, six.string_types):
        # Use YAML block literal for multi-line strings
        if '\\n' in result:
            return LiteralUniCode(result.replace('\\n', '\n'))
        try:
            new_result = yaml.load('value: ' + result, Loader=yaml.FullLoader)['value']
            if isinstance(new_result, (dict, list)):
                result = new_result
        except yaml.scanner.ScannerError:
            pass
        else:
            pass
    return result


for section in config.sections():
    group = section.split(':')
    if len(group) == 1:  # section contains host group
        for name, value in config.items(section):
            if value:
                value = name + '=' + value
            else:
                value = name
            host = re.split(' |\t', value, 1)
            hostname = host[0]
            hostvars = host[1] if len(host) > 1 else ''
            hostvars = varRegex.findall(hostvars)

            inventory.setdefault('all', {}).setdefault('children', {}).setdefault(
                group[0], {}).setdefault('hosts', {})[hostname] = {}
            for hostvar in hostvars:
                value = parse_value(hostvar[1])
                inventory.setdefault('all', {}).setdefault('children', {}).setdefault(
                    group[0], {}).setdefault('hosts', {})[hostname][hostvar[0]] = value
    elif group[1] == 'vars':  # section contains group vars
        for name, value in config.items(section):
            value = parse_value(value)
            inventory.setdefault('all', {}).setdefault('children', {}).setdefault(
                group[0], {}).setdefault('vars', {})[name] = value
    elif group[1] == 'children':  # section contains group of groups
        for name, value in config.items(section):
            inventory.setdefault('all', {}).setdefault('children', {}).setdefault(
                group[0], {}).setdefault('children', {})[name] = {}

print(yaml.safe_dump(inventory, default_flow_style=False, width=float("inf")))