refactor

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;}"