我正在尝试建立一个调用 databricks 笔记本的 AWS lambda 函数(在 s3 触发器的情况下)。我知道我必须在 lambda 函数(python)代码中使用 databricks 的 Jobs API 来使用以下命令发出 POST 请求:运行提交函数的 JSON 负载。
虽然文档不是很清楚,但我能够调用测试脚本,并在检查响应文本时看到 databricks 登录页面 html 代码,这意味着它没有经过身份验证。
我确实阅读了用户令牌,但我不确定如何将它们纳入身份验证。
任何以其他方式完成此工作或帮助我使用 user_tokens 进行身份验证以便流程到达笔记本的执行而不是在身份验证页面停止的帮助都会有所帮助。
提前致谢。
代码示例:
import requests
import json
job_payload = {
"run_name": 'just_a_run',
"existing_cluster_id": '****',
"notebook_task":
{
"notebook_path": 'https://databricks.cloud.company.com/****'
}
}
resp = requests.post('https://databricks.cloud.company.com/2.0/jobs/runs/submit', json=job_payload)
print(resp.status_code)
print(resp.text)
200
<!DOCTYPE html>
<html>
<head>
<meta charset="utf-8"/>
<meta http-equiv="Content-Language" content="en"/>
<title>Databricks - Sign In</title>
<meta name="viewport" content="width=960">
<link rel="stylesheet" href="/login/bootstrap.min.css">
<link rel="icon" type="image/png" href="login/favicon.ico" />
<meta http-equiv="content-type" content="text/html; charset=UTF8">
<link rel="shortcut icon" href="favicon.ico"><link href="login/login.e555bb48.css" rel="stylesheet"></head>
<body>
<div id="login-page"></div>
<script type="text/javascript" src="login/login.dabd48fd.js"></script></body>
</html>
Run Code Online (Sandbox Code Playgroud)
mun*_*aba 10
解决了:
1) 您需要创建一个用于授权的用户令牌,并在执行 REST 请求时将其作为“headers”参数发送。
2) headers={'Authorization': 'Bearer token'} 代替令牌的必须是您从 databricks 获取的实际令牌。
3) api链接必须以/api开头
4)databricks笔记本的路径必须是绝对路径,即“/Users/$USER_NAME/book_name”
最终工作代码:
import requests
import json
job_payload = {
"run_name": 'just_a_run',
"existing_cluster_id": 'id_of_cluster',
"notebook_task":
{
"notebook_path": '/Users/username/notebook_name'
}
}
resp = requests.post('https://databricks.cloud.company.com/api/2.0/jobs/runs/submit', json=job_payload, headers={'Authorization': 'Bearer token'})
print(resp.status_code)
print(resp.text)
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
13969 次 |
| 最近记录: |