afonso/core-extra
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

updated grpc client to have a proxy flag, updated coretk script to have a flag to enable proxy, disabled by default

Browse source
This commit is contained in:
Blake Harnden 2020-01-14 22:15:00 -08:00
parent 583a552bdb
commit d4bce0d25c
4 changed files with 18 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

5
daemon/core/api/grpc/client.py
View file

@ -137,7 +137,7 @@ class CoreGrpcClient:
Provides convenience methods for interfacing with the CORE grpc server.
"""
def __init__(self, address="localhost:50051"):
def __init__(self, address="localhost:50051", proxy=False):
"""
Creates a CoreGrpcClient instance.
@ -146,6 +146,7 @@ class CoreGrpcClient:
self.address = address
self.stub = None
self.channel = None
self.proxy = proxy
def start_session(
self,
@ -1036,7 +1037,7 @@ class CoreGrpcClient:
:return: nothing
"""
self.channel = grpc.insecure_channel(
self.address, options=[("grpc.enable_http_proxy", 0)]
self.address, options=[("grpc.enable_http_proxy", self.proxy)]
)
self.stub = core_pb2_grpc.CoreApiStub(self.channel)