Fix bug with unsaving data

eternnoir · Jun 4, 2022 · e689e96 · e689e96
1 parent 8da749e
commit e689e96
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()