Repolode
databricks

databricks/databricks-sdk-py

用于 Databricks 平台的 Python SDK,可在 Python 代码中调用 Databricks REST API。

Star
577
Fork
228
未关闭 Issue
344
最近提交
3天前

这是什么?

Databricks SDK for Python 是一个 Python 客户端库,用于在 Python 中操作 Databricks 工作区和账户。它覆盖 Databricks 公开的 REST API,并通过 WorkspaceClient 或 AccountClient 对象以类型化客户端的形式提供 clusters、jobs、repos 等各项服务。

该 SDK 支持 Databricks 原生令牌、OAuth、Azure 与 Google Cloud 凭证等多种认证方式,并能自动解析云厂商、工作区 ID 等配置。内置 HTTP 客户端会对可安全重试的请求(如 429、503 响应)进行带抖动的退避重试。长时间运行的操作提供等待工具,可阻塞直到资源进入预期状态;分页列表接口则以 Python 迭代器的形式提供。

该包通过 pip 安装,支持 Python 3.8 至 3.11,并已内置于较新版本的 Databricks Runtime 中。目前处于 Beta 阶段,后续版本接口可能有所调整。

核心功能

  • 01为 clusters、jobs、repos 等 Databricks REST API 服务提供类型化的 Python 客户端。
  • 02支持个人访问令牌、OAuth、Azure 及 Google Cloud 等多种认证方式。
  • 03对限流及临时不可用的响应进行带退避的自动重试。
  • 04为集群创建、作业运行等长时间操作提供等待工具。
  • 05通过 Python 迭代器自动处理分页结果。
  • 06可配置的日志与调试输出,记录 HTTP 请求和响应。

适用场景

  • 用 Python 脚本自动化管理集群、作业和工作区。
  • 构建通过用户 OAuth 登录操作 Databricks 资源的内部 Web 应用。
  • 编写定时脚本,查询作业运行状态和历史记录以检查数据管道。
  • 在基于 Python 的 CI/CD 流水线中管理 Databricks 资源。

快速上手

  1. 1

    从 PyPI 安装 SDK

    pip install databricks-sdk
  2. 2

    创建 WorkspaceClient 并列出集群

    from databricks.sdk import WorkspaceClient w = WorkspaceClient() for c in w.clusters.list(): print(c.cluster_name)

使用前须知

  • 需要 Databricks 工作区和有效凭据才能调用任何 API。
  • 目前处于 Beta 阶段,未来版本可能包含接口变更。

收录于专题

标签

常见问题

databricks-sdk-py 是什么?
用于 Databricks 平台的 Python SDK,可在 Python 代码中调用 Databricks REST API。 Databricks SDK for Python 是一个 Python 客户端库,用于在 Python 中操作 Databricks 工作区和账户。它覆盖 Databricks 公开的 REST API,并通过 WorkspaceClient 或 AccountClient 对象以类型化客户端的形式提供 clusters、jobs、repos 等各项服务。
databricks-sdk-py 可以免费使用吗?用的什么许可证?
databricks-sdk-py 以 Apache-2.0 许可证开源。用于商业产品前,请先确认许可证条款。
databricks-sdk-py 还在维护吗?
活跃维护中。databricks-sdk-py 最近一次提交是在 2026年11月5日。
databricks-sdk-py 适合用来做什么?
用 Python 脚本自动化管理集群、作业和工作区。 构建通过用户 OAuth 登录操作 Databricks 资源的内部 Web 应用。 编写定时脚本,查询作业运行状态和历史记录以检查数据管道。 在基于 Python 的 CI/CD 流水线中管理 Databricks 资源。
databricks-sdk-py 怎么安装和运行?
从 PyPI 安装 SDK: pip install databricks-sdk 创建 WorkspaceClient 并列出集群: from databricks.sdk import WorkspaceClient w = WorkspaceClient() for c in w.clusters.list(): print(c.cluster_name)
在哪里获取 databricks-sdk-py?
源代码和安装说明在 GitHub 上:https://github.com/databricks/databricks-sdk-py

概要由 AI 根据项目 README 生成;GitHub 数据每日更新。

相关项目