databricks/databricks-sdk-py

解决的问题

该 SDK 为开发者提供了一种与 Databricks Lakehouse 平台进行程序化交互的方式。它通过提供所有公开 API 操作的稳健、类型化的 Python 接口，消除了手动编写 Databricks REST API HTTP 请求的需要。

工作原理

该库的核心是 WorkspaceClient 和 AccountClient 类，它们负责连接到 Databricks 工作区和账户。它内置了一个 HTTP 客户端，可自动处理智能重试、分页响应以及长时间运行的操作。SDK 支持灵活的认证流程，可自动从环境变量、配置文件（.databrickscfg）或云原生身份提供商（AWS、Azure 和 GCP）中检测凭据。

适用人群

需要自动化管理 Databricks 集群、作业和 Unity Catalog 资源，或希望将 Databricks 功能集成到外部应用程序中的 Python 开发者和数据工程师。

主要亮点

• 完整 API 覆盖：涵盖所有公开的 Databricks REST API 操作。
• 自动认证：支持默认认证流程，可跨不同云提供商自动检测凭据。
• 稳健性：内置智能重试机制，用于处理 API 失败。
• 云原生集成：与 Azure 和 GCP 的原生认证方法深度集成。
• 开发者友好：提供自动补全支持和常见用例（如 OAuth Web 服务器）的集成示例。