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 26import glob 27import json 28import os 29import sys 30 31if 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 35files = [] 36for 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) 43files.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 49first = True 50print("{\"domains\":[") 51for 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()) 68print("]}") 69 70