mirror of
https://github.com/aiogram/aiogram.git
synced 2025-12-09 17:33:44 +00:00
91 lines
2.5 KiB
Python
91 lines
2.5 KiB
Python
import asyncio
|
|
|
|
from aiogram import Bot, types
|
|
from aiogram.dispatcher import Dispatcher
|
|
from aiogram.dispatcher.state import StateMachine, Controller
|
|
|
|
API_TOKEN = 'BOT TOKEN HERE'
|
|
|
|
loop = asyncio.get_event_loop()
|
|
bot = Bot(token=API_TOKEN, loop=loop)
|
|
dp = Dispatcher(bot)
|
|
|
|
|
|
@dp.message_handler(commands=['start'])
|
|
async def send_welcome(message: types.Message):
|
|
"""
|
|
Entry point to conversation
|
|
"""
|
|
await message.reply("Hi there! What's your name?")
|
|
|
|
# Set state
|
|
# This method lock all messages from user and all messages will be redirected to next step handler in state machine
|
|
state.set_state(message.chat.id, message.from_user.id, "name")
|
|
|
|
|
|
async def process_name(message: types.Message, controller: Controller):
|
|
"""
|
|
Process user name
|
|
"""
|
|
# Save name to storage
|
|
controller["name"] = message.text
|
|
|
|
await message.reply("How old are you?")
|
|
|
|
# Go to next state
|
|
controller.set_state('age')
|
|
|
|
|
|
async def process_age(message: types.Message, controller: Controller):
|
|
# Check age. Age must be is digit
|
|
if not message.text.isdigit():
|
|
return await message.reply("Age should be a number.\nHow old are you?")
|
|
|
|
# Save age to storage
|
|
controller["age"] = int(message.text)
|
|
|
|
# Configure ReplyKeyboardMarkup
|
|
markup = types.ReplyKeyboardMarkup()
|
|
markup.add("Male", "Female")
|
|
markup.add("Other")
|
|
|
|
await message.reply("What is your gender?", reply_markup=markup)
|
|
|
|
# Go to next state
|
|
controller.set_state("sex")
|
|
|
|
|
|
async def process_sex(message: types.Message, controller: Controller):
|
|
# Check reply
|
|
if message.text not in ["Male", "Female", "Other"]:
|
|
return await message.reply("Bad gender name. Choose you gender from keyboard.")
|
|
|
|
# Save sex to storage
|
|
controller["sex"] = message.text
|
|
|
|
# Remove keyboard
|
|
markup = types.ReplyKeyboardRemove()
|
|
# And send message
|
|
await bot.send_message(message.chat.id,
|
|
f"Hi!\n"
|
|
f"Nice to meet you, {controller['name']}.\n"
|
|
f"Age: {controller['age']}\n"
|
|
f"Sex: {controller['sex']}",
|
|
reply_markup=markup)
|
|
|
|
# Finish conversation
|
|
controller.clear()
|
|
|
|
|
|
# Configure state machine
|
|
state = StateMachine(dp, {
|
|
"name": process_name,
|
|
"age": process_age,
|
|
"sex": process_sex
|
|
})
|
|
|
|
if __name__ == '__main__':
|
|
try:
|
|
loop.run_until_complete(dp.start_pooling())
|
|
except KeyboardInterrupt:
|
|
loop.stop()
|