gistfile1.py

#!/usr/bin/env python
"""
Convert YAML to JSON and JSON to YAML
"""

import yaml
import json
import sys


if __name__ == '__main__':
    file_path = sys.argv[1]
    with open(file_path, 'r') as fp:
        raw = yaml.load(fp.read())
    if file_path.split('.')[-1] == 'json':
        sys.stdout.write(yaml.dump(raw))
    if file_path.split('.')[-1] == 'yaml':
        sys.stdout.write(json.dumps(raw))