mirror of
https://github.com/ada-dmitry/vol_bot.git
synced 2026-09-24 00:30:19 +00:00
60 lines
2.6 KiB
Python
60 lines
2.6 KiB
Python
from aiogram import Router, F
|
|
from aiogram.fsm.context import FSMContext
|
|
from aiogram.types import Message, CallbackQuery
|
|
from aiogram.fsm.state import State, StatesGroup
|
|
from sqlalchemy import select, insert, update
|
|
from models import User
|
|
from database import async_session
|
|
from keyboards import get_subscription_keyboard
|
|
|
|
router = Router()
|
|
|
|
# Машина состояний
|
|
class Register(StatesGroup):
|
|
waiting_for_fullname = State()
|
|
waiting_for_study_group = State()
|
|
|
|
@router.message(F.text == "/start")
|
|
async def start_handler(message: Message, state: FSMContext):
|
|
user_id = message.from_user.id
|
|
|
|
async with async_session() as session:
|
|
result = await session.execute(select(User).where(User.user_id == user_id))
|
|
user = result.scalar_one_or_none()
|
|
|
|
if user:
|
|
# Приветствие при повторном запуске
|
|
text = f"👋 Привет, {user.full_name or 'друг'}!\n" \
|
|
f"Группа: {user.study_group or 'не указана'}\n" \
|
|
"Используй кнопки ниже, чтобы управлять подпиской:"
|
|
await message.answer(text, reply_markup=get_subscription_keyboard(user.is_subscribed))
|
|
else:
|
|
# Первый запуск — регистрация
|
|
await state.set_state(Register.waiting_for_fullname)
|
|
await message.answer("👋 Привет! Пожалуйста, напиши свои ФИО одной строкой:")
|
|
|
|
@router.message(Register.waiting_for_fullname)
|
|
async def get_fullname(message: Message, state: FSMContext):
|
|
await state.update_data(full_name=message.text)
|
|
await state.set_state(Register.waiting_for_study_group)
|
|
await message.answer("Теперь укажи номер своей группы (например, Б11-111):")
|
|
|
|
@router.message(Register.waiting_for_study_group)
|
|
async def get_study_group(message: Message, state: FSMContext):
|
|
data = await state.get_data()
|
|
full_name = data["full_name"]
|
|
study_group = message.text
|
|
|
|
async with async_session() as session:
|
|
await session.execute(insert(User).values(
|
|
user_id=message.from_user.id,
|
|
full_name=full_name,
|
|
study_group=study_group,
|
|
is_subscribed=True
|
|
))
|
|
await session.commit()
|
|
|
|
await message.answer(f"✅ Регистрация завершена!\nФИО: {full_name}\nГруппа: {study_group}")
|
|
await message.answer("Теперь ты можешь управлять подпиской ниже 👇", reply_markup=get_subscription_keyboard(True))
|
|
await state.clear()
|