transformation script

transformation.py

import glob
import yaml


def read_yaml(filename):
    """ Read yaml file """
    with open(filename, 'r') as data:
        return yaml.load(data)


def write_yaml(filename, data):
    """ Write data """
    with open(filename, 'w') as outfile:
        outfile.write(yaml.dump(data, default_flow_style=False))


def convert_dict(dictionary):
    text = ''
    for key in sorted(dictionary):
        text += "#### {0}  \n{1}  \n".format(key, dictionary[key])
    return text


def file_loop():
    """ Loop though files """
    for component in glob.iglob("*/component.yaml"):
        data = read_yaml(component)
        for satisfies in data['satisfies']:
            if isinstance(satisfies['narrative'], dict):
                satisfies['narrative'] = convert_dict(satisfies['narrative'])
                write_yaml(component, data)


if __name__ == "__main__":
    file_loop()