fix xcom
This commit is contained in:
@@ -12,7 +12,7 @@ def extract_draft_data():
|
||||
return drafts.download_draft_json(marcel_vs_canuck)
|
||||
|
||||
def transform_draft(ti):
|
||||
draft_json=ti.xcom_pull(task_ids=['extract_draft1'])
|
||||
draft_json=ti.xcom_pull(task_ids='extract_draft1', key='return_value')
|
||||
if not draft_json:
|
||||
raise ValueError("No value stored in XComs")
|
||||
draft_picks=drafts.get_picked_maps(draft_json)
|
||||
@@ -32,13 +32,16 @@ with DAG("aoe_dag", start_date=datetime(2023, 1, 1), schedule_interval="@daily",
|
||||
#T
|
||||
transform_data = PythonOperator(
|
||||
task_id="transform_draft1",
|
||||
python_callable=transform_draft
|
||||
python_callable=transform_draft,
|
||||
do_xcom_push=True
|
||||
)
|
||||
|
||||
#L
|
||||
string_to_print = "{{ task_instance.xcom_pull(task_ids='transform_draft1', key='return_value') }}"
|
||||
print_output = BashOperator(
|
||||
task_id="print1",
|
||||
bash_command="echo 'bla'"
|
||||
bash_command="echo ${STRING_TO_PRINT}",
|
||||
env={"STRING_TO_PRINT": string_to_print}
|
||||
)
|
||||
|
||||
extract_data >> transform_data >> print_output
|
Reference in New Issue
Block a user