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

fix pydantic _get_value for MutableProxy

Browse Source
This commit is contained in:
Benedikt Bartscher 2024-02-29 22:44:46 +01:00
parent 7789118af3
commit 4380cd266d
No known key found for this signature in database
1 changed files with 23 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
reflex/state.py
View File

@ -1609,7 +1609,22 @@ class BaseState(Base, ABC, extra=pydantic.Extra.allow):
self.dirty_vars = set() self.dirty_vars = set()
self.dirty_substates = set() self.dirty_substates = set()
def get_value(self, key: str) -> Any: def get_value(self, value: Any) -> Any:
"""Get the value of a field (without proxying).
The returned value will NOT track dirty state updates.
Args:
value: The value of the field.
Returns:
The value of the field.
"""
if isinstance(value, MutableProxy):
return super().get_value(value.__wrapped__)
return super().get_value(value)
def _get_value(self, value: Any, **kwargs) -> Any:
"""Get the value of a field (without proxying). """Get the value of a field (without proxying).
The returned value will NOT track dirty state updates. The returned value will NOT track dirty state updates.
@ -1620,9 +1635,9 @@ class BaseState(Base, ABC, extra=pydantic.Extra.allow):
Returns: Returns:
The value of the field. The value of the field.
""" """
if isinstance(key, MutableProxy): if isinstance(value, MutableProxy):
return super().get_value(key.__wrapped__) return super()._get_value(value.__wrapped__, **kwargs)
return super().get_value(key) return super()._get_value(value, **kwargs)
def dict( def dict(
self, include_computed: bool = True, initial: bool = False, **kwargs self, include_computed: bool = True, initial: bool = False, **kwargs
@ -2085,6 +2100,10 @@ class StateManagerMemory(StateManager):
"_states_locks": {"exclude": True}, "_states_locks": {"exclude": True},
} }
# json_encoders = {
# MutableProxy: lambda v: v.__wrapped__,
# }
async def get_state(self, token: str) -> BaseState: async def get_state(self, token: str) -> BaseState:
"""Get the state for a token. """Get the state for a token.