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/storage/pickle_storage.py
View File

@ -1,6 +1,6 @@
from telebot.storage.base_storage import StateStorageBase, StateContext
import os
from telebot.handler_backends import State
import pickle
@ -53,7 +53,7 @@ class StatePickleStorage(StateStorageBase):
file.close()
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]: