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,
)