Load subscription_id from extra__azure__subscriptionId (#30556)

apache · Apr 10, 2023 · 7ce3b66 · 7ce3b66
1 parent 3a25234
commit 7ce3b66
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/airflow/providers/microsoft/azure/operators/data_factory.py b/airflow/providers/microsoft/azure/operators/data_factory.py
@@ -52,7 +52,9 @@ def get_link(
         conn_id = operator.azure_data_factory_conn_id  # type: ignore
         conn = BaseHook.get_connection(conn_id)
         extras = conn.extra_dejson
-        subscription_id = get_field(extras, "subscriptionId")
+        subscription_id = get_field(extras, "subscriptionId") or get_field(
+            extras, "extra__azure__subscriptionId"
+        )
         if not subscription_id:
             raise KeyError(f"Param subscriptionId not found in conn_id '{conn_id}'")
         # Both Resource Group Name and Factory Name can either be declared in the Azure Data Factory