src/datamigration/dags/redshift/redshift_data_load_dag.py [283:315]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
with models.DAG(
    DAG_ID,
    schedule=None,
    default_args=default_dag_args,
    render_template_as_native_obj=True,
) as dag:
    get_unique_id_from_config = PythonOperator(
        task_id=GET_CONFIG_TASK,
        python_callable=_get_config,
        dag=dag,
    )

    get_valid_tables = PythonOperator(
        task_id=GET_TABLE_LIST_TASK,
        python_callable=_get_table_list,
        dag=dag,
    )

    generate_batches = PythonOperator(
        task_id=GENERATE_BATCHES_DAG_TASK,
        python_callable=_generate_batches,
        dag=dag,
    )

    create_bq_transfer_config_json = PythonOperator.partial(
        task_id="create_bq_transfer_config_json",
        python_callable=_create_bq_transfer_config_json,
        dag=dag,
    ).expand(op_kwargs=XComArg(generate_batches, key="return_value"))

    create_bq_transfer = PythonOperator.partial(
        task_id="create_bq_transfer", python_callable=_create_bq_transfer, dag=dag
    ).expand(op_kwargs=XComArg(generate_batches, key="return_value"))
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/datamigration/dags/teradata/teradata_data_load_dag.py [369:401]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
with models.DAG(
    DAG_ID,
    schedule=None,
    default_args=default_dag_args,
    render_template_as_native_obj=True,
) as dag:
    get_unique_id_from_config = PythonOperator(
        task_id=GET_CONFIG_TASK,
        python_callable=_get_config,
        dag=dag,
    )

    get_valid_tables = PythonOperator(
        task_id=GET_TABLE_LIST_TASK,
        python_callable=_get_table_list,
        dag=dag,
    )

    generate_batches = PythonOperator(
        task_id=GENERATE_BATCHES_DAG_TASK,
        python_callable=_generate_batches,
        dag=dag,
    )

    create_bq_transfer_config_json = PythonOperator.partial(
        task_id="create_bq_transfer_config_json",
        python_callable=_create_bq_transfer_config_json,
        dag=dag,
    ).expand(op_kwargs=XComArg(generate_batches, key="return_value"))

    create_bq_transfer = PythonOperator.partial(
        task_id="create_bq_transfer", python_callable=_create_bq_transfer, dag=dag
    ).expand(op_kwargs=XComArg(generate_batches, key="return_value"))
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -