preprocess_spec.py 文件源码

def main():
    if len(sys.argv) != 3:
        print("Usage:\n\n\tpython preprocess_spec.py kuberneres_branch " \
              "output_spec_path")
        return 1
    spec_url = 'https://raw.githubusercontent.com/kubernetes/kubernetes/' \
               '%s/api/openapi-spec/swagger.json' % sys.argv[1]
    output_path = sys.argv[2]

    pool = urllib3.PoolManager()
    with pool.request('GET', spec_url, preload_content=False) as response:
        if response.status != 200:
            print("Error downloading spec file. Reason: %s" % response.reason)
            return 1
        in_spec = json.load(response, object_pairs_hook=OrderedDict)
        out_spec = process_swagger(in_spec)
        with open(output_path, 'w') as out:
            json.dump(out_spec, out, sort_keys=False, indent=2,
                      separators=(',', ': '), ensure_ascii=True)
    return 0