為 Visual Studio Code 的 Databricks 擴充功能設置授權

本文說明如果您尚未透過專案設定來設定延伸模組，如何在Visual Studio Code 的 Databricks 擴充功能和 Azure Databricks 工作區之間設定授權和驗證。 請參閱什麼是適用於 Visual Studio Code 的 Databricks 延伸模組？。

適用於 Visual Studio Code 的 Databricks 延伸模組會實作 Databricks 統一驗證 標準的部分，可讓您設定 Azure Databricks OAuth 2.0 型授權一次，然後跨多個 Azure Databricks 工具和 SDK 使用該設定，而不需要進一步的授權和驗證設定變更。 它也提供使用中存取令牌的自動重新整理，減少開發時的程序複雜度。

Databricks 建議 OAuth 互動式使用者授權，這很容易使用適用於Visual Studio Code 的 Databricks 擴充功能進行設定。 請參閱 設定 OAuth 授權和驗證。

否則，如果您使用的是 使用 Azure Databricks 個人存取權杖進行身份驗證（舊版）、使用 OAuth 為 Azure Databricks 授權服務主體，或 使用 Microsoft Entra 服務主體進行身份驗證，請參閱 配置設定檔以進行身份驗證。 如需 使用 Azure CLI 進行驗證，請參閱 設定 Azure CLI 驗證。

注意

適用於 Visual Studio Code 的 Databricks 擴充功能不支援使用 Azure 受控識別進行驗證（以前稱為 Azure 受控服務識別 (MSI)）。然而，不使用 Visual Studio Code 的 Databricks 擴充功能的 Databricks Connect 則支援此功能。 若要安裝 Databricks Connect，請參閱 安裝適用於 Python 的 Databricks Connect。

設定 OAuth 授權和驗證

Azure Databricks 支援以 OAuth 2.0 為基礎的授權和驗證，並使用 Databricks 統一驗證簡化存取權杖管理。 如需使用 Azure Databricks 進行使用者和服務主體授權的詳細資訊，請參閱 授權存取 Azure Databricks 資源。

若要完成 OAuth 授權的設定，請使用已啟用 Visual Studio Code 的 Databricks 延伸模組開啟 VSCode 專案：

1. 在 [組態] 檢視中，按下 [驗證類型]，然後按下齒輪 (登入 Databricks 工作區) 圖示。

   

2. 如果您在此清單中已經有驗證組態設定檔，其中包含 [使用 OAuth 進行驗證 (使用者對機器)] 標籤，且您知道對應至目標 Databricks 主機，請從清單中選取，然後執行下列動作：

   1. 如果出現提示，請在網頁瀏覽器中完成任何畫面上的指示，以使用 Azure Databricks 完成驗證。
   2. 如果也出現提示，請允許 all-apis 存取。
   3. 直接跳到 選擇用於執行程式碼和作業的計算資源。

3. 在 [命令選擇區] 中，針對 [選取驗證方法]，按下 [OAuth (使用者到機器)]。

4. 針對選取驗證方法，選擇OAuth (使用者到機器)。

5. 為相關聯的 Azure Databricks 驗證設定檔輸入一個名稱。

6. 在 [設定] 檢視中，點選 [登入 Databricks]。

   

7. 在 [命令選擇區] 中，針對 [選取驗證方法]，選取您剛才建立的驗證組態設定檔名稱。

8. 如果出現提示，請在網頁瀏覽器中完成任何畫面上的指示，以使用 Azure Databricks 完成驗證。 如果也出現提示，請允許 all-apis 存取。

9. 成功登入之後，返回 Visual Studio Code。

使用配置檔案設置驗證

假設您已經設定了帶有 Azure Databricks 驗證類型所需欄位的 Azure Databricks 組態設定檔，請參考下列指示。 若要建立設定檔，請參閱建立 Azure Databricks 組態設定檔。 若要設定設定檔的設定，請參閱設定檔索引標籤或授權存取 Azure Databricks 資源中的驗證類型區段。

若要選取您想要延伸模組使用的組態設定檔，請在專案和延伸模組開啟時執行下列動作：

1. 在 [組態] 檢視中，按下 [驗證類型]，然後按下齒輪 (登入 Databricks 工作區) 圖示。

   

2. 在 [命令選擇區] 中，選取清單中的目標 Azure Databricks 組態設定檔。

   提示

   稍後可以透過修改與專案關聯的 workspace 設定檔中的目標 databricks.yml 設定，來變更目標工作區主機。 請參閱 Databricks Asset Bundle 組態。

如果您沒有設定檔，您可以建立一個使用 Azure Databricks 個人存取權杖進行驗證的設定檔，如下所示：

1. 開啟您的專案和延伸模組之後，在 [組態] 檢視中，按下 [驗證類型] 或 [登入 Databricks]，然後按下齒輪 (登入 Databricks 工作區) 圖示。

   

2. 在 [命令選擇區] 中，選取 [個人存取權杖]。

3. 依照畫面上的提示完成下列動作：

   1. 為相關聯的 Azure Databricks 驗證設定檔輸入一個名稱。
   2. 在 Azure Databricks 工作區中產生個人存取令牌，然後複製令牌的值。
   3. 返回延伸模組，並輸入複製的令牌值。

如果尚未存在，該延伸模組會在專案中建立一個名為 .databricks 的資料夾，其中包含一個名為 databricks.env 的檔案。 此檔案包含您輸入的工作區主機，以及適用於 Visual Studio Code 的 Databricks 延伸模組所使用的其他 Azure Databricks 組態詳細資料。

如果檔案不存在，或在任何父資料夾中找不到現有 .gitignore，此延伸模組也會將隱藏 .gitignore 的檔案新增至專案。 如果已建立新的 .gitignore 檔案，延伸模組會將 .databricks/ 項目新增至這個新檔案。 如果延伸模組找到現有的 .gitignore 檔案，它會將 .databricks/ 項目新增至現有的檔案。

設定 Azure CLI 驗證

Azure Databricks 支援在本機安裝 Azure CLI 的驗證。 這可讓您產生短期的 (一小時) Microsoft Entra ID 權杖，從而避免不小心透過版本控制簽入或其他方式公開存留期更長的權杖的風險，例如 Azure Databricks 個人存取權杖。

在您的專案和延伸模組開啟，並在本機安裝 Azure CLI 時，請執行下列動作：

1. 在「設定」檢視中，點選「設定 Databricks」。

   注意

   如果看不到 [設定 Databricks]，請按一下 [配置] 旁的齒輪圖示 (即 [設定工作區])。

   

2. 在命令面板中，針對Databricks 主機，輸入您的每個工作區 URL，例如https://adb-1234567890123456.7.azuredatabricks.net。 然後按 Enter。

3. 選取 Azure CLI。