Create a Publish Collection DAG

dags/publish_dag.py

@@ -0,0 +1,31 @@
+from datetime import datetime
+
+from airflow.decorators import dag
+from airflow.models.param import Param
+
+from rikolti.dags.shared_tasks.indexing_tasks import (
+    publish_collection_task, get_version_pages)
+from rikolti.dags.shared_tasks.shared import get_registry_data_task
+from rikolti.dags.shared_tasks.shared import notify_dag_success
+from rikolti.dags.shared_tasks.shared import notify_dag_failure
+
+
+@dag(
+    dag_id="publish_collection",
+    schedule=None,
+    start_date=datetime(2023, 1, 1),
+    catchup=False,
+    params={
+        'collection_id': Param(None, description="Collection ID to publish"),
+        'version': Param(None, description="Version path to publish")
+    },
+    tags=["rikolti"],
+    on_failure_callback=notify_dag_failure,
+    on_success_callback=notify_dag_success,
+)
+def publish_collection_dag():
+    collection = get_registry_data_task()
+    version_pages = get_version_pages()
+    publish_collection_task(collection, version_pages)
+
+publish_collection_dag()