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