From e1fd3b67d288bc1ea5b98116b6f12e4f6e83a73f Mon Sep 17 00:00:00 2001 From: Martin Yushko <139386268+mayushko26@users.noreply.github.com> Date: Wed, 27 Nov 2024 15:13:26 -0500 Subject: [PATCH] Revert "Load default Airflow connections on initialization (#182)" This reverts commit f71d082cac983385349f3e30b8a650d5f5c5d619. --- .../airflow/2.10.1/python/mwaa/database/migrate.py | 13 +------------ .../airflow/2.9.2/python/mwaa/database/migrate.py | 12 +----------- 2 files changed, 2 insertions(+), 23 deletions(-) diff --git a/images/airflow/2.10.1/python/mwaa/database/migrate.py b/images/airflow/2.10.1/python/mwaa/database/migrate.py index e03b8d0..17a6512 100644 --- a/images/airflow/2.10.1/python/mwaa/database/migrate.py +++ b/images/airflow/2.10.1/python/mwaa/database/migrate.py @@ -55,21 +55,10 @@ def _migrate_db(): logging.info("The database is now migrated.") -@with_db_lock(1234) -def _create_default_connections(): - from airflow.cli.commands import connection_command as airflow_connection_command - - try: - args = Namespace(verbose=False) - airflow_connection_command.create_default_connections(args) - logging.info("The default Airflow connections have been created.") - except TimeoutError: - logging.warning("Timed out on initializatin of default Airflow connections.") - def _main(): _verify_environ() _migrate_db() - _create_default_connections() + if __name__ == "__main__": _main() diff --git a/images/airflow/2.9.2/python/mwaa/database/migrate.py b/images/airflow/2.9.2/python/mwaa/database/migrate.py index e319549..17a6512 100644 --- a/images/airflow/2.9.2/python/mwaa/database/migrate.py +++ b/images/airflow/2.9.2/python/mwaa/database/migrate.py @@ -54,21 +54,11 @@ def _migrate_db(): airflow_db_command.migratedb(args) logging.info("The database is now migrated.") -@with_db_lock(1234) -def _create_default_connections(): - from airflow.cli.commands import connection_command as airflow_connection_command - - try: - args = Namespace(verbose=False) - airflow_connection_command.create_default_connections(args) - logging.info("The default Airflow connections have been created.") - except TimeoutError: - logging.warning("Timed out on initializatin of default Airflow connections.") def _main(): _verify_environ() _migrate_db() - _create_default_connections() + if __name__ == "__main__": _main()