Xcom pull在`ti`上返回NameError

时间:2018-11-16 06:42:16

标签: python-3.x airflow airflow-scheduler apache-airflow-xcom

我正在尝试使用last_date中的键last_date_task将值推入xcom,然后在第二个任务ga_wh_task中将其从xcom中提取出来。当我通过运行python dag.py测试此dag时,会在此行返回错误,例如:

    provide_context=True, since={{ti.xcom_pull(task_ids="last_date_task", key='last_date')}})
NameError: name 'ti' is not defined

我在这里做错了什么?我尝试使用task_instance代替ticontext['ti']代替kwargs['ti']

这是dag文件:

default_args = {
    'owner': 'me',
    'start_date': dt.datetime(2017, 10, 30),
    'retries': 1,
    'retry_delay': dt.timedelta(minutes=10),
    'provide_context' : True,
}



def get_last_date(**kwargs):
    kwargs['ti'].xcom_push(key='last_date', value='2018-11-15')
    return True



with DAG('ga_mysql_dag2',
         default_args=default_args,
         schedule_interval=None,
         catchup=False,
         ) as dag:

    last_date_task = PythonOperator(task_id='last_date_task', python_callable=get_last_date, provide_context=True)

    ga_wh_task = GoogleAnalyticsReportingToMySqlOperator(task_id='ga_wh_task', google_analytics_conn_id='google_analytics', key_file=key_file,\
                                        view_id=view_id, until=until, dimensions=dimensions, metrics=metrics, database=database,\
                                        table = table, mysql_conn_id = mysql_conn_id,
                                        provide_context=True, since={{ti.xcom_pull(task_ids="last_date_task", key='last_date')}})
    sleep = BashOperator(task_id='sleep', bash_command='sleep 10')

# Dependencies
last_date_task >> ga_warehouse_task >> sleep

1 个答案:

答案 0 :(得分:0)

看起来GoogleAnalyticsReportingToMySqlOperator是您创建的运算符。

since参数应该是字符串。因此将其更改为since="{{ti.xcom_pull(task_ids='last_date_task', key='last_date')}}"