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

Use ._is_mutable() to account for parent state proxy (#3739)

Browse Source 
When a parent state proxy is set, also allow child StateProxy._self_mutable to
override the parent's `_is_mutable()`.
This commit is contained in:
Masen Furer 2024-08-05 14:03:49 -07:00 committed by GitHub
parent 7d9ed7e2ce
commit 3309c0e533
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
reflex/state.py
View File

@ -2059,7 +2059,7 @@ class StateProxy(wrapt.ObjectProxy):
Whether the state is mutable. Whether the state is mutable.
""" """
if self._self_parent_state_proxy is not None: if self._self_parent_state_proxy is not None:
return self._self_parent_state_proxy._is_mutable() return self._self_parent_state_proxy._is_mutable() or self._self_mutable
return self._self_mutable return self._self_mutable
async def __aenter__(self) -> StateProxy: async def __aenter__(self) -> StateProxy:
@ -3302,7 +3302,7 @@ class ImmutableMutableProxy(MutableProxy):
Raises: Raises:
ImmutableStateError: if the StateProxy is not mutable. ImmutableStateError: if the StateProxy is not mutable.
""" """
if not self._self_state._self_mutable: if not self._self_state._is_mutable():
raise ImmutableStateError( raise ImmutableStateError(
"Background task StateProxy is immutable outside of a context " "Background task StateProxy is immutable outside of a context "
"manager. Use `async with self` to modify state." "manager. Use `async with self` to modify state."