Skip to content

Commit 294b575

Browse files
authored
Output *_pb2_grpc.py when use_grpc_plugin=True
Start outputting missing *_pb2_grpc.py files when use_grpc_plugin is set to True for py_proto_library.
1 parent 8e44a86 commit 294b575

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

protobuf.bzl

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -45,8 +45,11 @@ def _CcSrcs(srcs, use_grpc_plugin=False):
4545
def _CcOuts(srcs, use_grpc_plugin=False):
4646
return _CcHdrs(srcs, use_grpc_plugin) + _CcSrcs(srcs, use_grpc_plugin)
4747

48-
def _PyOuts(srcs):
49-
return [s[:-len(".proto")] + "_pb2.py" for s in srcs]
48+
def _PyOuts(srcs, use_grpc_plugin=False):
49+
ret = [s[:-len(".proto")] + "_pb2.py" for s in srcs]
50+
if use_grpc_plugin:
51+
ret += [s[:-len(".proto")] + "_pb2_grpc.py" for s in srcs]
52+
return ret
5053

5154
def _RelativeOutputPath(path, include, dest=""):
5255
if include == None:
@@ -344,7 +347,7 @@ def py_proto_library(
344347
**kargs: other keyword arguments that are passed to cc_library.
345348
346349
"""
347-
outs = _PyOuts(srcs)
350+
outs = _PyOuts(srcs, use_grpc_plugin)
348351

349352
includes = []
350353
if include != None:

0 commit comments

Comments
 (0)