updates to grpc event streaming, client can now listen to a subset of desired events

This commit is contained in:
Blake Harnden 2019-12-13 16:11:58 -08:00
parent 47e087b365
commit 47cc20b567
4 changed files with 293 additions and 211 deletions

View file

@ -378,16 +378,17 @@ class CoreGrpcClient:
)
return self.stub.AddSessionServer(request)
def events(self, session_id, handler):
def events(self, session_id, handler, events=None):
"""
Listen for session events.
:param int session_id: id of session
:param handler: handler for every event
:param handler: handler for received events
:param list events: events to listen to, defaults to all
:return: nothing
:raises grpc.RpcError: when session doesn't exist
"""
request = core_pb2.EventsRequest(session_id=session_id)
request = core_pb2.EventsRequest(session_id=session_id, events=events)
stream = self.stub.Events(request)
start_streamer(stream, handler)