make reflex state import dynamic (#6258)

* make reflex state import dynamic

* remove that import

Author: adhami3310

reflex/app.py

@@ -86,7 +86,7 @@
     readable_name_from_component,
 )
 from reflex.experimental.memo import EXPERIMENTAL_MEMOS
-from reflex.istate.manager import StateModificationContext
+from reflex.istate.manager import StateManager, StateModificationContext
 from reflex.istate.proxy import StateProxy
 from reflex.page import DECORATED_PAGES
 from reflex.route import (
@@ -98,7 +98,6 @@
     BaseState,
     RouterData,
     State,
-    StateManager,
     StateUpdate,
     _split_substate_key,
     _substate_key,

reflex/istate/wrappers.py

@@ -2,8 +2,9 @@
 
 from typing import Any
 
+from reflex.istate.manager import get_state_manager
 from reflex.istate.proxy import ReadOnlyStateProxy
-from reflex.state import _split_substate_key, _substate_key, get_state_manager
+from reflex.state import _split_substate_key, _substate_key
 
 
 async def get_state(token: str, state_cls: Any | None = None) -> ReadOnlyStateProxy:

reflex/reflex.py

@@ -141,7 +141,7 @@ def _run(
     """Run the app in the given directory."""
     import atexit
 
-    from reflex.state import reset_disk_state_manager
+    from reflex.istate.manager import reset_disk_state_manager
     from reflex.utils import build, exec, prerequisites, processes, telemetry
     from reflex.utils.exec import should_use_granian
 

reflex/state.py

@@ -1617,6 +1617,7 @@ async def _get_state_from_redis(self, state_cls: type[T_STATE]) -> T_STATE:
             RuntimeError: If redis is not used in this backend process.
             StateMismatchError: If the state instance is not of the expected type.
         """
+        from reflex.istate.manager import get_state_manager
         from reflex.istate.manager.redis import StateManagerRedis
 
         # Then get the target state and all its substates.
@@ -2863,10 +2864,3 @@ def reload_state_module(
             state._var_dependencies = {}
             state._init_var_dependency_dicts()
     state.get_class_substate.cache_clear()
-
-
-from reflex.istate.manager import StateManager as StateManager  # noqa: E402
-from reflex.istate.manager import get_state_manager as get_state_manager  # noqa: E402
-from reflex.istate.manager import (  # noqa: E402
-    reset_disk_state_manager as reset_disk_state_manager,
-)

reflex/testing.py

@@ -39,15 +39,11 @@
 import reflex.utils.prerequisites
 import reflex.utils.processes
 from reflex.experimental.memo import EXPERIMENTAL_MEMOS
+from reflex.istate.manager import StateManager
 from reflex.istate.manager.disk import StateManagerDisk
 from reflex.istate.manager.memory import StateManagerMemory
 from reflex.istate.manager.redis import StateManagerRedis
-from reflex.state import (
-    BaseState,
-    StateManager,
-    _split_substate_key,
-    reload_state_module,
-)
+from reflex.state import BaseState, _split_substate_key, reload_state_module
 from reflex.utils import console, js_runtimes
 from reflex.utils.export import export
 from reflex.utils.token_manager import TokenManager

tests/units/test_state_tree.py

@@ -7,8 +7,9 @@
 from reflex_core.constants.state import FIELD_MARKER
 
 import reflex as rx
+from reflex.istate.manager import StateManager
 from reflex.istate.manager.redis import StateManagerRedis
-from reflex.state import BaseState, StateManager, _substate_key
+from reflex.state import BaseState, _substate_key
 
 
 class Root(BaseState):