From bd14142f1f85d11f1f92c630c06edfb8fd359b6f Mon Sep 17 00:00:00 2001 From: Michael Kotliar Date: Wed, 4 Sep 2019 18:47:40 -0400 Subject: [PATCH] Fix bug in trigerring dag if it wasn't read yet --- cwl_airflow/wes/backend.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/cwl_airflow/wes/backend.py b/cwl_airflow/wes/backend.py index 0dedd1f..54035b0 100644 --- a/cwl_airflow/wes/backend.py +++ b/cwl_airflow/wes/backend.py @@ -120,7 +120,20 @@ def post_dag_runs_legacy(self, dag_id): def trigger_dag(self, dag_id, run_id, conf): - return trigger_dag.trigger_dag(dag_id=dag_id, run_id=run_id, conf=conf, replace_microseconds=False) + try: + dag_path = DagModel.get_current(dag_id).fileloc + except Exception: + dag_path = path.join(DAGS_FOLDER, dag_id + ".py") + triggers = trigger_dag._trigger_dag( + dag_id=dag_id, + dag_run=DagRun(), + dag_bag=DagBag(dag_folder=dag_path), + run_id=run_id, + conf=conf, + execution_date=None, + replace_microseconds=False + ) + return triggers[0] if triggers else None def list_dags(self):