laipeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

minor pickle improvement (#4499)

Browse Source
This commit is contained in:
benedikt-bartscher 2024-12-12 11:07:59 +01:00 committed by GitHub
parent 5e026e4b27
commit ea90a3ebfa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
reflex/state.py
View File

@ -2122,14 +2122,26 @@ class BaseState(Base, ABC, extra=pydantic.Extra.allow):
state["__dict__"].pop("router", None) state["__dict__"].pop("router", None)
state["__dict__"].pop("router_data", None) state["__dict__"].pop("router_data", None)
# Never serialize parent_state or substates. # Never serialize parent_state or substates.
state["__dict__"]["parent_state"] = None state["__dict__"].pop("parent_state", None)
state["__dict__"]["substates"] = {} state["__dict__"].pop("substates", None)
state["__dict__"].pop("_was_touched", None) state["__dict__"].pop("_was_touched", None)
# Remove all inherited vars. # Remove all inherited vars.
for inherited_var_name in self.inherited_vars: for inherited_var_name in self.inherited_vars:
state["__dict__"].pop(inherited_var_name, None) state["__dict__"].pop(inherited_var_name, None)
return state return state
def __setstate__(self, state: dict[str, Any]):
"""Set the state from redis deserialization.
This method is called by pickle to deserialize the object.
Args:
state: The state dict for deserialization.
"""
state["__dict__"]["parent_state"] = None
state["__dict__"]["substates"] = {}
super().__setstate__(state)
def _check_state_size( def _check_state_size(
self, self,
pickle_state_size: int, pickle_state_size: int,