diff options
author | Benjamin Peterson <benjamin@python.org> | 2016-06-11 13:18:56 -0700 |
---|---|---|
committer | Benjamin Peterson <benjamin@python.org> | 2016-06-11 13:18:56 -0700 |
commit | 5a991226e84e935f2a50deec664eaf3860893354 (patch) | |
tree | e674175fd5bc642be83ceb674d57e303b26f4326 /Lib/json/tool.py | |
parent | 9997d0207649b71b5862d3c036c1f7dcacf4f7b1 (diff) | |
parent | 1ccde8e1ad13ea6943f6ef5ea4329c9827ba493e (diff) | |
download | cpython-5a991226e84e935f2a50deec664eaf3860893354.tar.gz |
merge 3.4
Diffstat (limited to 'Lib/json/tool.py')
-rw-r--r-- | Lib/json/tool.py | 39 |
1 files changed, 25 insertions, 14 deletions
diff --git a/Lib/json/tool.py b/Lib/json/tool.py index 7db4528571..4f3182c0c1 100644 --- a/Lib/json/tool.py +++ b/Lib/json/tool.py @@ -10,28 +10,39 @@ Usage:: Expecting property name enclosed in double quotes: line 1 column 3 (char 2) """ -import sys +import argparse +import collections import json +import sys + def main(): - if len(sys.argv) == 1: - infile = sys.stdin - outfile = sys.stdout - elif len(sys.argv) == 2: - infile = open(sys.argv[1], 'r') - outfile = sys.stdout - elif len(sys.argv) == 3: - infile = open(sys.argv[1], 'r') - outfile = open(sys.argv[2], 'w') - else: - raise SystemExit(sys.argv[0] + " [infile [outfile]]") + prog = 'python -m json.tool' + description = ('A simple command line interface for json module ' + 'to validate and pretty-print JSON objects.') + parser = argparse.ArgumentParser(prog=prog, description=description) + parser.add_argument('infile', nargs='?', type=argparse.FileType(), + help='a JSON file to be validated or pretty-printed') + parser.add_argument('outfile', nargs='?', type=argparse.FileType('w'), + help='write the output of infile to outfile') + parser.add_argument('--sort-keys', action='store_true', default=False, + help='sort the output of dictionaries alphabetically by key') + options = parser.parse_args() + + infile = options.infile or sys.stdin + outfile = options.outfile or sys.stdout + sort_keys = options.sort_keys with infile: try: - obj = json.load(infile) + if sort_keys: + obj = json.load(infile) + else: + obj = json.load(infile, + object_pairs_hook=collections.OrderedDict) except ValueError as e: raise SystemExit(e) with outfile: - json.dump(obj, outfile, sort_keys=True, indent=4) + json.dump(obj, outfile, sort_keys=sort_keys, indent=4) outfile.write('\n') |