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

Fix States

Browse Source
This commit is contained in:
_run
2022-02-02 14:44:02 +04:00
parent 7d2915c7f9
commit cd4a9add68
9 changed files with 29 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
telebot/asyncio_storage/memory_storage.py
View File

@ -8,6 +8,8 @@ class StateMemoryStorage(StateStorageBase):
async def set_state(self, chat_id, user_id, state):
if isinstance(state, object):
state = state.name
if chat_id in self.data:
if user_id in self.data[chat_id]:
self.data[chat_id][user_id]['state'] = state

2
telebot/asyncio_storage/pickle_storage.py
View File

@ -47,6 +47,8 @@ class StatePickleStorage(StateStorageBase):
file.close()
async def set_state(self, chat_id, user_id, state):
if isinstance(state, object):
state = state.name
if chat_id in self.data:
if user_id in self.data[chat_id]:
self.data[chat_id][user_id]['state'] = state

2
telebot/asyncio_storage/redis_storage.py
View File

@ -65,6 +65,8 @@ class StateRedisStorage(StateStorageBase):
"""
response = await self.get_record(chat_id)
user_id = str(user_id)
if isinstance(state, object):
state = state.name
if response:
if user_id in response:
response[user_id]['state'] = state