databricks - 如何访问 Databricks Repos 元数据？

Question

有没有办法在 Repo 的笔记本中访问诸如 Repo url 和 Branch name 之类的数据？也许在 dbutils 中有些东西。

Answer 1

您可以为此使用 Repos API - 特别是Get 命令。您可以通过 dbutils 从可用的笔记本上下文中提取笔记本路径，然后执行以下两个查询：

1. 通过Workspace API按路径获取 repo ID （repo 路径始终包含 3 个组件 - /Repos、目录（用于用户或自定义）和实际存储库名称）
2. 获取回购数据

像这样的东西：

import json
import requests

ctx = json.loads(
  dbutils.notebook.entry_point.getDbutils().notebook().getContext().toJson())

notebook_path = ctx['extraContext']['notebook_path']
repo_path = '/'.join(notebook_path.split('/')[:4])
api_url = ctx['extraContext']['api_url']
api_token = ctx['extraContext']['api_token']

repo_dir_data = requests.get(f"{api_url}/api/2.0/workspace/get-status",  
                             headers = {"Authorization": f"Bearer {api_token}"},
                             json={"path": repo_path}).json()
repo_id = repo_dir_data['object_id']
repo_data = requests.get(f"{api_url}/api/2.0/repos/{repo_id}",  
                         headers = {"Authorization": f"Bearer {api_token}"}
                        ).json()