YAML reformatting

reformat_yaml.py

# to install yaml:
#     sudo pip install pyyaml

import argparse
import yaml
import sys


parser = argparse.ArgumentParser(description="normalize YAML formatting")
parser.add_argument("--mode", help="how to pretty-print innermost nodes",
                        choices=["flow", "block"], required=True)
parser.add_argument("input", help="input YAML (or JSON) file")
parser.add_argument("output", help="output YAML file")

args = parser.parse_args()
use_flow = args.mode == "flow"

with open(args.input, "r") as f:
    data = yaml.load(f)
with open(args.output, "w") as f:
    yaml.dump(data, f, default_flow_style=use_flow)