File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -45,8 +45,11 @@ def _CcSrcs(srcs, use_grpc_plugin=False):
45
45
def _CcOuts (srcs , use_grpc_plugin = False ):
46
46
return _CcHdrs (srcs , use_grpc_plugin ) + _CcSrcs (srcs , use_grpc_plugin )
47
47
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
50
53
51
54
def _RelativeOutputPath (path , include , dest = "" ):
52
55
if include == None :
@@ -344,7 +347,7 @@ def py_proto_library(
344
347
**kargs: other keyword arguments that are passed to cc_library.
345
348
346
349
"""
347
- outs = _PyOuts (srcs )
350
+ outs = _PyOuts (srcs , use_grpc_plugin )
348
351
349
352
includes = []
350
353
if include != None :
You can’t perform that action at this time.
0 commit comments