diff options
author | Jisi Liu <liujisi@google.com> | 2018-03-12 15:27:10 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-03-12 15:27:10 -0700 |
commit | 5487d8ca0f5d76f930ea24ae62f67703592b2cdf (patch) | |
tree | d80d292b49b277194e54e563c09a58c369e25a28 | |
parent | 478751903ab6ee34d52ace35b6f2620ea417f33f (diff) | |
parent | 294b5758c373cbab4b72f35f4cb62dc1d8332b68 (diff) |
Merge pull request #4380 from mateuszmatejczyk/patch-1
Output *_pb2_grpc.py files when use_grpc_plugin=True
-rw-r--r-- | protobuf.bzl | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/protobuf.bzl b/protobuf.bzl index 1c2d63c6..78f19c62 100644 --- a/protobuf.bzl +++ b/protobuf.bzl @@ -45,8 +45,11 @@ def _CcSrcs(srcs, use_grpc_plugin=False): def _CcOuts(srcs, use_grpc_plugin=False): return _CcHdrs(srcs, use_grpc_plugin) + _CcSrcs(srcs, use_grpc_plugin) -def _PyOuts(srcs): - return [s[:-len(".proto")] + "_pb2.py" for s in srcs] +def _PyOuts(srcs, use_grpc_plugin=False): + ret = [s[:-len(".proto")] + "_pb2.py" for s in srcs] + if use_grpc_plugin: + ret += [s[:-len(".proto")] + "_pb2_grpc.py" for s in srcs] + return ret def _RelativeOutputPath(path, include, dest=""): if include == None: @@ -344,7 +347,7 @@ def py_proto_library( **kargs: other keyword arguments that are passed to cc_library. """ - outs = _PyOuts(srcs) + outs = _PyOuts(srcs, use_grpc_plugin) includes = [] if include != None: |