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 5

add stategroup filter (#659)

Browse source
This commit is contained in:
darksidecat 2021-08-17 00:09:11 +03:00 • committed by GitHub
parent 71eb5fc44e
commit 04bbc8211c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 41 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

11
aiogram/dispatcher/fsm/state.py
View file

@ -111,8 +111,8 @@ class StatesGroupMeta(type):
return item in cls.__all_states_names__
if isinstance(item, State):
return item in cls.__all_states__
# if isinstance(item, StatesGroup):
# return item in cls.__all_childs__
if isinstance(item, StatesGroupMeta):
return item in cls.__all_childs__
return False
def __str__(self) -> str:
@ -126,8 +126,11 @@ class StatesGroup(metaclass=StatesGroupMeta):
return cls
return cls.__parent__.get_root()
# def __call__(cls, event: TelegramObject, raw_state: Optional[str] = None) -> bool:
# return raw_state in cls.__all_states_names__
def __call__(cls, event: TelegramObject, raw_state: Optional[str] = None) -> bool:
return raw_state in type(cls).__all_states_names__
def __str__(self) -> str:
return f"StatesGroup {type(self).__full_group_name__}"
default_state = State()