def sync_get_external_schedule_execution_data_grpc( api_client: "DagsterGrpcClient", instance: DagsterInstance, repository_handle: RepositoryHandle, schedule_name: str, scheduled_execution_time: Any, ) -> ScheduleExecutionData: check.inst_param(repository_handle, "repository_handle", RepositoryHandle) check.str_param(schedule_name, "schedule_name") check.opt_inst_param(scheduled_execution_time, "scheduled_execution_time", PendulumDateTime) origin = repository_handle.get_external_origin() result = deserialize_as( api_client.external_schedule_execution( external_schedule_execution_args=ExternalScheduleExecutionArgs( repository_origin=origin, instance_ref=instance.get_ref(), schedule_name=schedule_name, scheduled_execution_timestamp=scheduled_execution_time. timestamp() if scheduled_execution_time else None, scheduled_execution_timezone=scheduled_execution_time.timezone. name if scheduled_execution_time else None, )), (ScheduleExecutionData, ExternalScheduleExecutionErrorData), ) if isinstance(result, ExternalScheduleExecutionErrorData): raise DagsterUserCodeProcessError.from_error_info(result.error) return result
def sync_get_external_partition_set_execution_param_data_grpc( api_client: "DagsterGrpcClient", repository_handle: RepositoryHandle, partition_set_name: str, partition_names: List[str], ) -> ExternalPartitionSetExecutionParamData: from dagster.grpc.client import DagsterGrpcClient check.inst_param(api_client, "api_client", DagsterGrpcClient) check.inst_param(repository_handle, "repository_handle", RepositoryHandle) check.str_param(partition_set_name, "partition_set_name") check.list_param(partition_names, "partition_names", of_type=str) repository_origin = repository_handle.get_external_origin() result = deserialize_as( api_client.external_partition_set_execution_params( partition_set_execution_param_args=PartitionSetExecutionParamArgs( repository_origin=repository_origin, partition_set_name=partition_set_name, partition_names=partition_names, ), ), (ExternalPartitionSetExecutionParamData, ExternalPartitionExecutionErrorData), ) if isinstance(result, ExternalPartitionExecutionErrorData): raise DagsterUserCodeProcessError.from_error_info(result.error) return result
def sync_get_external_sensor_execution_data_grpc( api_client: "DagsterGrpcClient", instance: "DagsterInstance", repository_handle: RepositoryHandle, sensor_name: str, last_completion_time: Optional[float], last_run_key: Optional[str], cursor: Optional[str], ) -> SensorExecutionData: check.inst_param(repository_handle, "repository_handle", RepositoryHandle) check.str_param(sensor_name, "sensor_name") check.opt_float_param(last_completion_time, "last_completion_time") check.opt_str_param(last_run_key, "last_run_key") check.opt_str_param(cursor, "cursor") origin = repository_handle.get_external_origin() result = deserialize_as( api_client.external_sensor_execution( sensor_execution_args=SensorExecutionArgs( repository_origin=origin, instance_ref=instance.get_ref(), sensor_name=sensor_name, last_completion_time=last_completion_time, last_run_key=last_run_key, cursor=cursor, ) ), (SensorExecutionData, ExternalSensorExecutionErrorData), ) if isinstance(result, ExternalSensorExecutionErrorData): raise DagsterUserCodeProcessError.from_error_info(result.error) return result
def sync_get_external_schedule_execution_data_ephemeral_grpc( instance: DagsterInstance, repository_handle: RepositoryHandle, schedule_name: str, scheduled_execution_time: Any, ): from dagster.grpc.client import ephemeral_grpc_api_client origin = repository_handle.get_external_origin() with ephemeral_grpc_api_client(origin.repository_location_origin. loadable_target_origin) as api_client: return sync_get_external_schedule_execution_data_grpc( api_client, instance, repository_handle, schedule_name, scheduled_execution_time, )