python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Precalculate StateGroup values

Browse source
This commit is contained in:
Oleg A 2024-06-11 02:17:57 +03:00
parent 69c359d23a
commit b05f1a3cf0
No known key found for this signature in database
GPG key ID: 5FE046817A9657C5
1 changed files with 13 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

19
aiogram/fsm/state.py
View file

@ -70,6 +70,9 @@ class StatesGroupMeta(type):
__childs__: "Tuple[Type[StatesGroup], ...]"
__states__: Tuple[State, ...]
__state_names__: Tuple[str, ...]
__all_childs__: Tuple[Type["StatesGroup"], ...]
__all_states__: Tuple[State, ...]
__all_states_names__: Tuple[str, ...]
@no_type_check
def __new__(mcs, name, bases, namespace, **kwargs):
@ -84,12 +87,14 @@ class StatesGroupMeta(type):
elif inspect.isclass(arg) and issubclass(arg, StatesGroup):
childs.append(arg)
arg.__parent__ = cls
arg.__update_all_values__()
cls.__parent__ = None
cls.__childs__ = tuple(childs)
cls.__states__ = tuple(states)
cls.__state_names__ = tuple(state.state for state in states)
cls.__update_all_values__()
return cls
@property
@ -98,22 +103,24 @@ class StatesGroupMeta(type):
return ".".join((cls.__parent__.__full_group_name__, cls.__name__))
return cls.__name__
@property
def __all_childs__(cls) -> Tuple[Type["StatesGroup"], ...]:
def __update_all_values__(cls) -> None:
cls.__all_childs__ = cls.__get_all_childs__()
cls.__all_states__ = cls.__get_all_states__()
cls.__all_states_names__ = cls.__get_all_states_names__()
def __get_all_childs__(cls) -> Tuple[Type["StatesGroup"], ...]:
result = cls.__childs__
for child in cls.__childs__:
result += child.__childs__
return result
@property
def __all_states__(cls) -> Tuple[State, ...]:
def __get_all_states__(cls) -> Tuple[State, ...]:
result = cls.__states__
for group in cls.__childs__:
result += group.__all_states__
return result
@property
def __all_states_names__(cls) -> Tuple[str, ...]:
def __get_all_states_names__(cls) -> Tuple[str, ...]:
return tuple(state.state for state in cls.__all_states__ if state.state)
def __contains__(cls, item: Any) -> bool: