This is an automated email from the ASF dual-hosted git repository.
potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new b95ea2093c Use params instead of dag_run.conf in example (#40759)
b95ea2093c is described below
commit b95ea2093c6f9bed53ee00fa88c948f1804dfd67
Author: Jens Scheffler <[email protected]>
AuthorDate: Sat Jul 13 07:33:55 2024 +0200
Use params instead of dag_run.conf in example (#40759)
---
airflow/example_dags/example_params_trigger_ui.py | 19 ++++++-------------
1 file changed, 6 insertions(+), 13 deletions(-)
diff --git a/airflow/example_dags/example_params_trigger_ui.py
b/airflow/example_dags/example_params_trigger_ui.py
index 47465ad39d..d95a78df7f 100644
--- a/airflow/example_dags/example_params_trigger_ui.py
+++ b/airflow/example_dags/example_params_trigger_ui.py
@@ -24,17 +24,12 @@ from __future__ import annotations
import datetime
from pathlib import Path
-from typing import TYPE_CHECKING
from airflow.decorators import task
from airflow.models.dag import DAG
-from airflow.models.param import Param
+from airflow.models.param import Param, ParamsDict
from airflow.utils.trigger_rule import TriggerRule
-if TYPE_CHECKING:
- from airflow.models.dagrun import DagRun
- from airflow.models.taskinstance import TaskInstance
-
with DAG(
dag_id=Path(__file__).stem,
dag_display_name="Params Trigger UI",
@@ -60,20 +55,18 @@ with DAG(
@task(task_id="get_names", task_display_name="Get names")
def get_names(**kwargs) -> list[str]:
- ti: TaskInstance = kwargs["ti"]
- dag_run: DagRun = ti.dag_run
- if "names" not in dag_run.conf:
+ params: ParamsDict = kwargs["params"]
+ if "names" not in params:
print("Uuups, no names given, was no UI used to trigger?")
return []
- return dag_run.conf["names"]
+ return params["names"]
@task.branch(task_id="select_languages", task_display_name="Select
languages")
def select_languages(**kwargs) -> list[str]:
- ti: TaskInstance = kwargs["ti"]
- dag_run: DagRun = ti.dag_run
+ params: ParamsDict = kwargs["params"]
selected_languages = []
for lang in ["english", "german", "french"]:
- if dag_run.conf.get(lang):
+ if params[lang]:
selected_languages.append(f"generate_{lang}_greeting")
return selected_languages
