Fix onload events for substates (#473)

2023-02-08 06:55:12 +01:00 · 2023-02-08 06:55:12 +01:00 · f2d0fbbd8d
commit f2d0fbbd8d
parent 4ef2ba779c
1 changed files with 3 additions and 1 deletions
--- a/pynecone/middleware/hydrate_middleware.py
+++ b/pynecone/middleware/hydrate_middleware.py
@ -35,5 +35,7 @@ class HydrateMiddleware(Middleware):
            else:
                load_event = None
            if load_event:
-                load_event.fn(state)
+                substate_path = utils.format_event_handler(load_event).split(".")
+                ex_state = state.get_substate(substate_path[:-1])
+                load_event.fn(ex_state)
            return utils.format_state({state.get_name(): state.dict()})