cursus.mods.exe_doc.registration_helper

Registration Helper for execution document generation.

This module provides the RegistrationHelper class that extracts execution document configurations from registration step configurations.

class RegistrationHelper

Bases: ExecutionDocumentHelper

Helper for extracting execution document configurations from registration steps.

This helper ports the logic from DynamicPipelineTemplate._fill_registration_configurations() and _create_execution_doc_config() methods to generate execution document configurations.

can_handle_step(step_name, config)

Check if this helper can handle the given step configuration.

Parameters:

• step_name (str) – Name of the step

• config – Step configuration object

Returns:

True if this helper can handle the configuration, False otherwise

Return type:

bool

get_execution_step_name(step_name, config)

Get execution document step name following step builder naming convention.

Transforms step names from DAG format to execution document format: - “Registration” -> “Registration-NA” (adds region suffix)

This follows the same logic as RegistrationStepBuilder.create_step(): step_name = self._get_step_name() + “-” + self.config.region

Parameters:

• step_name (str) – Original step name from DAG (e.g., “Registration”)

• config – Configuration object containing region

Returns:

Execution document step name (e.g., “Registration-NA”)

Return type:

str

extract_step_config(step_name, config, all_configs=None)

Extract execution document configuration from registration step config.

Parameters:

• step_name (str) – Name of the step

• config – Registration configuration object

• all_configs (Dict[str, Any] | None) – Optional dictionary of all available configurations for related config lookup

Returns:

Dictionary containing the execution document configuration

Raises:

ExecutionDocumentGenerationError – If configuration extraction fails

Return type:

Dict[str, Any]

create_execution_doc_config_with_related_configs(registration_config, payload_config=None, package_config=None)

Create execution document configuration with related payload and package configs.

This method extends the basic execution document configuration with load testing information from payload and package configurations.

Parameters:

• registration_config – Registration configuration object

• payload_config – Optional payload configuration object

• package_config – Optional package configuration object

Returns:

Dictionary with complete execution document configuration

Return type:

Dict[str, Any]

find_registration_step_patterns(step_names, region='')

Find step name patterns that likely correspond to registration steps.

This method is ported from DynamicPipelineTemplate._fill_registration_configurations().

Parameters:

• step_names (List[str]) – List of step names to search through

• region (str) – Optional region string to create region-specific patterns

Returns:

List of step name patterns for registration steps

Return type:

List[str]

validate_registration_config(config)

Validate that the registration config has all required fields.

Parameters:

config – Registration configuration object

Returns:

True if all required fields are present, False otherwise

Return type:

bool