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

inline memoize components in compiled page files, to prevent hot reload errors (#2527)

Browse Source
This commit is contained in:
jackie-pc 2024-02-05 12:31:06 -08:00 committed by GitHub
parent 1bf4e23bf3
commit 0f3857ec9a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
reflex/compiler/compiler.py
View File

@ -63,6 +63,10 @@ def _compile_theme(theme: dict) -> str:
return templates.THEME.render(theme=theme)
def _is_dev_mode() -> bool:
return os.environ.get("REFLEX_ENV_MODE", "dev") == "dev"
def _compile_contexts(state: Optional[Type[BaseState]]) -> str:
"""Compile the initial state and contexts.
@ -72,16 +76,15 @@ def _compile_contexts(state: Optional[Type[BaseState]]) -> str:
Returns:
The compiled context file.
"""
is_dev_mode = os.environ.get("REFLEX_ENV_MODE", "dev") == "dev"
return (
templates.CONTEXT.render(
initial_state=utils.compile_state(state),
state_name=state.get_name(),
client_storage=utils.compile_client_storage(state),
is_dev_mode=is_dev_mode,
is_dev_mode=_is_dev_mode(),
)
if state
else templates.CONTEXT.render(is_dev_mode=is_dev_mode)
else templates.CONTEXT.render(is_dev_mode=_is_dev_mode())
)
@ -238,7 +241,12 @@ def _compile_stateful_components(
# When the component is referenced by more than one page, render it
# to be included in the STATEFUL_COMPONENTS module.
if isinstance(component, StatefulComponent) and component.references > 1:
# Skip this step in dev mode, thereby avoiding potential hot reload errors for larger apps
if (
isinstance(component, StatefulComponent)
and component.references > 1
and not _is_dev_mode()
):
# Reset this flag to render the actual component.
component.rendered_as_shared = False