blob: 1d0d2b1d0b27839d6b6adac2400e06a0d4e54041 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
import sys
import subprocess
import os.path
def has_grpcio_protoc():
# type: () -> bool
""" checks if grpcio-tools protoc is installed"""
try:
import grpc_tools.protoc
except ImportError:
return False
return True
def get_proto_builtin_include_path():
"""Find include path for standard google/protobuf includes and for
nanopb.proto.
"""
if getattr(sys, 'frozen', False):
# Pyinstaller package
paths = [
os.path.join(os.path.dirname(os.path.abspath(sys.executable)), 'proto'),
os.path.join(os.path.dirname(os.path.abspath(sys.executable)), 'grpc_tools', '_proto')
]
else:
# Stand-alone script
paths = [
os.path.dirname(os.path.abspath(__file__))
]
if has_grpcio_protoc():
import pkg_resources
paths.append(pkg_resources.resource_filename('grpc_tools', '_proto'))
return paths
def invoke_protoc(argv):
# type: (list) -> typing.Any
"""
Invoke protoc.
This routine will use grpcio-provided protoc if it exists,
using system-installed protoc as a fallback.
Args:
argv: protoc CLI invocation, first item must be 'protoc'
"""
# Add current directory to include path if nothing else is specified
if not [x for x in argv if x.startswith('-I')]:
argv.append("-I.")
# Add default protoc include paths
for incpath in get_proto_builtin_include_path():
argv.append('-I' + incpath)
if has_grpcio_protoc():
import grpc_tools.protoc as protoc
return protoc.main(argv)
else:
return subprocess.call(argv)
def print_versions():
try:
if has_grpcio_protoc():
import grpc_tools.protoc
sys.stderr.write("Using grpcio-tools protoc from " + grpc_tools.protoc.__file__ + "\n")
else:
sys.stderr.write("Using protoc from system path\n")
invoke_protoc(['protoc', '--version'])
except Exception as e:
sys.stderr.write("Failed to determine protoc version: " + str(e) + "\n")
try:
sys.stderr.write("protoc builtin include path: " + str(get_proto_builtin_include_path()) + "\n")
except Exception as e:
sys.stderr.write("Failed to construct protoc include path: " + str(e) + "\n")
try:
import google.protobuf
sys.stderr.write("Python version " + sys.version + "\n")
sys.stderr.write("Using python-protobuf from " + google.protobuf.__file__ + "\n")
sys.stderr.write("Python-protobuf version: " + google.protobuf.__version__ + "\n")
except Exception as e:
sys.stderr.write("Failed to determine python-protobuf version: " + str(e) + "\n")
if __name__ == '__main__':
print_versions()
|
