如何在Python脚本中复用Snowflake外部浏览器SSO认证信息？

嘿，我完全懂你的烦恼——每次运行Python脚本都要跳浏览器输一遍凭证，确实有点折腾，毕竟DataGrip里都能维持好几个小时的认证状态对吧？其实Snowflake的Python生态（包括Snowpark）早就考虑到了这个需求，咱们可以通过官方支持的配置来实现认证信息的复用，不用每次都重新登录。

方法一：用官方自带的本地会话缓存（最推荐） #

这是最简单也最稳妥的方式，只需要在你的连接参数里新增两个配置项，让Snowpark自动帮你缓存认证令牌：

from snowflake.snowpark import Session

connection_parameters = {
    "account": "xyz",
    "authenticator": "externalbrowser",
    "user": "abcde",
    "role": "role_one",
    "warehouse": "tiny",
    "database": "db1234",
    # 开启会话保持，让认证状态持续有效
    "client_session_keep_alive": True,
    # 允许驱动将临时凭证存储到本地
    "client_store_temporary_credential": True
}

# 创建会话
session = Session.builder.configs(connection_parameters).create()

原理说明：

第一次运行脚本时，你还是需要通过浏览器完成SSO登录，但之后Snowpark会把有效的认证令牌存储在你本地用户目录下的.snowflake文件夹里。后续再运行脚本时，驱动会自动读取这个缓存的令牌，只要令牌还在有效期内（和DataGrip的有效期一致，通常是几小时），就不会再弹出浏览器让你登录了。

方法二：手动管理会话令牌（适合需要自定义缓存逻辑的场景） #

如果你需要更灵活地控制缓存的存储位置或者有效期，可以手动提取会话的令牌信息，保存到本地文件，下次启动时直接加载：

import json
from snowflake.snowpark import Session

# 尝试从本地缓存文件加载会话信息
try:
    with open("snowflake_session_cache.json", "r") as cache_file:
        cached_config = json.load(cache_file)
        # 用缓存的令牌直接创建会话
        session = Session.builder.configs(cached_config).create()
        print("已复用缓存的认证信息")
except (FileNotFoundError, Exception):
    # 缓存不存在或失效时，重新进行SSO登录
    connection_parameters = {
        "account": "xyz",
        "authenticator": "externalbrowser",
        "user": "abcde",
        "role": "role_one",
        "warehouse": "tiny",
        "database": "db1234"
    }
    session = Session.builder.configs(connection_parameters).create()
    print("已完成新的SSO登录")
    
    # 提取当前会话的认证令牌信息，保存到本地缓存
    conn = session._conn
    cache_data = {
        "account": "xyz",
        "user": "abcde",
        "role": "role_one",
        "warehouse": "tiny",
        "database": "db1234",
        "session_token": conn._session_token,
        "master_token": conn._master_token,
        "session_id": conn._session_id
    }
    with open("snowflake_session_cache.json", "w") as cache_file:
        json.dump(cache_data, cache_file)

注意事项：

• 这个方法里用到的_conn、_session_token等都是Snowpark的私有属性，未来版本更新可能会有变动，所以优先推荐第一种官方缓存方式。
• 令牌是有有效期的，过期后还是需要重新登录，你可以在代码里加个过期判断逻辑来优化体验。

额外提醒 #

• 本地缓存的凭证是存在你的用户专属目录里的，如果你在共享机器上使用，要注意不要泄露这个缓存文件，避免权限风险。
• 如果需要强制重新登录，只需要删除本地的.snowflake文件夹（对应方法一）或者手动删除你创建的缓存文件（对应方法二）即可。
• 确保你的Snowpark和Snowflake Python驱动版本是较新的，旧版本可能不支持这些缓存配置项。

备注：内容来源于stack exchange，提问作者flubber