aboutsummaryrefslogtreecommitdiff
path: root/components
diff options
context:
space:
mode:
authorTyler <tyler@tylerhoang.xyz>2026-05-17 01:01:04 -0700
committerTyler <tyler@tylerhoang.xyz>2026-05-17 01:01:04 -0700
commit9c681a2292b4ab7ca9d8975889fe78b6bf4e0305 (patch)
tree10e2d10ded75523b7e3de3c4ac925ff0e2979b62 /components
parent2ea1abbe88b2b4acb1c58a74022fcec42d02a4c2 (diff)
parent48f3bc1960d71785aad44292205845612a4f1cb8 (diff)
Merge feature/session-persistence: session persistence + watchlist exit button
- Persists watchlist and active ticker to localStorage across page reloads - Adds "← Watchlist" button in sidebar to return to landing page - Fixes write-before-read race that destroyed stored data on fresh load - Fixes click swallow caused by st.rerun() aborting render before click receivers were registered
Diffstat (limited to 'components')
-rw-r--r--components/persistence.py62
1 files changed, 62 insertions, 0 deletions
diff --git a/components/persistence.py b/components/persistence.py
new file mode 100644
index 0000000..552347d
--- /dev/null
+++ b/components/persistence.py
@@ -0,0 +1,62 @@
+import json
+import streamlit as st
+import streamlit.components.v1 as components
+from utils.security import escape_html
+
+
+def render_persistence_bridge() -> None:
+ """Sync watchlist + ticker to/from localStorage."""
+ restored_wl = st.session_state.get("_persist_wl", "")
+ restored_tk = st.session_state.get("_persist_tk", "")
+ if restored_wl and not st.session_state.get("_persist_loaded"):
+ try:
+ st.session_state["watchlist"] = json.loads(restored_wl)
+ except Exception:
+ pass
+ if restored_tk:
+ st.session_state["ticker"] = restored_tk
+ st.session_state["_persist_loaded"] = True
+ # No st.rerun() — the bridge runs at the top of the render cycle so the
+ # restored values propagate naturally to components below. Calling
+ # st.rerun() here would abort the render before click-receiver inputs
+ # (qt_click_receiver, wl_click_receiver) are registered, causing
+ # Streamlit to clear their pending values and swallow the click.
+
+ st.text_input("persist_wl", key="_persist_wl", label_visibility="collapsed")
+ st.text_input("persist_tk", key="_persist_tk", label_visibility="collapsed")
+
+ loaded = "1" if st.session_state.get("_persist_loaded") else "0"
+ wl_json = escape_html(json.dumps(st.session_state.get("watchlist", [])))
+ tk_val = escape_html(st.session_state.get("ticker") or "")
+
+ html = (
+ "<div id='d' data-loaded='" + loaded + "' data-wl='" + wl_json + "' data-tk='" + tk_val + "'></div>"
+ "<script>"
+ "(function() {"
+ " var d = document.getElementById('d');"
+ " function setInput(label, val) {"
+ " var inputs = window.parent.document.querySelectorAll('input[type=text]');"
+ " for (var i = 0; i < inputs.length; i++) {"
+ " if (inputs[i].getAttribute('aria-label') === label) {"
+ " var setter = Object.getOwnPropertyDescriptor(window.parent.HTMLInputElement.prototype, 'value').set;"
+ " setter.call(inputs[i], val);"
+ " inputs[i].dispatchEvent(new window.parent.Event('input', { bubbles: true }));"
+ " return;"
+ " }"
+ " }"
+ " }"
+ " if (d.dataset.loaded === '1') {"
+ " localStorage.setItem('prism_watchlist', d.dataset.wl);"
+ " if (d.dataset.tk) localStorage.setItem('prism_ticker', d.dataset.tk);"
+ " else localStorage.removeItem('prism_ticker');"
+ " } else {"
+ " var stored_wl = localStorage.getItem('prism_watchlist');"
+ " var stored_tk = localStorage.getItem('prism_ticker') || '';"
+ " if (!stored_wl) return;"
+ " setInput('persist_wl', stored_wl);"
+ " if (stored_tk) setInput('persist_tk', stored_tk);"
+ " }"
+ "})();"
+ "</script>"
+ )
+ components.html(html, height=0, scrolling=False)