concatenate_protocols.py 1.11 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13
#!/usr/bin/env python
# Copyright 2016 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import os.path
import sys

try:
    import json
except ImportError:
    import simplejson as json

14
import pdl
15 16 17 18 19 20 21 22 23 24 25 26 27 28

def main(argv):
    if len(argv) < 1:
        sys.stderr.write("Usage: %s <protocol-1> [<protocol-2> [, <protocol-3>...]] <output-file>\n" % sys.argv[0])
        return 1

    domains = []
    version = None
    for protocol in argv[:-1]:
        file_name = os.path.normpath(protocol)
        if not os.path.isfile(file_name):
            sys.stderr.write("Cannot find %s\n" % file_name)
            return 1
        input_file = open(file_name, "r")
29
        parsed_json = pdl.loads(input_file.read(), file_name)
30 31 32 33 34 35 36 37 38 39
        domains += parsed_json["domains"]
        version = parsed_json["version"]

    output_file = open(argv[-1], "w")
    json.dump({"version": version, "domains": domains}, output_file, indent=4, sort_keys=False, separators=(',', ': '))
    output_file.close()


if __name__ == '__main__':
    sys.exit(main(sys.argv[1:]))