| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128 | #!/usr/bin/env python# Copyright 2016, Google Inc.# All rights reserved.## Redistribution and use in source and binary forms, with or without# modification, are permitted provided that the following conditions are# met:##     * Redistributions of source code must retain the above copyright# notice, this list of conditions and the following disclaimer.#     * Redistributions in binary form must reproduce the above# copyright notice, this list of conditions and the following disclaimer# in the documentation and/or other materials provided with the# distribution.#     * Neither the name of Google Inc. nor the names of its# contributors may be used to endorse or promote products derived from# this software without specific prior written permission.## THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.import osimport os.pathimport shutilimport subprocessDEPS_FILE_CONTENT="""# Copyright 2016, Google Inc.# All rights reserved.## Redistribution and use in source and binary forms, with or without# modification, are permitted provided that the following conditions are# met:##     * Redistributions of source code must retain the above copyright# notice, this list of conditions and the following disclaimer.#     * Redistributions in binary form must reproduce the above# copyright notice, this list of conditions and the following disclaimer# in the documentation and/or other materials provided with the# distribution.#     * Neither the name of Google Inc. nor the names of its# contributors may be used to endorse or promote products derived from# this software without specific prior written permission.## THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.# AUTO-GENERATED BY make_grpcio_tools.py!CC_FILES={}"""# Bazel query result prefix for expected source files in protobuf.PROTOBUF_CC_PREFIX = '//:src/'GRPC_ROOT = os.path.abspath(    os.path.join(os.path.dirname(os.path.abspath(__file__)),                 '..', '..', '..'))GRPC_PYTHON_ROOT = os.path.join(GRPC_ROOT, 'tools/distrib/python/grpcio_tools')GRPC_PROTOBUF = os.path.join(GRPC_ROOT, 'third_party/protobuf/src')GRPC_PROTOC_PLUGINS = os.path.join(GRPC_ROOT, 'src/compiler')GRPC_PYTHON_PROTOBUF = os.path.join(GRPC_PYTHON_ROOT,                                    'third_party/protobuf/src')GRPC_PYTHON_PROTOC_PLUGINS = os.path.join(GRPC_PYTHON_ROOT,                                          'grpc_root/src/compiler')GRPC_PYTHON_PROTOC_LIB_DEPS = os.path.join(GRPC_PYTHON_ROOT,                                           'protoc_lib_deps.py')GRPC_INCLUDE = os.path.join(GRPC_ROOT, 'include')GRPC_PYTHON_INCLUDE = os.path.join(GRPC_PYTHON_ROOT, 'grpc_root/include')BAZEL_DEPS = os.path.join(GRPC_ROOT, 'tools/distrib/python/bazel_deps.sh')BAZEL_DEPS_PROTOC_LIB_QUERY = '//:protoc_lib'def write_deps(query, out_file):  """Write the result of the bazel query `query` against protobuf to     `out_file`."""  output = subprocess.check_output([BAZEL_DEPS, query])  output = output.splitlines()  cc_files = [      name for name in output      if name.endswith('.cc') and name.startswith(PROTOBUF_CC_PREFIX)]  cc_files = [cc_file[len(PROTOBUF_CC_PREFIX):] for cc_file in cc_files]  deps_file_content = DEPS_FILE_CONTENT.format(cc_files)  out_file.write(deps_file_content)def main():  os.chdir(GRPC_ROOT)  for tree in [GRPC_PYTHON_PROTOBUF,               GRPC_PYTHON_PROTOC_PLUGINS,               GRPC_PYTHON_INCLUDE]:    try:      shutil.rmtree(tree)    except Exception as _:      pass  shutil.copytree(GRPC_PROTOBUF, GRPC_PYTHON_PROTOBUF)  shutil.copytree(GRPC_PROTOC_PLUGINS, GRPC_PYTHON_PROTOC_PLUGINS)  shutil.copytree(GRPC_INCLUDE, GRPC_PYTHON_INCLUDE)  with open(GRPC_PYTHON_PROTOC_LIB_DEPS, 'w') as deps_file:    write_deps(BAZEL_DEPS_PROTOC_LIB_QUERY, deps_file)if __name__ == '__main__':  main()
 |