From 907f4abd988091b7c9aa106db5c2694f5d3ae60a Mon Sep 17 00:00:00 2001 From: Sven Augustin Date: Fri, 25 Nov 2022 23:08:51 +0100 Subject: [PATCH] added rerunall and get_session_ids; simplified rerun a bit --- stand/utils/st_utils.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/stand/utils/st_utils.py b/stand/utils/st_utils.py index 0d576cd..117928c 100644 --- a/stand/utils/st_utils.py +++ b/stand/utils/st_utils.py @@ -1,12 +1,28 @@ import streamlit as st +def rerunall(): + server = st.server.server.Server.get_current() + for si in server._session_info_by_id.values(): + print("rerun:", si.session.id) + client_state = None + si.session.request_rerun(client_state) + + + +def get_session_ids(): + server = st.server.server.Server.get_current() + sids = server._session_info_by_id.keys() + return set(sids) + + + def rerun(session_id=None): if session_id is None: session_id = get_session_id() server = st.server.server.Server.get_current() - session = server._get_session_info(session_id).session + session = server.get_session_by_id(session_id) client_state = None session.request_rerun(client_state)