Function reload_state_module

reflex/state.py:2650–2684

Reset rx.State subclasses to avoid conflict when reloading. Args: module: The module to reload. state: Recursive argument for the state class to reload.

(
    module: str,
    state: type[BaseState] = State,
)

def reload_state_module(
    module: str,
    state: type[BaseState] = State,
) -> None:
    """Reset rx.State subclasses to avoid conflict when reloading.

    Args:
        module: The module to reload.
        state: Recursive argument for the state class to reload.

    """
    from reflex_base.registry import RegistrationContext

    # Reset the _app_ref of OnLoadInternalState to avoid stale references.
    if state is OnLoadInternalState:
        state._app_ref = None
    # Clean out all potentially dirty states of reloaded modules.
    for pd_state in tuple(state._potentially_dirty_states):
        with contextlib.suppress(ValueError):
            if (
                state.get_root_state().get_class_substate(pd_state).__module__ == module
                and module is not None
            ):
                state._potentially_dirty_states.remove(pd_state)
    reg_ctx = RegistrationContext.get()
    substates = reg_ctx.get_substates(state)
    for subclass in tuple(substates):
        reload_state_module(module=module, state=subclass)
        if subclass.__module__ == module and module is not None:
            all_base_state_classes.pop(subclass.get_full_name(), None)
            substates.remove(subclass)
            state._always_dirty_substates.discard(subclass.get_name())
            state._var_dependencies = {}
            state._init_var_dependency_dicts()
    state.get_class_substate.cache_clear()