mirror/pyTelegramBotAPI
1
0
Fork 0
mirror of https://github.com/eternnoir/pyTelegramBotAPI.git synced 2023-08-10 21:12:57 +03:00
Code Releases Activity

Allow only state objects

Browse Source
This commit is contained in:
coder2020official
2022-05-02 02:08:48 +05:00
parent 825827cb1e
commit 836130a718
6 changed files with 15 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
telebot/asyncio_storage/memory_storage.py
View File

@ -1,5 +1,7 @@
from telebot.asyncio_storage.base_storage import StateStorageBase, StateContext
from telebot.asyncio_handler_backends import State
class StateMemoryStorage(StateStorageBase):
def __init__(self) -> None:
self.data = {}
@ -8,7 +10,7 @@ class StateMemoryStorage(StateStorageBase):
async def set_state(self, chat_id, user_id, state):
if isinstance(state, object):
if isinstance(state, State):
state = state.name
if chat_id in self.data:
if user_id in self.data[chat_id]: