Format YAML with Python 3rdparty module ruamel.yaml to preserve comments and quotes

format-yaml.py

#!/usr/bin/env python3

import argparse
import io

import ruamel.yaml


if __name__ == "__main__":
  p = argparse.ArgumentParser()
  p.add_argument("fp", default=[], nargs="+", help="files to format")
  args = p.parse_args()

  y = ruamel.yaml.YAML()
  y.preserve_quotes = True
  y.width = 1024

  for fp in args.fp:
    buf = io.StringIO()
    with open(fp, "r+") as f:
      d = y.load(f)
      f.seek(0)
      y.dump(d, f)
      f.truncate()