Allow to store data in storage.

examples/state_machine.py

@@ -6,7 +6,6 @@ from aiogram.dispatcher import Dispatcher
 from aiogram.dispatcher.state import StateMachine
 
 API_TOKEN = 'BOT TOKEN HERE'
-API_TOKEN = '380294876:AAFbdYYgq1hBi9hQDcxD3bj8QCNnVec5aHk'
 
 logging.basicConfig(level=logging.DEBUG)
 
@@ -14,8 +13,6 @@ loop = asyncio.get_event_loop()
 bot = Bot(token=API_TOKEN, loop=loop)
 dp = Dispatcher(bot)
 
-users = {}
-
 
 @dp.message_handler(commands=['start'])
 async def send_welcome(message: types.Message):
@@ -24,39 +21,41 @@ async def send_welcome(message: types.Message):
 
 
 async def process_name(message, controller):
-    users[message.from_user.id] = {"name": message.text}
+    controller["name"] = message.text
 
     await message.reply("How old are you?")
 
-    controller.set('age')
+    controller.set_state('age')
 
 
 async def process_age(message, controller):
     if not message.text.isdigit():
         return await message.reply("Age should be a number.\nHow old are you?")
 
-    users[message.from_user.id].update({"age": int(message.text)})
+    controller["age"] = int(message.text)
 
     markup = types.ReplyKeyboardMarkup()
     markup.add("Male", "Female")
     markup.add("Other")
     await message.reply("What is your gender?", reply_markup=markup)
-    controller.set("sex")
+
+    controller.set_state("sex")
 
 
 async def process_sex(message, controller):
     if message.text not in ["Male", "Female", "Other"]:
         return await message.reply("Bad gender name. Choose you gender from keyboard.")
 
-    users[message.from_user.id].update({"sex": message.text})
-    controller.clear()
-
-    user = users[message.from_user.id]
+    controller["sex"] = message.text
 
     markup = types.ReplyKeyboardRemove()
     await bot.send_message(message.chat.id,
-                           f"Hi!\nNice to meet you, {user['name']}.\nAge: {user['age']}\nSex: {user['sex']}",
+                           f"Hi!\n"
+                           f"Nice to meet you, {controller['name']}.\n"
+                           f"Age: {controller['age']}\n"
+                           f"Sex: {controller['sex']}",
                            reply_markup=markup)
+    controller.clear()
 
 
 state = StateMachine(dp, {