• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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 os
17import os.path
18import sys
19
20import setuptools
21
22import grpc_tools.command
23
24PY3 = sys.version_info.major == 3
25
26# Ensure we're in the proper directory whether or not we're being used by pip.
27os.chdir(os.path.dirname(os.path.abspath(__file__)))
28
29# Break import-style to ensure we can actually find our in-repo dependencies.
30import commands
31import grpc_version
32
33LICENSE = 'Apache License 2.0'
34
35PACKAGE_DIRECTORIES = {
36    '': '.',
37}
38
39INSTALL_REQUIRES = (
40    'coverage>=4.0', 'enum34>=1.0.4',
41    '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',)
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
97setuptools.setup(
98    name='grpcio-tests',
99    version=grpc_version.VERSION,
100    license=LICENSE,
101    packages=list(PACKAGES),
102    package_dir=PACKAGE_DIRECTORIES,
103    package_data=PACKAGE_DATA,
104    install_requires=INSTALL_REQUIRES,
105    cmdclass=COMMAND_CLASS,
106    tests_require=TESTS_REQUIRE,
107    test_suite=TEST_SUITE,
108    test_loader=TEST_LOADER,
109    test_runner=TEST_RUNNER,
110)
111