Connections: DB connection update

cern-sis · Jul 24, 2024 · afbcd7b · afbcd7b
1 parent 33bd15e
commit afbcd7b
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/dags/library/cern_publication_records.py b/dags/library/cern_publication_records.py
@@ -45,7 +45,7 @@ def join(values, **kwargs):
 
     PostgresOperator(
         task_id="populate_library_cern_publication_records_table",
-        postgres_conn_id="superset_qa",
+        postgres_conn_id="superset",
         sql="""
         INSERT INTO library_cern_publication_records (year,
         publications_total_count, conference_proceedings_count,

diff --git a/dags/migrations/migrations.py b/dags/migrations/migrations.py
@@ -14,7 +14,7 @@
 ) as dag:
     AlembicOperator(
         task_id="alembic_op",
-        conn_id="superset_qa",
+        conn_id="superset",
         command="{{ params.command }}",
         revision="{{ params.revision }}",
         script_location=f"{os.environ['AIRFLOW_HOME']}/dags/migrations/",

diff --git a/dags/open_access/gold_open_access_mechanisms.py b/dags/open_access/gold_open_access_mechanisms.py
@@ -70,7 +70,7 @@ def join_and_add_year(counts, **kwargs):
 
     populate_golden_open_access = PostgresOperator(
         task_id="populate_golden_open_access",
-        postgres_conn_id="superset_qa",
+        postgres_conn_id="superset",
         sql="""
         INSERT INTO oa_golden_open_access (year, cern_read_and_publish, cern_individual_apcs,
         scoap3, other, other_collective_models, created_at, updated_at)

diff --git a/dags/open_access/open_access.py b/dags/open_access/open_access.py
@@ -51,7 +51,7 @@ def join(values, **kwargs):
 
     PostgresOperator(
         task_id="populate_open_access_table",
-        postgres_conn_id="superset_qa",
+        postgres_conn_id="superset",
         sql="""
         INSERT INTO oa_open_access (year, closed_access, bronze_open_access,
         green_open_access, gold_open_access, created_at, updated_at)

diff --git a/docker-compose.yaml b/docker-compose.yaml
@@ -60,7 +60,7 @@ x-airflow-common: &airflow-common
     AIRFLOW__CORE__LOAD_EXAMPLES: "true"
     AIRFLOW__API__AUTH_BACKENDS: "airflow.api.auth.backend.basic_auth,airflow.api.auth.backend.session"
     AIRFLOW_CONN_CDS: "http://cds.cern.ch"
-    AIRFLOW_CONN_SUPERSET_QA: "postgresql://bi_local:bi_local@postgres_local:5432/bi_local"
+    AIRFLOW_CONN_SUPERSET: "postgresql://bi_local:bi_local@postgres_local:5432/bi_local"
     # yamllint disable rule:line-length
     # Use simple http server on scheduler for health checks
     # See https://airflow.apache.org/docs/apache-airflow/stable/administration-and-deployment/logging-monitoring/check-health.html#scheduler-health-check-server