如何在Apache Superset中通过REST连接动态密码的Pgmaker数据库

可以通过自定义连接逻辑或Superset API结合REST密码接口实现对接，以下是两种可行方案：

方案1：自定义数据库连接逻辑（推荐） #

Superset支持通过配置钩子动态修改数据库连接参数，你可以在建立连接前调用REST接口获取最新密码：

• 1. 编写动态密码获取函数
     在Superset的superset_config.py中添加密码获取逻辑：

  import requests
  
  def get_pgmaker_latest_password():
      # 调用你的REST接口获取动态密码
      resp = requests.get("https://your-rest-endpoint/pgmaker-password", headers={"X-API-Key": "your-secret-key"})
      resp.raise_for_status()
      return resp.json()["password"]
  

• 2. 配置连接参数修改钩子
     同样在superset_config.py中添加连接修改逻辑，仅针对Pgmaker数据库替换密码：

  def mutate_db_connection(uri, params, username, password, security_manager):
      # 识别Pgmaker连接（可通过URI中的特征字符串判断）
      if "pgmaker" in uri.lower():
          password = get_pgmaker_latest_password()
      return uri, params, username, password
  
  DB_CONNECTION_MUTATOR = mutate_db_connection
  

• 3. 配置Superset数据库连接
     在Superset界面添加Pgmaker连接时，填写基础SQLAlchemy URI（如postgresql://pgmaker_user:@pgmaker-host:5432/pgmaker_db），密码字段留空即可——钩子会自动替换为REST接口获取的最新密码。

方案2：通过Superset API定期更新密码 #

如果不想修改代码，可通过定时任务调用Superset API更新连接密码：

• 1. 获取Superset API令牌

  curl -X POST -H "Content-Type: application/json" \
    -d '{"username":"superset-admin","password":"your-admin-pass","provider":"db"}' \
    http://superset-host/api/v1/security/login
  

• 2. 定期更新Pgmaker连接密码
     先通过Superset API查询目标数据库的ID，再调用更新接口替换密码：

  # 更新密码（替换{DB_ID}、{NEW_PASSWORD}、{ACCESS_TOKEN}）
  curl -X PUT -H "Content-Type: application/json" \
    -H "Authorization: Bearer {ACCESS_TOKEN}" \
    -d '{"password":"{NEW_PASSWORD}"}' \
    http://superset-host/api/v1/database/{DB_ID}
  

  可通过crontab或定时任务工具（如Airflow）定期执行该脚本，确保密码与Pgmaker的动态更新同步。

关键注意事项 #

• 确保REST密码接口使用HTTPS传输，并添加API密钥等身份验证机制，防止密码泄露。
• 自定义连接逻辑中需添加异常处理（如接口超时、密码获取失败），避免Superset连接崩溃。
• 测试时验证密码更新后的连接有效性，确保每次建立连接都能获取最新的有效密码。

内容的提问来源于stack exchange，提问作者Deepak Tibhe