tests/docker/travis.py

#!/usr/bin/env python
#
# Travis YAML config parser
#
# Copyright (c) 2016 Red Hat Inc.
#
# Authors:
#  Fam Zheng <famz@redhat.com>
#
# This work is licensed under the terms of the GNU GPL, version 2
# or (at your option) any later version. See the COPYING file in
# the top-level directory.

from __future__ import print_function
import sys
import yaml
import itertools

def load_yaml(fname):
    return yaml.safe_load(open(fname, "r").read())

def conf_iter(conf):
    # If "compiler" is omitted from the included env then Travis picks the
    # first entry of the global compiler list.
    default_compiler = conf["compiler"][0]
    def env_to_list(env):
        return env if isinstance(env, list) else [env]
    for entry in conf["matrix"]["include"]:
        yield {"env": env_to_list(entry["env"]),
               "compiler": entry.get("compiler", default_compiler)}

def main():
    if len(sys.argv) < 2:
        sys.stderr.write("Usage: %s <travis-file>\n" % sys.argv[0])
        return 1
    conf = load_yaml(sys.argv[1])
    print("\n".join((": ${%s}" % var for var in conf["env"]["global"])))
    for config in conf_iter(conf):
        print("(")
        print("\n".join(config["env"]))
        print("alias cc=" + config["compiler"])
        print("\n".join(conf["before_script"]))
        print("\n".join(conf["script"]))
        print(")")
    return 0

if __name__ == "__main__":
    sys.exit(main())