在Databricks社区版挂载OneDrive for Business的可行方案咨询

Databricks社区版支持挂载OneDrive for Business，但需根据目标文件夹类型（个人OneDrive/团队站点文档库）选择合适的认证方式，以下是具体方案及你的代码问题排查：

一、核心问题排查：你的AccessToken获取失败原因 #

你当前使用的client_credentials（服务主体）认证方式，仅适用于访问SharePoint团队站点/组织文档库，无法直接访问用户个人OneDrive for Business。此外，你的请求还存在两个潜在问题：

1. client_credentials流不需要redirect_uri参数，可移除
2. 若使用Microsoft Graph v2.0端点，需用scope替代resource参数

二、可行挂载方案 #

方案1：挂载团队站点/组织文档库（服务主体认证） #

适用于挂载属于团队、部门的共享OneDrive文件夹，步骤如下：

1. 应用注册配置：
   • 在Azure AD中为应用添加Files.ReadWrite.All或Sites.ReadWrite.All应用权限（注意是应用权限，不是委派权限）
   • 要求租户管理员完成权限同意
2. 修正后的代码：

import requests

# 替换为你的实际信息
client_id = "你的客户端ID"
client_secret = "你的客户端密钥"
tenant_id = "你的租户ID"
site_id = "目标站点ID"
folder_id = "目标文件夹ID"
mount_point = "/mnt/onedrive-team"

# 获取访问令牌
response = requests.post(
    f"https://login.microsoftonline.com/{tenant_id}/oauth2/token",
    data={
        "client_id": client_id,
        "client_secret": client_secret,
        "grant_type": "client_credentials",
        "resource": "https://graph.microsoft.com"
    }
)

# 检查令牌获取结果
if response.status_code != 200:
    raise Exception(f"令牌获取失败: {response.json()}")
access_token = response.json()["access_token"]

# 挂载到DBFS
dbutils.fs.mount(
    source="graph",
    mount_point=mount_point,
    extra_configs={
        "graph.access_token": access_token,
        "graph.site_id": site_id,
        "graph.folder_id": folder_id
    }
)

方案2：挂载个人OneDrive for Business（授权码流认证） #

服务主体无法访问个人OneDrive，需通过用户授权获取令牌，步骤如下：

1. 应用注册配置：
   • 添加Files.ReadWrite和offline_access委派权限
   • 设置重定向URI为http://localhost:8080或https://login.microsoftonline.com/common/oauth2/nativeclient
2. 本地获取授权码与令牌：

# 本地运行此代码，在浏览器中完成授权
import requests
from urllib.parse import urlencode

client_id = "你的客户端ID"
client_secret = "你的客户端密钥"
tenant_id = "你的租户ID"
redirect_uri = "http://localhost:8080"
scope = "https://graph.microsoft.com/Files.ReadWrite offline_access"

# 生成授权URL，复制到浏览器打开
auth_url = f"https://login.microsoftonline.com/{tenant_id}/oauth2/v2.0/authorize?{urlencode({
    'client_id': client_id,
    'response_type': 'code',
    'redirect_uri': redirect_uri,
    'scope': scope,
    'prompt': 'consent'
})}"
print("请访问以下URL获取授权码:", auth_url)

# 输入浏览器返回的授权码
code = input("输入授权码: ")

# 交换访问令牌与刷新令牌
token_response = requests.post(
    f"https://login.microsoftonline.com/{tenant_id}/oauth2/v2.0/token",
    data={
        "client_id": client_id,
        "client_secret": client_secret,
        "code": code,
        "redirect_uri": redirect_uri,
        "grant_type": "authorization_code"
    }
)
token_data = token_response.json()
print("访问令牌:", token_data["access_token"])
print("刷新令牌:", token_data["refresh_token"])

3. 在Databricks中挂载：
   • 将获取到的access_token、refresh_token、client_secret存入Databricks Secrets（避免明文泄露）
   • 执行挂载代码：

mount_point = "/mnt/personal-onedrive"
dbutils.fs.mount(
    source="graph",
    mount_point=mount_point,
    extra_configs={
        "graph.access_token": dbutils.secrets.get("你的密钥范围", "onedrive-access-token"),
        "graph.refresh_token": dbutils.secrets.get("你的密钥范围", "onedrive-refresh-token"),
        "graph.client_id": client_id,
        "graph.client_secret": dbutils.secrets.get("你的密钥范围", "onedrive-client-secret"),
        "graph.folder_id": "你的个人OneDrive文件夹ID"
    }
)

三、社区版额外注意事项 #

• 挂载的存储会占用Databricks社区版的免费存储配额（10GB）
• 访问令牌有效期通常为1小时，使用刷新令牌可自动续期
• 若遇到网络问题，可通过requests.get("https://graph.microsoft.com/v1.0/sites")测试连通性

内容的提问来源于stack exchange，提问作者Saravana Kumar