从“google\u cloud\u platform”类型的已保存连接中提取Keyfile JSON

Question

我在气流中保存了一个类型为“google\u cloud\u platform”的连接，如下所述https://cloud.google.com/composer/docs/how-to/managing/connections

现在在我的DAG中，我需要从保存的连接中提取Keyfile JSON

要使用的正确挂钩是什么？你知道吗

Answer 1

您可以按如下方式使用BaseHook：

from airflow.hooks.base_hook import BaseHook

GCP_CONNECTION_ID="my-gcp-connection"
BaseHook.get_connection(GCP_CONNECTION_ID).extras["extra__google_cloud_platform__keyfile_dict"]

Answer 2

使用^{}获取存储的连接。例如

from airflow.contrib.hooks.gcp_api_base_hook import GoogleCloudBaseHook

gcp_hook = GoogleCloudBaseHook(gcp_conn_id="<your-conn-id>")
keyfile_dict = gcp_hook._get_field('keyfile_dict')