airflow.providers.microsoft.azure.hooks.batch

模組內容

類別

AzureBatchHook

Azure Batch API 的 Hook。
class airflow.providers.microsoft.azure.hooks.batch.AzureBatchHook(azure_batch_conn_id=default_conn_name)[原始碼]

基底類別: airflow.hooks.base.BaseHook

Azure Batch API 的 Hook。

參數

azure_batch_conn_id (str) – 將用於啟動容器執行個體的服務主體的 Azure Batch 連線 ID。

conn_name_attr = 'azure_batch_conn_id'[原始碼]
default_conn_name = 'azure_batch_default'[原始碼]
conn_type = 'azure_batch'[原始碼]
hook_name = 'Azure Batch Service'[原始碼]
classmethod get_connection_form_widgets()[原始碼]

傳回要新增至連線表單的連線小工具。

classmethod get_ui_field_behaviour()[原始碼]

傳回自訂欄位行為。

connection()[原始碼]

取得 Batch 用戶端連線 (已快取)。

get_conn()[原始碼]

取得 Batch 用戶端連線。

傳回

Azure Batch 用戶端

傳回類型

azure.batch.BatchServiceClient

configure_pool(pool_id, vm_size, vm_node_agent_sku_id, vm_publisher=None, vm_offer=None, sku_starts_with=None, vm_sku=None, vm_version=None, os_family=None, os_version=None, display_name=None, target_dedicated_nodes=None, use_latest_image_and_sku=False, **kwargs)[原始碼]

設定集區。

參數

  • pool_id (str) – 在帳戶中唯一識別集區的字串

  • vm_size (str) – 集區中虛擬機器的大小。

  • display_name (str | None) – 集區的顯示名稱

  • target_dedicated_nodes (int | None) – 集區中所需的專用計算節點數。

  • use_latest_image_and_sku (bool) – 是否使用最新的已驗證 VM 映像和 SKU

  • vm_publisher (str | None) – Azure 虛擬機器 Marketplace 映像的發行者。例如，Canonical 或 MicrosoftWindowsServer。

  • vm_offer (str | None) – Azure 虛擬機器 Marketplace 映像的供應項目類型。例如，UbuntuServer 或 WindowsServer。

  • sku_starts_with (str | None) – 要搜尋的 SKU 的起始名稱

  • vm_sku (str | None) – 要使用的虛擬機器 SKU 名稱

  • vm_version (str | None) – 虛擬機器的版本

  • vm_version – str

  • vm_node_agent_sku_id (str) – 虛擬機器的節點代理程式 SKU ID

  • os_family (str | None) – 要安裝在集區中虛擬機器上的 Azure 客體作業系統系列。

  • os_version (str | None) – 作業系統系列版本

create_pool(pool)[原始碼]

建立集區 (如果尚不存在)。

參數

pool (azure.batch.models.PoolAddParameter) – 要建立的集區物件

wait_for_all_node_state(pool_id, node_state)[原始碼]

等待集區中的所有節點達到給定狀態。

參數

  • pool_id (str) – 識別集區的字串

  • node_state (set) – batch_models.ComputeNodeState 的集合

configure_job(job_id, pool_id, display_name=None, **kwargs)[原始碼]

設定要在集區中使用的作業。

參數

  • job_id (str) – 在帳戶中唯一識別作業的字串

  • pool_id (str) – 識別集區的字串

  • display_name (str | None) – 作業的顯示名稱

create_job(job)[原始碼]

在集區中建立作業。

參數

job (azure.batch.models.JobAddParameter) – 要建立的作業物件

configure_task(task_id, command_line, display_name=None, container_settings=None, **kwargs)[原始碼]

建立任務。

參數

  • task_id (str) – 識別要建立之任務的字串

  • command_line (str) – 任務的命令列。

  • display_name (str | None) – 任務的顯示名稱

  • container_settings – 任務在其下執行的容器設定。如果將執行此任務的集區已設定 containerConfiguration，則也必須設定此項。如果將執行此任務的集區未設定 containerConfiguration，則不得設定此項。

add_single_task_to_job(job_id, task)[原始碼]

將單一任務新增至給定作業 (如果任務不存在)。

參數

  • job_id (str) – 識別給定作業的字串

  • task (azure.batch.models.TaskAddParameter) – 要新增的任務

wait_for_job_tasks_to_complete(job_id, timeout)[原始碼]

等待特定作業中的任務完成。

參數

  • job_id (str) – 識別作業的字串

  • timeout (int) – 在逾時 (以分鐘為單位) 之前等待的時間量

test_connection()[原始碼]

測試已設定的 Azure Batch 連線。

此條目是否有幫助？