b8b3f8910e
* add more type annotations through the code * add typing in reflex/utils * misc typing * more typings * state typing * keep typing * typing init and utils * more typing for components * fix attempt for 3.9 * need more __future * more typings * type event plz * type model * type vars/base.py * enable 'ANN001' for reflex folder (ignore tests and benchmarks) * fix pyi * add missing annotations * use more precise error when ignoring
28 lines
909 B
Python
28 lines
909 B
Python
"""Wrappers for the state manager."""
|
|
|
|
from typing import Any
|
|
|
|
from reflex.istate.proxy import ReadOnlyStateProxy
|
|
from reflex.state import _split_substate_key, _substate_key, get_state_manager
|
|
|
|
|
|
async def get_state(token: str, state_cls: Any | None = None) -> ReadOnlyStateProxy:
|
|
"""Get the instance of a state for a token.
|
|
|
|
Args:
|
|
token: The token for the state.
|
|
state_cls: The class of the state.
|
|
|
|
Returns:
|
|
A read-only proxy of the state instance.
|
|
"""
|
|
mng = get_state_manager()
|
|
if state_cls is not None:
|
|
root_state = await mng.get_state(_substate_key(token, state_cls))
|
|
else:
|
|
root_state = await mng.get_state(token)
|
|
_, state_path = _split_substate_key(token)
|
|
state_cls = root_state.get_class_substate(tuple(state_path.split(".")))
|
|
instance = await root_state.get_state(state_cls)
|
|
return ReadOnlyStateProxy(instance)
|