Merge pull request #57 from cisco-ie/expose_channel

Expose gRPC Channel in Client and ClientBuilder

Author: remingtonc

src/cisco_gnmi/__init__.py

@@ -30,4 +30,4 @@
 from .xe import XEClient
 from .builder import ClientBuilder
 
-__version__ = "1.0.9"
+__version__ = "1.0.10"

src/cisco_gnmi/builder.py

@@ -281,7 +281,7 @@ def set_channel_option(self, name, value):
                 self.__channel_options.append(new_option)
         return self
 
-    def construct(self):
+    def construct(self, return_channel=False):
         """Constructs and returns the desired Client object.
         The instance of this class will reset to default values for further building.
 
@@ -347,7 +347,10 @@ def construct(self):
                 ],
             )
         self._reset()
-        return client
+        if return_channel:
+            return client, channel
+        else:
+            return client
 
     def _reset(self):
         """Resets the builder.

src/cisco_gnmi/client.py

@@ -98,6 +98,7 @@ def __init__(self, grpc_channel, timeout=_C_MAX_LONG, default_call_metadata=None
         """
         self.service = proto.gnmi_pb2_grpc.gNMIStub(grpc_channel)
         self.default_call_metadata = default_call_metadata
+        self._channel = grpc_channel
 
     def capabilities(self):
         """Capabilities allows the client to retrieve the set of capabilities that