Изменение БД, добавление регистрации, добавление сложной рассылки, кнопки для СР

handlers/user.py

@@ -1,5 +1,7 @@
 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
@@ -7,27 +9,51 @@ 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):
-    async with async_session() as session:
+async def start_handler(message: Message, state: FSMContext):
+    user_id = message.from_user.id
 
-        result = await session.execute(select(User).where(User.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 not user:
-            await session.execute(insert(User).values(user_id=message.from_user.id))
-            await session.commit()
-        text = "Добро пожаловать! Управляйте подпиской:"
-        await message.answer(text, reply_markup=get_subscription_keyboard(user.is_subscribed if user else True))
 
-@router.callback_query(F.data.in_(["subscribe", "unsubscribe"]))
-async def toggle_subscription(callback: CallbackQuery):
-    subscribe = callback.data == "subscribe"
+        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(
-            update(User)
-            .where(User.user_id == callback.from_user.id)
-            .values(is_subscribed=subscribe)
-        )
+        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 callback.message.edit_reply_markup(reply_markup=get_subscription_keyboard(subscribe))
-    await callback.answer("Вы подписались!" if subscribe else "Вы отписались.")
+
+    await message.answer(f"✅ Регистрация завершена!\nФИО: {full_name}\nГруппа: {study_group}")
+    await message.answer("Теперь ты можешь управлять подпиской ниже 👇", reply_markup=get_subscription_keyboard(True))
+    await state.clear()