From f7f8aa171b3100244b7ac8587c7eada0f32c7012 Mon Sep 17 00:00:00 2001 From: Elijah Ahianyo Date: Tue, 14 Nov 2023 18:26:25 +0000 Subject: [PATCH] Set is_hydrated var explicitly on rx.State (#2168) --- reflex/middleware/hydrate_middleware.py | 4 ---- reflex/state.py | 3 +++ 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/reflex/middleware/hydrate_middleware.py b/reflex/middleware/hydrate_middleware.py index 062ff33b3..3992919a2 100644 --- a/reflex/middleware/hydrate_middleware.py +++ b/reflex/middleware/hydrate_middleware.py @@ -13,9 +13,6 @@ if TYPE_CHECKING: from reflex.app import App -State.add_var(constants.CompileVars.IS_HYDRATED, type_=bool, default_value=False) - - class HydrateMiddleware(Middleware): """Middleware to handle initial app hydration.""" @@ -57,7 +54,6 @@ class HydrateMiddleware(Middleware): # Get the route for on_load events. route = event.router_data.get(constants.RouteVar.PATH, "") - # Add the on_load events and set is_hydrated to True. events = [*app.get_load_events(route), type(state).set_is_hydrated(True)] # type: ignore events = fix_events(events, event.token, router_data=event.router_data) diff --git a/reflex/state.py b/reflex/state.py index 99d907d3d..11bb5e9a4 100644 --- a/reflex/state.py +++ b/reflex/state.py @@ -194,6 +194,9 @@ class State(Base, ABC, extra=pydantic.Extra.allow): # The router data for the current page router: RouterData = RouterData() + # The hydrated bool. + is_hydrated: bool = False + def __init__(self, *args, parent_state: State | None = None, **kwargs): """Initialize the state.