Little fix for states

examples/custom_states.py

@@ -5,6 +5,12 @@ from telebot import custom_filters
 bot = telebot.TeleBot("")
 
 
+class MyStates:
+    name = 1
+    surname = 2
+    age = 3
+
+
 
 @bot.message_handler(commands=['start'])
 def start_ex(message):
@@ -24,7 +30,7 @@ def any_state(message):
     bot.send_message(message.chat.id, "Your state was cancelled.")
     bot.delete_state(message.chat.id)
 
-@bot.message_handler(state=1)
+@bot.message_handler(state=MyStates.name)
 def name_get(message):
     """
     State 1. Will process when user's state is 1.
@@ -35,7 +41,7 @@ def name_get(message):
         data['name'] = message.text
  
  
-@bot.message_handler(state=2)
+@bot.message_handler(state=MyStates.surname)
 def ask_age(message):
     """
     State 2. Will process when user's state is 2.
@@ -46,14 +52,14 @@ def ask_age(message):
         data['surname'] = message.text
  
 # result
-@bot.message_handler(state=3, is_digit=True)
+@bot.message_handler(state=MyStates.age, is_digit=True)
 def ready_for_answer(message):
     with bot.retrieve_data(message.chat.id) as data:
         bot.send_message(message.chat.id, "Ready, take a look:\n<b>Name: {name}\nSurname: {surname}\nAge: {age}</b>".format(name=data['name'], surname=data['surname'], age=message.text), parse_mode="html")
     bot.delete_state(message.chat.id)
 
 #incorrect number
-@bot.message_handler(state=3, is_digit=False)
+@bot.message_handler(state=MyStates.age, is_digit=False)
 def age_incorrect(message):
     bot.send_message(message.chat.id, 'Looks like you are submitting a string in the field age. Please enter a number')
 

tests/.state-save/states.pkl

@@ -0,0 +1 @@
+<EFBFBD>}<7D>.

tests/test_types.py

@@ -235,6 +235,7 @@ def test_json_chat_invite_link():
     assert invite_link.pending_join_request_count == 1
     assert invite_link.creates_join_request
 
+
 def test_chat_member_updated():
     json_string = r'{"chat": {"id": -1234567890123, "type": "supergroup", "title": "No Real Group", "username": "NoRealGroup"}, "from": {"id": 133869498, "is_bot": false, "first_name": "Vincent"}, "date": 1624119999, "old_chat_member": {"user": {"id": 77777777, "is_bot": false, "first_name": "Pepe"}, "status": "member"}, "new_chat_member": {"user": {"id": 77777777, "is_bot": false, "first_name": "Pepe"}, "status": "administrator"}}'
     cm_updated = types.ChatMemberUpdated.de_json(json_string)