Merge pull request #1575 from coder2020official/bugfixes

Fix bug with unsaving data
eternnoir · Jun 4, 2022 · a21ab20 · a21ab20
2 parents 808810e + e689e96
commit a21ab20
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/telebot/asyncio_storage/pickle_storage.py b/telebot/asyncio_storage/pickle_storage.py
@@ -51,9 +51,11 @@ async def set_state(self, chat_id, user_id, state):
         if chat_id in self.data:
             if user_id in self.data[chat_id]:
                 self.data[chat_id][user_id]['state'] = state
+                self.update_data()
                 return True
             else:
                 self.data[chat_id][user_id] = {'state': state, 'data': {}}
+                self.update_data()
                 return True
         self.data[chat_id] = {user_id: {'state': state, 'data': {}}}
         self.update_data()

diff --git a/telebot/storage/pickle_storage.py b/telebot/storage/pickle_storage.py
@@ -57,9 +57,11 @@ def set_state(self, chat_id, user_id, state):
         if chat_id in self.data:
             if user_id in self.data[chat_id]:
                 self.data[chat_id][user_id]['state'] = state
+                self.update_data()
                 return True
             else:
                 self.data[chat_id][user_id] = {'state': state, 'data': {}}
+                self.update_data()
                 return True
         self.data[chat_id] = {user_id: {'state': state, 'data': {}}}
         self.update_data()