Skip to content

Commit d466da3

Browse files
committed
Add state_list to StatesGroup
1 parent 1797f07 commit d466da3

File tree

2 files changed

+11
-1
lines changed

2 files changed

+11
-1
lines changed

telebot/asyncio_handler_backends.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,11 @@ def __init_subclass__(cls) -> None:
8181
value.name = ':'.join((cls.__name__, name))
8282
value.group = cls
8383

84+
@property
85+
def state_list(self):
86+
return [value for name, value in self.__dict__.items()
87+
if not name.startswith('__') and not callable(value) and isinstance(value, State)]
88+
8489

8590
class SkipHandler:
8691
"""

telebot/handler_backends.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -191,7 +191,12 @@ def __init_subclass__(cls) -> None:
191191
value.name = ':'.join((cls.__name__, name))
192192
value.group = cls
193193

194-
194+
@property
195+
def state_list(self):
196+
return [value for name, value in self.__dict__.items()
197+
if not name.startswith('__') and not callable(value) and isinstance(value, State)]
198+
199+
195200
class BaseMiddleware:
196201
"""
197202
Base class for middleware.

0 commit comments

Comments
 (0)