Fix connection resolution in CLI by setting server process context in decorators

Author: uplsh580

airflow-core/src/airflow/cli/commands/connection_command.py

@@ -65,8 +65,6 @@ def _connection_mapper(conn: Connection) -> dict[str, Any]:
 @providers_configuration_loaded
 def connections_get(args):
     """Get a connection."""
-    os.environ["_AIRFLOW_PROCESS_CONTEXT"] = "server"
-
     try:
         conn = Connection.get_connection_from_secrets(args.conn_id)
     except AirflowNotFoundException:
@@ -357,8 +355,6 @@ def _import_helper(file_path: str, overwrite: bool) -> None:
 @providers_configuration_loaded
 def connections_test(args) -> None:
     """Test an Airflow connection."""
-    os.environ["_AIRFLOW_PROCESS_CONTEXT"] = "server"
-
     console = AirflowConsole()
     if conf.get("core", "test_connection", fallback="Disabled").lower().strip() != "enabled":
         console.print(

airflow-core/src/airflow/utils/cli.py

@@ -107,6 +107,7 @@ def wrapper(*args, **kwargs):
                     from airflow.configuration import conf
                     from airflow.utils.db import check_and_run_migrations, synchronize_log_template
 
+                    os.environ["_AIRFLOW_PROCESS_CONTEXT"] = "server"
                     if conf.getboolean("database", "check_migrations"):
                         check_and_run_migrations()
                     synchronize_log_template()
@@ -436,6 +437,7 @@ def suppress_logs_and_warning(f: T) -> T:
     @functools.wraps(f)
     def _wrapper(*args, **kwargs):
         _check_cli_args(args)
+        os.environ["_AIRFLOW_PROCESS_CONTEXT"] = "server"
         if args[0].verbose:
             f(*args, **kwargs)
         else: