airflow.providers.databricks.hooks.databricks_base

Databricks hook。

此 hook 啟用向 Databricks 平台提交和運行任務。 內部運算子與 api/2.0/jobs/runs/submit 端點 通訊。

模組內容

類別

BaseDatabricksHook

與 Databricks 互動的基礎類別。

BearerAuth

aiohttp 僅提供 BasicAuth，對於 Bearer 驗證，我們需要 BasicAuth 的子類別。

屬性

AZURE_METADATA_SERVICE_TOKEN_URL

AZURE_METADATA_SERVICE_INSTANCE_URL

TOKEN_REFRESH_LEAD_TIME

AZURE_MANAGEMENT_ENDPOINT

DEFAULT_DATABRICKS_SCOPE

OIDC_TOKEN_SERVICE_URL

DEFAULT_AZURE_CREDENTIAL_SETTING_KEY

airflow.providers.databricks.hooks.databricks_base.AZURE_METADATA_SERVICE_TOKEN_URL = 'http://169.254.169.254/metadata/identity/oauth2/token'[source]
airflow.providers.databricks.hooks.databricks_base.AZURE_METADATA_SERVICE_INSTANCE_URL = 'http://169.254.169.254/metadata/instance'[source]
airflow.providers.databricks.hooks.databricks_base.TOKEN_REFRESH_LEAD_TIME = 120[source]
airflow.providers.databricks.hooks.databricks_base.AZURE_MANAGEMENT_ENDPOINT = 'https://management.core.windows.net/'[source]
airflow.providers.databricks.hooks.databricks_base.DEFAULT_DATABRICKS_SCOPE = '2ff814a6-3304-4ab8-85cb-cd0e6f879c1d'[source]
airflow.providers.databricks.hooks.databricks_base.OIDC_TOKEN_SERVICE_URL = '{}/oidc/v1/token'[source]
airflow.providers.databricks.hooks.databricks_base.DEFAULT_AZURE_CREDENTIAL_SETTING_KEY = 'use_default_azure_credential'[source]
class airflow.providers.databricks.hooks.databricks_base.BaseDatabricksHook(databricks_conn_id=default_conn_name, timeout_seconds=180, retry_limit=3, retry_delay=1.0, retry_args=None, caller='Unknown')[source]

基礎類別：airflow.hooks.base.BaseHook

與 Databricks 互動的基礎類別。

參數

  • databricks_conn_id (str) – 參考 Databricks 連線。

  • timeout_seconds (int) – requests 函式庫在逾時前等待的時間（秒）。

  • retry_limit (int) – 服務中斷時重試連線的次數。

  • retry_delay (float) – 重試之間等待的秒數（可以是浮點數）。

  • retry_args (dict[Any, Any] | None) – 傳遞給 tenacity.Retrying 類別的可選參數字典。

  • caller (str) – 呼叫此 hook 的運算子名稱。

conn_name_attr: str = 'databricks_conn_id'[source]
default_conn_name = 'databricks_default'[source]
conn_type = 'databricks'[source]
extra_parameters[source]
databricks_conn()[source]
get_conn()[source]

傳回 hook 的連線。

user_agent_header()[source]
user_agent_value()[source]
host()[source]
async __aenter__()[source]
async __aexit__(*err)[source]
class airflow.providers.databricks.hooks.databricks_base.BearerAuth(token)[source]

基礎類別：aiohttp.BasicAuth

aiohttp 僅提供 BasicAuth，對於 Bearer 驗證，我們需要 BasicAuth 的子類別。

encode()[source]

編碼憑證。

這個條目有幫助嗎？