Advertisement
Guest User

Untitled

a guest
Jul 19th, 2019
80
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.81 KB | None | 0 0
  1. args = {
  2. 'owner': 'me',
  3. 'start_date': airflow.utils.dates.days_ago(2),
  4. 'provide_context': True
  5. }
  6.  
  7. dag = DAG('test_timeout',
  8. schedule_interval=None,
  9. default_args=args,
  10. dagrun_timeout=timedelta(seconds=20))
  11.  
  12. def this_passes(**kwargs):
  13. return
  14.  
  15. def this_passes_with_delay(**kwargs):
  16. time.sleep(120)
  17. return
  18.  
  19. would_succeed = PythonOperator(task_id='would_succeed',
  20. dag=dag,
  21. python_callable=this_passes,
  22. email=to)
  23.  
  24. would_succeed_with_delay = PythonOperator(task_id='would_succeed_with_delay',
  25. dag=dag,
  26. python_callable=this_passes_with_delay,
  27. email=to)
  28.  
  29. would_succeed >> would_succeed_with_delay
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement