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

Added inherited backend vars (#616)

Browse Source
This commit is contained in:
Lucas 2023-02-28 21:29:16 +00:00 committed by GitHub
parent 55ac6e4973
commit f06de5289e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
pynecone/state.py
View File

@ -47,6 +47,9 @@ class State(Base, ABC):
# Backend vars that are never sent to the client. # Backend vars that are never sent to the client.
backend_vars: ClassVar[Dict[str, Any]] = {} backend_vars: ClassVar[Dict[str, Any]] = {}
# Backend vars inherited
inherited_backend_vars: ClassVar[Dict[str, Any]] = {}
# The parent state. # The parent state.
parent_state: Optional[State] = None parent_state: Optional[State] = None
@ -130,13 +133,17 @@ class State(Base, ABC):
parent_state = cls.get_parent_state() parent_state = cls.get_parent_state()
if parent_state is not None: if parent_state is not None:
cls.inherited_vars = parent_state.vars cls.inherited_vars = parent_state.vars
cls.inherited_backend_vars = parent_state.backend_vars
cls.backend_vars = { cls.new_backend_vars = {
name: value name: value
for name, value in cls.__dict__.items() for name, value in cls.__dict__.items()
if utils.is_backend_variable(name) if utils.is_backend_variable(name)
and name not in cls.inherited_backend_vars
} }
cls.backend_vars = {**cls.inherited_backend_vars, **cls.new_backend_vars}
# Set the base and computed vars. # Set the base and computed vars.
skip_vars = set(cls.inherited_vars) | { skip_vars = set(cls.inherited_vars) | {
"parent_state", "parent_state",
@ -450,8 +457,11 @@ class State(Base, ABC):
Returns: Returns:
The value of the var. The value of the var.
""" """
# Get the var from the parent state. inherited_vars = {
if name in super().__getattribute__("inherited_vars"): **super().__getattribute__("inherited_vars"),
**super().__getattribute__("inherited_backend_vars"),
}
if name in inherited_vars:
return getattr(super().__getattribute__("parent_state"), name) return getattr(super().__getattribute__("parent_state"), name)
elif name in super().__getattribute__("backend_vars"): elif name in super().__getattribute__("backend_vars"):
return super().__getattribute__("backend_vars").__getitem__(name) return super().__getattribute__("backend_vars").__getitem__(name)
@ -467,7 +477,8 @@ class State(Base, ABC):
value: The value of the attribute. value: The value of the attribute.
""" """
# Set the var on the parent state. # Set the var on the parent state.
if name in self.inherited_vars: inherited_vars = {**self.inherited_vars, **self.inherited_backend_vars}
if name in inherited_vars:
setattr(self.parent_state, name, value) setattr(self.parent_state, name, value)
return return