fix insert-elk-sirene dag

annuaire-entreprises-data-gouv-fr · Nov 22, 2023 · 3c1ea79 · 3c1ea79
1 parent 23b171c
commit 3c1ea79
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/DAG-insert-elk-sirene.py b/DAG-insert-elk-sirene.py
@@ -126,4 +126,5 @@
         python_callable=fill_elastic_siren_index,
     )
 
-    fill_elastic_siren_index.set_upstream(get_colors)
+    create_elastic_index.set_upstream(get_colors)
+    fill_elastic_siren_index.set_upstream(create_elastic_index)
diff --git a/task_functions/get_colors.py b/task_functions/get_colors.py
@@ -6,8 +6,8 @@
 
 
 def get_colors(**kwargs):
-    kwargs["ti"].xcom_push(key="next_color", value="20231122")
-    kwargs["ti"].xcom_push(key="current_color", value="green")
+    kwargs["ti"].xcom_push(key="next_color", value="20231123")
+    kwargs["ti"].xcom_push(key="current_color", value="20231122")
 
     # try:
     #     with urlopen(COLOR_URL, timeout=5) as url: