Check if router is ready on page load (reflex-dev#241)

ACucos1 · Feb 15, 2023 · 482bf82 · 482bf82
1 parent a6c81cd
commit 482bf82
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/pynecone/compiler/templates.py b/pynecone/compiler/templates.py
@@ -158,9 +158,12 @@ def format_state(
 STATE = constants.STATE
 EVENTS = constants.EVENTS
 SET_RESULT = format_state_setter(RESULT)
+READY = f"const {{ isReady }} = {ROUTER};"
 USE_EFFECT = join(
     [
         "useEffect(() => {{",
+        "  if(!isReady)",
+        f"    return;",
         f"  if (!{SOCKET}.current) {{{{",
         f"    connect({SOCKET}, {{state}}, {RESULT}, {SET_RESULT}, {ROUTER}, {EVENT_ENDPOINT})",
         "  }}",

diff --git a/pynecone/compiler/utils.py b/pynecone/compiler/utils.py
@@ -119,7 +119,8 @@ def compile_state(state: Type[State]) -> str:
     )
     router = templates.ROUTER
     socket = templates.SOCKET
-    return templates.join([synced_state, result, router, socket])
+    ready = templates.READY
+    return templates.join([synced_state, result, router, socket, ready])
 
 
 def compile_events(state: Type[State]) -> str: