1# Copyright 2015 gRPC authors. 2# 3# Licensed under the Apache License, Version 2.0 (the "License"); 4# you may not use this file except in compliance with the License. 5# You may obtain a copy of the License at 6# 7# http://www.apache.org/licenses/LICENSE-2.0 8# 9# Unless required by applicable law or agreed to in writing, software 10# distributed under the License is distributed on an "AS IS" BASIS, 11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12# See the License for the specific language governing permissions and 13# limitations under the License. 14"""A setup module for the gRPC Python package.""" 15 16import multiprocessing 17import os 18import os.path 19import sys 20 21import setuptools 22 23import grpc_tools.command 24 25PY3 = sys.version_info.major == 3 26 27# Ensure we're in the proper directory whether or not we're being used by pip. 28os.chdir(os.path.dirname(os.path.abspath(__file__))) 29 30# Break import-style to ensure we can actually find our in-repo dependencies. 31import commands 32import grpc_version 33 34LICENSE = 'Apache License 2.0' 35 36PACKAGE_DIRECTORIES = { 37 '': '.', 38} 39 40INSTALL_REQUIRES = ( 41 'coverage>=4.0', 'grpcio>={version}'.format(version=grpc_version.VERSION), 42 'grpcio-channelz>={version}'.format(version=grpc_version.VERSION), 43 'grpcio-status>={version}'.format(version=grpc_version.VERSION), 44 'grpcio-tools>={version}'.format(version=grpc_version.VERSION), 45 'grpcio-health-checking>={version}'.format(version=grpc_version.VERSION), 46 'oauth2client>=1.4.7', 'protobuf>=3.6.0', 'six>=1.10', 47 'google-auth>=1.17.2', 'requests>=2.14.2') 48 49if not PY3: 50 INSTALL_REQUIRES += ('futures>=2.2.0', 'enum34>=1.0.4') 51 52COMMAND_CLASS = { 53 # Run `preprocess` *before* doing any packaging! 54 'preprocess': commands.GatherProto, 55 'build_package_protos': grpc_tools.command.BuildPackageProtos, 56 'build_py': commands.BuildPy, 57 'run_fork': commands.RunFork, 58 'run_interop': commands.RunInterop, 59 'test_lite': commands.TestLite, 60 'test_gevent': commands.TestGevent, 61 'test_aio': commands.TestAio, 62 'test_py3_only': commands.TestPy3Only, 63} 64 65PACKAGE_DATA = { 66 'tests.interop': [ 67 'credentials/ca.pem', 68 'credentials/server1.key', 69 'credentials/server1.pem', 70 ], 71 'tests.protoc_plugin.protos.invocation_testing': ['same.proto',], 72 'tests.protoc_plugin.protos.invocation_testing.split_messages': [ 73 'messages.proto', 74 ], 75 'tests.protoc_plugin.protos.invocation_testing.split_services': [ 76 'services.proto', 77 ], 78 'tests.testing.proto': [ 79 'requests.proto', 80 'services.proto', 81 ], 82 'tests.unit': [ 83 'credentials/ca.pem', 84 'credentials/server1.key', 85 'credentials/server1.pem', 86 ], 87 'tests': ['tests.json'], 88} 89 90TEST_SUITE = 'tests' 91TEST_LOADER = 'tests:Loader' 92TEST_RUNNER = 'tests:Runner' 93TESTS_REQUIRE = INSTALL_REQUIRES 94 95PACKAGES = setuptools.find_packages('.') 96 97if __name__ == "__main__": 98 multiprocessing.freeze_support() 99 setuptools.setup( 100 name='grpcio-tests', 101 version=grpc_version.VERSION, 102 license=LICENSE, 103 packages=list(PACKAGES), 104 package_dir=PACKAGE_DIRECTORIES, 105 package_data=PACKAGE_DATA, 106 install_requires=INSTALL_REQUIRES, 107 cmdclass=COMMAND_CLASS, 108 tests_require=TESTS_REQUIRE, 109 test_suite=TEST_SUITE, 110 test_loader=TEST_LOADER, 111 test_runner=TEST_RUNNER, 112 ) 113