config_parse.py 709 Bytes
Newer Older
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
"""
Helper functions for configuration parsing
"""
import collections


def convert_tokens(tokens):
    """
    This function is called on the token
    dictionary that is imported from a yaml file.
    It returns a new dictionary where
    all strings containing 'None' are converted
    to a literal None due to a bug in Ansible
    """

    if tokens == 'None':
        return None
    elif isinstance(tokens, basestring) or (not isinstance(tokens, collections.Iterable)):
        return tokens
    elif isinstance(tokens, dict):
        return {
            convert_tokens(k): convert_tokens(v)
            for k, v in tokens.items()
        }
    else:
        return [convert_tokens(v) for v in tokens]