src/datamigration/dags/redshift/redshift_transfer_run_log_dag.py [54:86]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
def _load_parameters(ti, **kwargs) -> None:
    """
    Load necessary variables for downstream tasks
    """
    if kwargs["dag_run"].conf is not None:
        full_run_id = ast.literal_eval(kwargs["dag_run"].conf["config"])[
            "TRANSFER_RUN_ID"
        ]
        run_state = ast.literal_eval(kwargs["dag_run"].conf["config"])[
            "TRANSFER_RUN_STATE"
        ]

    else:
        logging.error(
            "Method _load_parameters: full_run_id not provided in DAG conf.\n \
            Please provide full_run_id in form 'projects/{projectId}/transferConfigs/{configId}/runs/{run_id}'\n \
            or 'projects/{projectId}/locations/{locationId}/transferConfigs/{configId}/runs/{run_id}'"
        )
        raise Exception(
            "Method _load_parameters: full_run_id not provided in DAG conf.\n \
            Please provide full_run_id in form 'projects/{projectId}/transferConfigs/{configId}/runs/{run_id}'\n \
            or 'projects/{projectId}/locations/{locationId}/transferConfigs/{configId}/runs/{run_id}'"
        )

    (
        _,
        _,
        transfer_config_id,
        transfer_run_id,
    ) = dts_logs_utils.parse_full_transfer_runID(full_run_id)
    tracking_info = dts_logs_utils.get_tracking_info(
        transfer_config_id, BQ_TRANSFER_TRACKING_TABLE_NAME
    )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/datamigration/dags/teradata/teradata_transfer_run_log_dag.py [64:96]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
def _load_parameters(ti, **kwargs) -> None:
    """
    Load necessary variables for downstream tasks
    """
    if kwargs["dag_run"].conf is not None:
        full_run_id = ast.literal_eval(kwargs["dag_run"].conf["config"])[
            "TRANSFER_RUN_ID"
        ]
        run_state = ast.literal_eval(kwargs["dag_run"].conf["config"])[
            "TRANSFER_RUN_STATE"
        ]

    else:
        logging.error(
            "Method _load_parameters: full_run_id not provided in DAG conf.\n \
            Please provide full_run_id in form 'projects/{projectId}/transferConfigs/{configId}/runs/{run_id}'\n \
            or 'projects/{projectId}/locations/{locationId}/transferConfigs/{configId}/runs/{run_id}'"
        )
        raise Exception(
            "Method _load_parameters: full_run_id not provided in DAG conf.\n \
            Please provide full_run_id in form 'projects/{projectId}/transferConfigs/{configId}/runs/{run_id}'\n \
            or 'projects/{projectId}/locations/{locationId}/transferConfigs/{configId}/runs/{run_id}'"
        )

    (
        _,
        _,
        transfer_config_id,
        transfer_run_id,
    ) = dts_logs_utils.parse_full_transfer_runID(full_run_id)
    tracking_info = dts_logs_utils.get_tracking_info(
        transfer_config_id, BQ_TRANSFER_TRACKING_TABLE_NAME
    )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -