From a84a4f8faf750215e3ea328ca439f458dcced335 Mon Sep 17 00:00:00 2001 From: Sven Augustin Date: Fri, 25 Nov 2022 23:50:03 +0100 Subject: [PATCH] refactor --- stand/utils/st_utils.py | 21 +++++++++++++-------- 1 file changed, 13 insertions(+), 8 deletions(-) diff --git a/stand/utils/st_utils.py b/stand/utils/st_utils.py index 117928c..f9549ba 100644 --- a/stand/utils/st_utils.py +++ b/stand/utils/st_utils.py @@ -1,39 +1,44 @@ import streamlit as st +get_server = st.server.server.Server.get_current + + def rerunall(): - server = st.server.server.Server.get_current() - for si in server._session_info_by_id.values(): + sibi = get_session_info_by_id() + for si in sibi.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() + sibi = get_session_info_by_id() + sids = sibi.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() + server = get_server() session = server.get_session_by_id(session_id) client_state = None session.request_rerun(client_state) - def get_session_id(): ctx = st.scriptrunner.script_run_context.get_script_run_ctx() return ctx.session_id +def get_session_info_by_id(): + server = get_server() + return server._session_info_by_id + + def hide_UI_elements(menu=True, header=True, footer=True): HIDDEN = " {visibility: hidden;}"