Streamlit多页面应用session_state丢失原因排查

可能的原因和排查方向 #

1. setup_page函数逻辑存在错误判断 #

如果setup_page里的检查逻辑出问题，比如页面切换时误判access token不存在，就会把已有的session_state值覆盖或清空。比如：

• 函数是不是每次页面加载都硬要从查询参数拿授权码，完全忽略已经存在的token？
• 有没有不小心写了st.session_state.clear()这类代码，或者误重置了access_token这个键？

排查：在setup_page里加个打印语句，比如print(st.session_state)，观察切换页面前后session_state的变化，定位是哪一步导致token丢失。

2. 页面脚本的执行顺序不对 #

Streamlit多页面中每个页面都是独立脚本，切换页面时会重新执行目标页面的全部代码。如果setup_page不是在页面最顶部调用，而是在st.title或其他组件之后才执行，可能会导致session_state的读写时机异常。

排查：把setup_page放在每个页面脚本的最开头，在任何其他Streamlit组件之前执行，保证session_state的初始化逻辑优先运行。

3. OAuth2重定向逻辑冲突 #

handle_redirect函数如果在页面切换时被误触发，比如目标页面加载时硬要解析当前URL的查询参数（但此时根本没有授权码），可能会清空或覆盖session_state里的token。

排查：给handle_redirect添加触发条件——只有当URL中存在code授权码参数时，才执行兑换token的逻辑，否则直接跳过，保留现有session_state值。

4. session_state的键名不统一 #

要是不同页面使用的session_state键名不一致，比如首页用access_token，page1里写成accessToken，这种大小写或拼写差异会导致读不到已有的值，误以为token不存在。

排查：统一所有页面的session_state键名，比如固定使用access_token、user_info、user_groups这类名称。

5. Streamlit版本或环境存在bug #

旧版本的Streamlit可能存在多页面session_state共享的bug，比如切换页面时session_state没有被正确保留。

排查：将Streamlit升级到最新稳定版本，执行pip install --upgrade streamlit后，再测试问题是否解决。

示例修正思路 #

调整setup_page的逻辑，优先检查已有token，仅在无token且有授权码时执行兑换操作：

def setup_page():
    # 先检查是否有已保存的token
    if "access_token" in st.session_state and st.session_state["access_token"]:
        return
    # 再检查URL中是否存在授权码参数
    query_params = st.experimental_get_query_params()
    if "code" in query_params:
        # 用授权码兑换access token
        token = get_access_token(query_params["code"][0])
        if token:
            st.session_state["access_token"] = token
            st.session_state["user_info"] = get_user_info(token)
            st.session_state["user_groups"] = get_user_groups(token)
            # 清除查询参数，避免重复处理
            st.experimental_set_query_params()
            st.experimental_rerun()
        else:
            # 兑换失败，显示登录按钮
            show_login_button()
    else:
        # 无token也无授权码，显示登录按钮
        show_login_button()

内容的提问来源于stack exchange，提问作者Luis Valencia