# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! """Client and server classes corresponding to protobuf-defined services.""" import grpc import warnings from . import dispatcher_pb2 as dispatcher__pb2 GRPC_GENERATED_VERSION = '1.64.1' GRPC_VERSION = grpc.__version__ EXPECTED_ERROR_RELEASE = '1.65.0' SCHEDULED_RELEASE_DATE = 'June 25, 2024' _version_not_supported = False try: from grpc._utilities import first_version_is_lower _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION) except ImportError: _version_not_supported = True if _version_not_supported: warnings.warn( f'The grpc package installed is at version {GRPC_VERSION},' + f' but the generated code in dispatcher_pb2_grpc.py depends on' + f' grpcio>={GRPC_GENERATED_VERSION}.' + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}' + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.' + f' This warning will become an error in {EXPECTED_ERROR_RELEASE},' + f' scheduled for release on {SCHEDULED_RELEASE_DATE}.', RuntimeWarning ) class DispatcherStub(object): """Missing associated documentation comment in .proto file.""" def __init__(self, channel): """Constructor. Args: channel: A grpc.Channel. """ self.Register = channel.unary_unary( '/Dispatcher/Register', request_serializer=dispatcher__pb2.WorkerRegisterRequest.SerializeToString, response_deserializer=dispatcher__pb2.WorkerRegisterResponse.FromString, _registered_method=True) self.Listen = channel.unary_stream( '/Dispatcher/Listen', request_serializer=dispatcher__pb2.WorkerListenRequest.SerializeToString, response_deserializer=dispatcher__pb2.AssignedAction.FromString, _registered_method=True) self.ListenV2 = channel.unary_stream( '/Dispatcher/ListenV2', request_serializer=dispatcher__pb2.WorkerListenRequest.SerializeToString, response_deserializer=dispatcher__pb2.AssignedAction.FromString, _registered_method=True) self.Heartbeat = channel.unary_unary( '/Dispatcher/Heartbeat', request_serializer=dispatcher__pb2.HeartbeatRequest.SerializeToString, response_deserializer=dispatcher__pb2.HeartbeatResponse.FromString, _registered_method=True) self.SubscribeToWorkflowEvents = channel.unary_stream( '/Dispatcher/SubscribeToWorkflowEvents', request_serializer=dispatcher__pb2.SubscribeToWorkflowEventsRequest.SerializeToString, response_deserializer=dispatcher__pb2.WorkflowEvent.FromString, _registered_method=True) self.SubscribeToWorkflowRuns = channel.stream_stream( '/Dispatcher/SubscribeToWorkflowRuns', request_serializer=dispatcher__pb2.SubscribeToWorkflowRunsRequest.SerializeToString, response_deserializer=dispatcher__pb2.WorkflowRunEvent.FromString, _registered_method=True) self.SendStepActionEvent = channel.unary_unary( '/Dispatcher/SendStepActionEvent', request_serializer=dispatcher__pb2.StepActionEvent.SerializeToString, response_deserializer=dispatcher__pb2.ActionEventResponse.FromString, _registered_method=True) self.SendGroupKeyActionEvent = channel.unary_unary( '/Dispatcher/SendGroupKeyActionEvent', request_serializer=dispatcher__pb2.GroupKeyActionEvent.SerializeToString, response_deserializer=dispatcher__pb2.ActionEventResponse.FromString, _registered_method=True) self.PutOverridesData = channel.unary_unary( '/Dispatcher/PutOverridesData', request_serializer=dispatcher__pb2.OverridesData.SerializeToString, response_deserializer=dispatcher__pb2.OverridesDataResponse.FromString, _registered_method=True) self.Unsubscribe = channel.unary_unary( '/Dispatcher/Unsubscribe', request_serializer=dispatcher__pb2.WorkerUnsubscribeRequest.SerializeToString, response_deserializer=dispatcher__pb2.WorkerUnsubscribeResponse.FromString, _registered_method=True) self.RefreshTimeout = channel.unary_unary( '/Dispatcher/RefreshTimeout', request_serializer=dispatcher__pb2.RefreshTimeoutRequest.SerializeToString, response_deserializer=dispatcher__pb2.RefreshTimeoutResponse.FromString, _registered_method=True) self.ReleaseSlot = channel.unary_unary( '/Dispatcher/ReleaseSlot', request_serializer=dispatcher__pb2.ReleaseSlotRequest.SerializeToString, response_deserializer=dispatcher__pb2.ReleaseSlotResponse.FromString, _registered_method=True) self.UpsertWorkerLabels = channel.unary_unary( '/Dispatcher/UpsertWorkerLabels', request_serializer=dispatcher__pb2.UpsertWorkerLabelsRequest.SerializeToString, response_deserializer=dispatcher__pb2.UpsertWorkerLabelsResponse.FromString, _registered_method=True) class DispatcherServicer(object): """Missing associated documentation comment in .proto file.""" def Register(self, request, context): """Missing associated documentation comment in .proto file.""" context.set_code(grpc.StatusCode.UNIMPLEMENTED) context.set_details('Method not implemented!') raise NotImplementedError('Method not implemented!') def Listen(self, request, context): """Missing associated documentation comment in .proto file.""" context.set_code(grpc.StatusCode.UNIMPLEMENTED) context.set_details('Method not implemented!') raise NotImplementedError('Method not implemented!') def ListenV2(self, request, context): """ListenV2 is like listen, but implementation does not include heartbeats. This should only used by SDKs against engine version v0.18.1+ """ context.set_code(grpc.StatusCode.UNIMPLEMENTED) context.set_details('Method not implemented!') raise NotImplementedError('Method not implemented!') def Heartbeat(self, request, context): """Heartbeat is a method for workers to send heartbeats to the dispatcher """ context.set_code(grpc.StatusCode.UNIMPLEMENTED) context.set_details('Method not implemented!') raise NotImplementedError('Method not implemented!') def SubscribeToWorkflowEvents(self, request, context): """Missing associated documentation comment in .proto file.""" context.set_code(grpc.StatusCode.UNIMPLEMENTED) context.set_details('Method not implemented!') raise NotImplementedError('Method not implemented!') def SubscribeToWorkflowRuns(self, request_iterator, context): """Missing associated documentation comment in .proto file.""" context.set_code(grpc.StatusCode.UNIMPLEMENTED) context.set_details('Method not implemented!') raise NotImplementedError('Method not implemented!') def SendStepActionEvent(self, request, context): """Missing associated documentation comment in .proto file.""" context.set_code(grpc.StatusCode.UNIMPLEMENTED) context.set_details('Method not implemented!') raise NotImplementedError('Method not implemented!') def SendGroupKeyActionEvent(self, request, context): """Missing associated documentation comment in .proto file.""" context.set_code(grpc.StatusCode.UNIMPLEMENTED) context.set_details('Method not implemented!') raise NotImplementedError('Method not implemented!') def PutOverridesData(self, request, context): """Missing associated documentation comment in .proto file.""" context.set_code(grpc.StatusCode.UNIMPLEMENTED) context.set_details('Method not implemented!') raise NotImplementedError('Method not implemented!') def Unsubscribe(self, request, context): """Missing associated documentation comment in .proto file.""" context.set_code(grpc.StatusCode.UNIMPLEMENTED) context.set_details('Method not implemented!') raise NotImplementedError('Method not implemented!') def RefreshTimeout(self, request, context): """Missing associated documentation comment in .proto file.""" context.set_code(grpc.StatusCode.UNIMPLEMENTED) context.set_details('Method not implemented!') raise NotImplementedError('Method not implemented!') def ReleaseSlot(self, request, context): """Missing associated documentation comment in .proto file.""" context.set_code(grpc.StatusCode.UNIMPLEMENTED) context.set_details('Method not implemented!') raise NotImplementedError('Method not implemented!') def UpsertWorkerLabels(self, request, context): """Missing associated documentation comment in .proto file.""" context.set_code(grpc.StatusCode.UNIMPLEMENTED) context.set_details('Method not implemented!') raise NotImplementedError('Method not implemented!') def add_DispatcherServicer_to_server(servicer, server): rpc_method_handlers = { 'Register': grpc.unary_unary_rpc_method_handler( servicer.Register, request_deserializer=dispatcher__pb2.WorkerRegisterRequest.FromString, response_serializer=dispatcher__pb2.WorkerRegisterResponse.SerializeToString, ), 'Listen': grpc.unary_stream_rpc_method_handler( servicer.Listen, request_deserializer=dispatcher__pb2.WorkerListenRequest.FromString, response_serializer=dispatcher__pb2.AssignedAction.SerializeToString, ), 'ListenV2': grpc.unary_stream_rpc_method_handler( servicer.ListenV2, request_deserializer=dispatcher__pb2.WorkerListenRequest.FromString, response_serializer=dispatcher__pb2.AssignedAction.SerializeToString, ), 'Heartbeat': grpc.unary_unary_rpc_method_handler( servicer.Heartbeat, request_deserializer=dispatcher__pb2.HeartbeatRequest.FromString, response_serializer=dispatcher__pb2.HeartbeatResponse.SerializeToString, ), 'SubscribeToWorkflowEvents': grpc.unary_stream_rpc_method_handler( servicer.SubscribeToWorkflowEvents, request_deserializer=dispatcher__pb2.SubscribeToWorkflowEventsRequest.FromString, response_serializer=dispatcher__pb2.WorkflowEvent.SerializeToString, ), 'SubscribeToWorkflowRuns': grpc.stream_stream_rpc_method_handler( servicer.SubscribeToWorkflowRuns, request_deserializer=dispatcher__pb2.SubscribeToWorkflowRunsRequest.FromString, response_serializer=dispatcher__pb2.WorkflowRunEvent.SerializeToString, ), 'SendStepActionEvent': grpc.unary_unary_rpc_method_handler( servicer.SendStepActionEvent, request_deserializer=dispatcher__pb2.StepActionEvent.FromString, response_serializer=dispatcher__pb2.ActionEventResponse.SerializeToString, ), 'SendGroupKeyActionEvent': grpc.unary_unary_rpc_method_handler( servicer.SendGroupKeyActionEvent, request_deserializer=dispatcher__pb2.GroupKeyActionEvent.FromString, response_serializer=dispatcher__pb2.ActionEventResponse.SerializeToString, ), 'PutOverridesData': grpc.unary_unary_rpc_method_handler( servicer.PutOverridesData, request_deserializer=dispatcher__pb2.OverridesData.FromString, response_serializer=dispatcher__pb2.OverridesDataResponse.SerializeToString, ), 'Unsubscribe': grpc.unary_unary_rpc_method_handler( servicer.Unsubscribe, request_deserializer=dispatcher__pb2.WorkerUnsubscribeRequest.FromString, response_serializer=dispatcher__pb2.WorkerUnsubscribeResponse.SerializeToString, ), 'RefreshTimeout': grpc.unary_unary_rpc_method_handler( servicer.RefreshTimeout, request_deserializer=dispatcher__pb2.RefreshTimeoutRequest.FromString, response_serializer=dispatcher__pb2.RefreshTimeoutResponse.SerializeToString, ), 'ReleaseSlot': grpc.unary_unary_rpc_method_handler( servicer.ReleaseSlot, request_deserializer=dispatcher__pb2.ReleaseSlotRequest.FromString, response_serializer=dispatcher__pb2.ReleaseSlotResponse.SerializeToString, ), 'UpsertWorkerLabels': grpc.unary_unary_rpc_method_handler( servicer.UpsertWorkerLabels, request_deserializer=dispatcher__pb2.UpsertWorkerLabelsRequest.FromString, response_serializer=dispatcher__pb2.UpsertWorkerLabelsResponse.SerializeToString, ), } generic_handler = grpc.method_handlers_generic_handler( 'Dispatcher', rpc_method_handlers) server.add_generic_rpc_handlers((generic_handler,)) server.add_registered_method_handlers('Dispatcher', rpc_method_handlers) # This class is part of an EXPERIMENTAL API. class Dispatcher(object): """Missing associated documentation comment in .proto file.""" @staticmethod def Register(request, target, options=(), channel_credentials=None, call_credentials=None, insecure=False, compression=None, wait_for_ready=None, timeout=None, metadata=None): return grpc.experimental.unary_unary( request, target, '/Dispatcher/Register', dispatcher__pb2.WorkerRegisterRequest.SerializeToString, dispatcher__pb2.WorkerRegisterResponse.FromString, options, channel_credentials, insecure, call_credentials, compression, wait_for_ready, timeout, metadata, _registered_method=True) @staticmethod def Listen(request, target, options=(), channel_credentials=None, call_credentials=None, insecure=False, compression=None, wait_for_ready=None, timeout=None, metadata=None): return grpc.experimental.unary_stream( request, target, '/Dispatcher/Listen', dispatcher__pb2.WorkerListenRequest.SerializeToString, dispatcher__pb2.AssignedAction.FromString, options, channel_credentials, insecure, call_credentials, compression, wait_for_ready, timeout, metadata, _registered_method=True) @staticmethod def ListenV2(request, target, options=(), channel_credentials=None, call_credentials=None, insecure=False, compression=None, wait_for_ready=None, timeout=None, metadata=None): return grpc.experimental.unary_stream( request, target, '/Dispatcher/ListenV2', dispatcher__pb2.WorkerListenRequest.SerializeToString, dispatcher__pb2.AssignedAction.FromString, options, channel_credentials, insecure, call_credentials, compression, wait_for_ready, timeout, metadata, _registered_method=True) @staticmethod def Heartbeat(request, target, options=(), channel_credentials=None, call_credentials=None, insecure=False, compression=None, wait_for_ready=None, timeout=None, metadata=None): return grpc.experimental.unary_unary( request, target, '/Dispatcher/Heartbeat', dispatcher__pb2.HeartbeatRequest.SerializeToString, dispatcher__pb2.HeartbeatResponse.FromString, options, channel_credentials, insecure, call_credentials, compression, wait_for_ready, timeout, metadata, _registered_method=True) @staticmethod def SubscribeToWorkflowEvents(request, target, options=(), channel_credentials=None, call_credentials=None, insecure=False, compression=None, wait_for_ready=None, timeout=None, metadata=None): return grpc.experimental.unary_stream( request, target, '/Dispatcher/SubscribeToWorkflowEvents', dispatcher__pb2.SubscribeToWorkflowEventsRequest.SerializeToString, dispatcher__pb2.WorkflowEvent.FromString, options, channel_credentials, insecure, call_credentials, compression, wait_for_ready, timeout, metadata, _registered_method=True) @staticmethod def SubscribeToWorkflowRuns(request_iterator, target, options=(), channel_credentials=None, call_credentials=None, insecure=False, compression=None, wait_for_ready=None, timeout=None, metadata=None): return grpc.experimental.stream_stream( request_iterator, target, '/Dispatcher/SubscribeToWorkflowRuns', dispatcher__pb2.SubscribeToWorkflowRunsRequest.SerializeToString, dispatcher__pb2.WorkflowRunEvent.FromString, options, channel_credentials, insecure, call_credentials, compression, wait_for_ready, timeout, metadata, _registered_method=True) @staticmethod def SendStepActionEvent(request, target, options=(), channel_credentials=None, call_credentials=None, insecure=False, compression=None, wait_for_ready=None, timeout=None, metadata=None): return grpc.experimental.unary_unary( request, target, '/Dispatcher/SendStepActionEvent', dispatcher__pb2.StepActionEvent.SerializeToString, dispatcher__pb2.ActionEventResponse.FromString, options, channel_credentials, insecure, call_credentials, compression, wait_for_ready, timeout, metadata, _registered_method=True) @staticmethod def SendGroupKeyActionEvent(request, target, options=(), channel_credentials=None, call_credentials=None, insecure=False, compression=None, wait_for_ready=None, timeout=None, metadata=None): return grpc.experimental.unary_unary( request, target, '/Dispatcher/SendGroupKeyActionEvent', dispatcher__pb2.GroupKeyActionEvent.SerializeToString, dispatcher__pb2.ActionEventResponse.FromString, options, channel_credentials, insecure, call_credentials, compression, wait_for_ready, timeout, metadata, _registered_method=True) @staticmethod def PutOverridesData(request, target, options=(), channel_credentials=None, call_credentials=None, insecure=False, compression=None, wait_for_ready=None, timeout=None, metadata=None): return grpc.experimental.unary_unary( request, target, '/Dispatcher/PutOverridesData', dispatcher__pb2.OverridesData.SerializeToString, dispatcher__pb2.OverridesDataResponse.FromString, options, channel_credentials, insecure, call_credentials, compression, wait_for_ready, timeout, metadata, _registered_method=True) @staticmethod def Unsubscribe(request, target, options=(), channel_credentials=None, call_credentials=None, insecure=False, compression=None, wait_for_ready=None, timeout=None, metadata=None): return grpc.experimental.unary_unary( request, target, '/Dispatcher/Unsubscribe', dispatcher__pb2.WorkerUnsubscribeRequest.SerializeToString, dispatcher__pb2.WorkerUnsubscribeResponse.FromString, options, channel_credentials, insecure, call_credentials, compression, wait_for_ready, timeout, metadata, _registered_method=True) @staticmethod def RefreshTimeout(request, target, options=(), channel_credentials=None, call_credentials=None, insecure=False, compression=None, wait_for_ready=None, timeout=None, metadata=None): return grpc.experimental.unary_unary( request, target, '/Dispatcher/RefreshTimeout', dispatcher__pb2.RefreshTimeoutRequest.SerializeToString, dispatcher__pb2.RefreshTimeoutResponse.FromString, options, channel_credentials, insecure, call_credentials, compression, wait_for_ready, timeout, metadata, _registered_method=True) @staticmethod def ReleaseSlot(request, target, options=(), channel_credentials=None, call_credentials=None, insecure=False, compression=None, wait_for_ready=None, timeout=None, metadata=None): return grpc.experimental.unary_unary( request, target, '/Dispatcher/ReleaseSlot', dispatcher__pb2.ReleaseSlotRequest.SerializeToString, dispatcher__pb2.ReleaseSlotResponse.FromString, options, channel_credentials, insecure, call_credentials, compression, wait_for_ready, timeout, metadata, _registered_method=True) @staticmethod def UpsertWorkerLabels(request, target, options=(), channel_credentials=None, call_credentials=None, insecure=False, compression=None, wait_for_ready=None, timeout=None, metadata=None): return grpc.experimental.unary_unary( request, target, '/Dispatcher/UpsertWorkerLabels', dispatcher__pb2.UpsertWorkerLabelsRequest.SerializeToString, dispatcher__pb2.UpsertWorkerLabelsResponse.FromString, options, channel_credentials, insecure, call_credentials, compression, wait_for_ready, timeout, metadata, _registered_method=True)