def ExternalPartitionConfig(self, request, _context):
partition_args = deserialize_json_to_dagster_namedtuple(
request.serialized_partition_args)
check.inst_param(partition_args, 'partition_args', PartitionArgs)
recon_repo = recon_repository_from_origin(
partition_args.repository_origin)
definition = recon_repo.get_definition()
partition_set_def = definition.get_partition_set_def(
partition_args.partition_set_name)
partition = partition_set_def.get_partition(
partition_args.partition_name)
try:
with user_code_error_boundary(
PartitionExecutionError,
lambda:
'Error occurred during the evaluation of the `run_config_for_partition` '
'function for partition set {partition_set_name}'.format(
partition_set_name=partition_set_def.name),
):
run_config = partition_set_def.run_config_for_partition(
partition)
return api_pb2.ExternalPartitionConfigReply(
serialized_external_partition_config_or_external_partition_execution_error
=serialize_dagster_namedtuple(
ExternalPartitionConfigData(name=partition.name,
run_config=run_config)))
except PartitionExecutionError:
return api_pb2.ExternalPartitionConfigReply(
serialized_external_partition_config_or_external_partition_execution_error
=serialize_dagster_namedtuple(
ExternalPartitionExecutionErrorData(
serializable_error_info_from_exc_info(
sys.exc_info()))))
def get_partition_config(recon_repo, partition_set_name, partition_name):
definition = recon_repo.get_definition()
partition_set_def = definition.get_partition_set_def(partition_set_name)
partition = partition_set_def.get_partition(partition_name)
try:
with user_code_error_boundary(
PartitionExecutionError,
lambda:
"Error occurred during the evaluation of the `run_config_for_partition` "
"function for partition set {partition_set_name}".format(
partition_set_name=partition_set_def.name),
):
run_config = partition_set_def.run_config_for_partition(partition)
return ExternalPartitionConfigData(name=partition.name,
run_config=run_config)
except PartitionExecutionError:
return ExternalPartitionExecutionErrorData(
serializable_error_info_from_exc_info(sys.exc_info()))
def get_partition_config(args):
check.inst_param(args, 'args', PartitionArgs)
recon_repo = recon_repository_from_origin(args.repository_origin)
definition = recon_repo.get_definition()
partition_set_def = definition.get_partition_set_def(args.partition_set_name)
partition = partition_set_def.get_partition(args.partition_name)
try:
with user_code_error_boundary(
PartitionExecutionError,
lambda: 'Error occurred during the evaluation of the `run_config_for_partition` '
'function for partition set {partition_set_name}'.format(
partition_set_name=partition_set_def.name
),
):
run_config = partition_set_def.run_config_for_partition(partition)
return ExternalPartitionConfigData(name=partition.name, run_config=run_config)
except PartitionExecutionError:
return ExternalPartitionExecutionErrorData(
serializable_error_info_from_exc_info(sys.exc_info())
)
