diff --git a/dags/library/cern_publication_records.py b/dags/library/cern_publication_records.py index 3e4a4cc..9eedeed 100644 --- 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 index 6d01466..32ec247 100644 --- 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 index fb5b91f..9fb46f1 100644 --- 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 index b2731b4..bdf7c1f 100644 --- 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 index 1bdf9f0..c51ac70 100644 --- 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