Xcom pull returns a NameError on `ti`

Viewed 3564

I'm trying to push into xcom a value with key last_date in last_date_task and then pull it from xcom in second task ga_wh_task. when I test this dag by running python dag.py it returns an error at this line like:

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

What am I doing wrong here? I've tried using task_instance instead of ti and context['ti'] instead of kwargs['ti']

Here's the dag file:

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 Answers

Looks like GoogleAnalyticsReportingToMySqlOperator is an operator you created.

since param should be string. So change it to since="{{ti.xcom_pull(task_ids='last_date_task', key='last_date')}}"

Related