inspector/scripts/generate-combined-inspector-json.py

   1 #!/usr/bin/python
   2
   3 # Copyright (C) 2013 Apple Inc. All rights reserved.
   4 #
   5 # Redistribution and use in source and binary forms, with or without
   6 # modification, are permitted provided that the following conditions
   7 # are met:
   8 # 1. Redistributions of source code must retain the above copyright
   9 #    notice, this list of conditions and the following disclaimer.
  10 # 2. Redistributions in binary form must reproduce the above copyright
  11 #    notice, this list of conditions and the following disclaimer in the
  12 #    documentation and/or other materials provided with the distribution.
  13 #
  14 # THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
  15 # AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  16 # THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  17 # PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
  18 # BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  19 # CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  20 # SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  21 # INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  22 # CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  23 # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  24 # THE POSSIBILITY OF SUCH DAMAGE.
  25
  26 import glob
  27 import json
  28 import os
  29 import sys
  30
  31 if len(sys.argv) < 2:
  32     print("usage: %s [json files or directory of json files ...]" % os.path.basename(sys.argv[0]))
  33     sys.exit(1)
  34
  35 files = []
  36 for arg in sys.argv[1:]:
  37     if not os.access(arg, os.F_OK):
  38         raise Exception("File \"%s\" not found" % arg)
  39     elif os.path.isdir(arg):
  40         files.extend(glob.glob(os.path.join(arg, "*.json")))
  41     else:
  42         files.append(arg)
  43 files.sort()
  44
  45 # To keep as close to the original JSON formatting as possible, just
  46 # dump each JSON input file unmodified into an array of "domains".
  47 # Validate each file is valid JSON and that there is a "domain" key.
  48
  49 first = True
  50 print("{\"domains\":[")
  51 for file in files:
  52     if first:
  53         first = False
  54     else:
  55         print(",")
  56
  57     string = open(file).read()
  58
  59     try:
  60         dictionary = json.loads(string)
  61         if not "domain" in dictionary:
  62             raise Exception("File \"%s\" does not contains a \"domain\" key." % file)
  63     except ValueError:
  64         sys.stderr.write("File \"%s\" does not contain valid JSON:\n" % file)
  65         raise
  66
  67     print(string.rstrip())
  68 print("]}")
  69