From fec47cecafa511c25cec591547cda13fd933f9b4 Mon Sep 17 00:00:00 2001 From: _run Date: Wed, 29 Jun 2022 20:21:42 +0500 Subject: [PATCH] Add the possibility to getbase class of a State object --- telebot/asyncio_handler_backends.py | 1 + telebot/handler_backends.py | 4 ++++ 2 files changed, 5 insertions(+) diff --git a/telebot/asyncio_handler_backends.py b/telebot/asyncio_handler_backends.py index e5e2e4f..4ef4555 100644 --- a/telebot/asyncio_handler_backends.py +++ b/telebot/asyncio_handler_backends.py @@ -29,6 +29,7 @@ class StatesGroup: 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 class SkipHandler: diff --git a/telebot/handler_backends.py b/telebot/handler_backends.py index b838231..4304317 100644 --- a/telebot/handler_backends.py +++ b/telebot/handler_backends.py @@ -163,6 +163,10 @@ class StatesGroup: 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 + + + class BaseMiddleware: