yaml-reader.py

#! /usr/bin/env python

from __future__ import print_function

import sys
import pprint
import argparse

import json
import yaml

def main():

    parser = argparse.ArgumentParser()
    parser.add_argument("-f", "--format",
                        metavar="<pretty|json|yaml>",
                        default="pretty",
                        help="Output format: pretty,json,yaml")
    parser.add_argument("input", help="YAML file to read.")
    args = parser.parse_args()

    try:
        content = yaml.load(open(args.input))
    except Exception as err:
        print("ERROR:")
        print(err)
        return
    
    
    if args.format.lower() == "yaml":
        print(yaml.dump(content, default_flow_style=False))
    elif args.format.lower() == "json":
        print(json.dumps(content, indent=4))
    else:
        # Just a Python pretty print.
        pprint.PrettyPrinter().pprint(content)

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