mirror of https://github.com/python/cpython.git
122 lines
4.5 KiB
Python
122 lines
4.5 KiB
Python
"""Command-line tool to validate and pretty-print JSON
|
|
|
|
See `json.__main__` for a usage example (invocation as
|
|
`python -m json.tool` is supported for backwards compatibility).
|
|
"""
|
|
import argparse
|
|
import json
|
|
import re
|
|
import sys
|
|
from _colorize import get_theme, can_colorize
|
|
|
|
|
|
# The string we are colorizing is valid JSON,
|
|
# so we can use a looser but simpler regex to match
|
|
# the various parts, most notably strings and numbers,
|
|
# where the regex given by the spec is much more complex.
|
|
_color_pattern = re.compile(r'''
|
|
(?P<key>"(\\.|[^"\\])*")(?=:) |
|
|
(?P<string>"(\\.|[^"\\])*") |
|
|
(?P<number>NaN|-?Infinity|[0-9\-+.Ee]+) |
|
|
(?P<boolean>true|false) |
|
|
(?P<null>null)
|
|
''', re.VERBOSE)
|
|
|
|
_group_to_theme_color = {
|
|
"key": "definition",
|
|
"string": "string",
|
|
"number": "number",
|
|
"boolean": "keyword",
|
|
"null": "keyword",
|
|
}
|
|
|
|
|
|
def _colorize_json(json_str, theme):
|
|
def _replace_match_callback(match):
|
|
for group, color in _group_to_theme_color.items():
|
|
if m := match.group(group):
|
|
return f"{theme[color]}{m}{theme.reset}"
|
|
return match.group()
|
|
|
|
return re.sub(_color_pattern, _replace_match_callback, json_str)
|
|
|
|
|
|
def main():
|
|
description = ('A simple command line interface for json module '
|
|
'to validate and pretty-print JSON objects.')
|
|
parser = argparse.ArgumentParser(description=description, color=True)
|
|
parser.add_argument('infile', nargs='?',
|
|
help='a JSON file to be validated or pretty-printed',
|
|
default='-')
|
|
parser.add_argument('outfile', nargs='?',
|
|
help='write the output of infile to outfile',
|
|
default=None)
|
|
parser.add_argument('--sort-keys', action='store_true', default=False,
|
|
help='sort the output of dictionaries alphabetically by key')
|
|
parser.add_argument('--no-ensure-ascii', dest='ensure_ascii', action='store_false',
|
|
help='disable escaping of non-ASCII characters')
|
|
parser.add_argument('--json-lines', action='store_true', default=False,
|
|
help='parse input using the JSON Lines format. '
|
|
'Use with --no-indent or --compact to produce valid JSON Lines output.')
|
|
group = parser.add_mutually_exclusive_group()
|
|
group.add_argument('--indent', default=4, type=int,
|
|
help='separate items with newlines and use this number '
|
|
'of spaces for indentation')
|
|
group.add_argument('--tab', action='store_const', dest='indent',
|
|
const='\t', help='separate items with newlines and use '
|
|
'tabs for indentation')
|
|
group.add_argument('--no-indent', action='store_const', dest='indent',
|
|
const=None,
|
|
help='separate items with spaces rather than newlines')
|
|
group.add_argument('--compact', action='store_true',
|
|
help='suppress all whitespace separation (most compact)')
|
|
options = parser.parse_args()
|
|
|
|
dump_args = {
|
|
'sort_keys': options.sort_keys,
|
|
'indent': options.indent,
|
|
'ensure_ascii': options.ensure_ascii,
|
|
}
|
|
if options.compact:
|
|
dump_args['indent'] = None
|
|
dump_args['separators'] = ',', ':'
|
|
|
|
try:
|
|
if options.infile == '-':
|
|
infile = sys.stdin
|
|
else:
|
|
infile = open(options.infile, encoding='utf-8')
|
|
try:
|
|
if options.json_lines:
|
|
objs = (json.loads(line) for line in infile)
|
|
else:
|
|
objs = (json.load(infile),)
|
|
finally:
|
|
if infile is not sys.stdin:
|
|
infile.close()
|
|
|
|
if options.outfile is None:
|
|
outfile = sys.stdout
|
|
else:
|
|
outfile = open(options.outfile, 'w', encoding='utf-8')
|
|
with outfile:
|
|
if can_colorize(file=outfile):
|
|
t = get_theme(tty_file=outfile).syntax
|
|
for obj in objs:
|
|
json_str = json.dumps(obj, **dump_args)
|
|
outfile.write(_colorize_json(json_str, t))
|
|
outfile.write('\n')
|
|
else:
|
|
for obj in objs:
|
|
json.dump(obj, outfile, **dump_args)
|
|
outfile.write('\n')
|
|
except ValueError as e:
|
|
raise SystemExit(e)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
try:
|
|
main()
|
|
except BrokenPipeError as exc:
|
|
raise SystemExit(exc.errno)
|