gaifutar

from aiogram import Bot, Dispatcher, types, executor
from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton
from aiogram.dispatcher.filters import Command
from aiogram.dispatcher.filters.state import StatesGroup, State
from aiogram.dispatcher import FSMContext
from aiogram.contrib.fsm_storage.memory import MemoryStorage
from API_KEY_1196494 import API_KEY

# making a class for a state to move from one state to another later and tie commands to a current state
class reg_state (StatesGroup):
    activity = State()
    full_name = State()
    phone = State()

# button names content
button1_text = 'Activity 1' 
button2_text = 'Activity 2'

# buttons
button1 = InlineKeyboardButton (text = button1_text, callback_data='activity1')
button2 = InlineKeyboardButton (text = button2_text, callback_data='activity2')

#keyboard
keyboard_inline = InlineKeyboardMarkup().add(button1).add(button2)

# answers
enter_your_name_details = 'Enter your full name'
enter_your_phone = "Enter your phone number"

bot = Bot(API_KEY)
storage = MemoryStorage()
dp = Dispatcher(bot, storage = storage)


# job initiated, asking to choose activity
@dp.message_handler(commands=['start'])
async def activity_reg (message: types.Message):
    user_name_greeting = message.from_user.first_name
    await message.answer (text = f"{user_name_greeting}, Which activity do you want to choose?", reply_markup=keyboard_inline)

@dp.callback_query_handler(lambda call: call.data in ['activity1', 'activity2'])
async def inline_buttons_handler(call: types.CallbackQuery, state: FSMContext):
    await bot.answer_callback_query(callback_query_id=call.id)
    await state.update_data(activity=call.data)
    await reg_state.full_name.set()
    await call.message.answer (enter_your_name_details)


# grabbing the info about the user's name and asking for phone
@dp.message_handler(state=reg_state.full_name)
async def grab_full_name (message: types.Message, state: FSMContext):
    await state.update_data(full_name = message.text)
    await message.answer (enter_your_phone)
    await reg_state.phone.set()

# grabbing user's phone info and wrapping it up
@dp.message_handler (state = reg_state.phone)
async def grab_phone (message: types.Message, state: FSMContext):
    await state.update_data (phone = message.text)
    data_full = await state.get_data()
    await message.answer(f"Activity: {data_full['activity']}'\n"
                         f"Full name: {data_full['full_name']}\n"
                         f"Phone: {data_full['phone']}")
    await state.finish()



if __name__ == '__main__':
    executor.start_polling(dp)