mirror
/
pyTelegramBotAPI
mirror of https://github.com/eternnoir/pyTelegramBotAPI.git
1
0
Fork 0
Code Releases Activity

Define state_list in __init_subclass__

This commit is contained in:
Cub11k 2023-04-14 22:11:08 +03:00
parent d466da3542
commit e4bddd91cb
2 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
telebot/asyncio_handler_backends.py
View File

@ -74,17 +74,18 @@ class StatesGroup:
my_state = State() # returns my_state:State string.
"""
def __init_subclass__(cls) -> None:
state_list = []
for name, value in cls.__dict__.items():
if not name.startswith('__') and not callable(value) and isinstance(value, State):
# change value of that variable
value.name = ':'.join((cls.__name__, name))
value.group = cls
state_list.append(value)
cls._state_list = state_list
@property
def state_list(self):
return [value for name, value in self.__dict__.items()
if not name.startswith('__') and not callable(value) and isinstance(value, State)]
return self._state_list
class SkipHandler:

6
telebot/handler_backends.py
View File

@ -185,16 +185,18 @@ class StatesGroup:
my_state = State() # returns my_state:State string.
"""
def __init_subclass__(cls) -> None:
state_list = []
for name, value in cls.__dict__.items():
if not name.startswith('__') and not callable(value) and isinstance(value, State):
# change value of that variable
value.name = ':'.join((cls.__name__, name))
value.group = cls
state_list.append(value)
cls._state_list = state_list
@property
def state_list(self):
return [value for name, value in self.__dict__.items()
if not name.startswith('__') and not callable(value) and isinstance(value, State)]
return self._state_list
class BaseMiddleware: