Source code for mpi_cmake_modules.yaml2oneline

#!/usr/bin/env python
""" Convert a YAML file in a one line string

Example:

    For example, a YAML file with the content.

        foo: 13
        bar: 42

    would be converted into
    
        {foo: 13, bar: 42}

License BSD-3-Clause
Copyright (c) 2021, New York University and Max Planck Gesellschaft.
"""

import sys
import yaml


def yaml2oneline(filename):
    """Convert a yaml file into a dictionnary one-line string

    This function is used here typically in order to parse the yaml config file
    defining the clang-format parameters.

    Args:
        filename: str `--` Path relative to the execution path or global path.
    """
    with open(filename, "r") as fh:
        data = yaml.load(fh, Loader=yaml.SafeLoader)
        return yaml.dump(data, default_flow_style=True).replace("\n", "")


if __name__ == "__main__":
    """ Execute yaml2oneline on the input yaml file. """

    ## input yaml file name
    filename = sys.argv[1]
    print(yaml2oneline(filename))