jsonfmt.py

   1 #!/usr/bin/env python3
   2 # -*- coding: utf-8 -*-
   3
   4 """jsonfmt.py - JSON formatter
   5
   6 File ID: 423dd854-4a64-11e4-97b1-c80aa9e67bbd
   7 License: GNU General Public License version 2 or later.
   8 Author: Øyvind A. Holm <sunny@sunbase.org>
   9
  10 """
  11
  12
  13 def format_json(text, indent=4):
  14     """Return formatted JSON"""
  15     import json
  16
  17     return json.dumps(
  18         json.loads(text),
  19         ensure_ascii=False,
  20         indent=indent,
  21         sort_keys=True,
  22     )
  23
  24
  25 def main():
  26     import argparse
  27     import os
  28     import sys
  29
  30     progname = os.path.basename(__file__)
  31
  32     parser = argparse.ArgumentParser(
  33         description="JSON formatter",
  34     )
  35     parser.add_argument(
  36         "-i",
  37         "--indent",
  38         default=2,
  39         help="Use X spaces as indent",
  40         metavar="X",
  41         type=int,
  42     )
  43     args = parser.parse_args()
  44
  45     try:
  46         print(
  47             format_json(
  48                 "".join(sys.stdin.readlines()),
  49                 indent=args.indent,
  50             )
  51         )
  52     except ValueError:
  53         sys.stderr.write("%s: Invalid JSON\n" % progname)
  54
  55
  56 if __name__ == "__main__":
  57     main()