Как получить значение XCOM в PostgresOperator - PullRequest
Новая
       21

Как получить значение XCOM в PostgresOperator

0 голосов
/ 27 марта 2020

Здесь я нажимаю значение XCOM: 

task_get_username_bash = BashOperator(
                task_id='execute_bash',
                bash_command='whoami',
                xcom_push=True)

Так что в XCOM он хранится как {'return_value': '$ USER'} (в моем случае $ USER = 'airflow').

Затем я хочу получить это return_value из XCOM: 

task_insert_new_row = PostgresOperator(
                task_id='insert_new_row',
                trigger_rule=TriggerRule.ALL_DONE,
                sql='''INSERT INTO table_name VALUES
                (%s, %s, %s);''',
                parameters=(uuid.uuid4().int % 123456789,
                            "{{ ti.xcom_pull(task_ids='execute_bash', key='return_value') }}",
                            datetime.now()))

Но PostgresOperator интерпретирует ссылку на макрос как str. Как вытащить XCOM в PostgresOperator?

1 Ответ

0 голосов
/ 28 марта 2020

Проблема решена: 

task_insert_new_row = PostgresOperator(
                task_id='insert_new_row',
                trigger_rule=TriggerRule.ALL_DONE,
                sql='''INSERT INTO table_name VALUES
                (%s, '{{ ti.xcom_pull(task_ids='execute_bash', key='return_value') }}', %s);''',
                parameters=(uuid.uuid4().int % 123456789, datetime.now()))
...