json2yaml.py

json2yaml.py

#!/usr/bin/env python3

# Uses PyYAML>=0.2.5
from yaml import dump as yaml_dump
from json import load as json_load, JSONDecodeError
from argparse import ArgumentParser
from os.path import exists as path_exists
from sys import exit, stderr


def main():
    argparser = ArgumentParser(description="Convert a JSON file to YAML")
    argparser.add_argument(
        "json_file",
        type=str,
        help="The JSON file to convert"
    )
    args = argparser.parse_args()
    json_file = args.json_file

    if not path_exists(json_file):
        print("'{}' not found".format(json_file), file=stderr)
        return 1

    with open(json_file) as f:
        try:
            as_dict = json_load(f)
        except JSONDecodeError:
            print("'{}' is not valid JSON".format(json_file), file=stderr)
            return 1

    print(yaml_dump(as_dict))

    return 0


if __name__ == '__main__':
    exit(main())