236 lines
9.4 KiB
Python
236 lines
9.4 KiB
Python
from aiogram import Router, F
|
||
from aiogram.filters import CommandStart, Command
|
||
from aiogram.types import (
|
||
Message,
|
||
InlineKeyboardMarkup,
|
||
InlineKeyboardButton,
|
||
CallbackQuery,
|
||
ReplyKeyboardMarkup,
|
||
KeyboardButton,
|
||
)
|
||
from aiogram.fsm.context import FSMContext
|
||
|
||
from database.db import async_session_maker
|
||
from services.user_service import UserService
|
||
from utils.i18n import t
|
||
|
||
router = Router()
|
||
|
||
def main_menu_keyboard(lang: str = 'ru') -> ReplyKeyboardMarkup:
|
||
"""Клавиатура с основными командами (кнопки отправляют команды)."""
|
||
return ReplyKeyboardMarkup(
|
||
resize_keyboard=True,
|
||
keyboard=[
|
||
[
|
||
KeyboardButton(text=t(lang, "menu.add")),
|
||
KeyboardButton(text=t(lang, "menu.vocab")),
|
||
],
|
||
[
|
||
KeyboardButton(text=t(lang, "menu.task")),
|
||
KeyboardButton(text=t(lang, "menu.practice")),
|
||
],
|
||
[
|
||
KeyboardButton(text=t(lang, "menu.words")),
|
||
KeyboardButton(text=t(lang, "menu.import")),
|
||
],
|
||
[
|
||
KeyboardButton(text=t(lang, "menu.stats")),
|
||
KeyboardButton(text=t(lang, "menu.settings")),
|
||
],
|
||
],
|
||
)
|
||
|
||
|
||
@router.message(CommandStart())
|
||
async def cmd_start(message: Message, state: FSMContext):
|
||
"""Обработчик команды /start"""
|
||
async with async_session_maker() as session:
|
||
# Проверяем, существует ли пользователь
|
||
existing_user = await UserService.get_user_by_telegram_id(session, message.from_user.id)
|
||
is_new_user = existing_user is None
|
||
|
||
# Создаём или получаем пользователя
|
||
user = await UserService.get_or_create_user(
|
||
session,
|
||
telegram_id=message.from_user.id,
|
||
username=message.from_user.username
|
||
)
|
||
|
||
lang = (user.language_interface or 'ru')
|
||
|
||
if is_new_user:
|
||
# Новый пользователь
|
||
await message.answer(
|
||
t(lang, "start.new_intro", first_name=message.from_user.first_name),
|
||
reply_markup=main_menu_keyboard(lang),
|
||
)
|
||
|
||
# Предлагаем пройти тест уровня
|
||
keyboard = InlineKeyboardMarkup(inline_keyboard=[
|
||
[InlineKeyboardButton(text=t(lang, 'start.offer_btn'), callback_data="offer_level_test")],
|
||
[InlineKeyboardButton(text=t(lang, 'start.skip_btn'), callback_data="skip_level_test")]
|
||
])
|
||
|
||
await message.answer(t(lang, "start.offer_test"), reply_markup=keyboard)
|
||
else:
|
||
# Существующий пользователь
|
||
await message.answer(
|
||
t(lang, "start.return", first_name=message.from_user.first_name),
|
||
reply_markup=main_menu_keyboard(lang),
|
||
)
|
||
|
||
|
||
@router.message(Command("menu"))
|
||
async def cmd_menu(message: Message):
|
||
"""Показать клавиатуру с основными командами."""
|
||
# Определяем язык пользователя
|
||
async with async_session_maker() as session:
|
||
user = await UserService.get_user_by_telegram_id(session, message.from_user.id)
|
||
lang = (user.language_interface if user else 'ru') or 'ru'
|
||
await message.answer(t(lang, "menu.below"), reply_markup=main_menu_keyboard(lang))
|
||
|
||
|
||
# Обработчики кнопок главного меню (по тексту)
|
||
|
||
def _menu_match(key: str):
|
||
labels = {t('ru', key), t('en', key), t('ja', key)}
|
||
return lambda m: m.text in labels
|
||
|
||
|
||
@router.message(_menu_match('menu.add'))
|
||
async def btn_add_pressed(message: Message, state: FSMContext):
|
||
from bot.handlers.vocabulary import AddWordStates
|
||
async with async_session_maker() as session:
|
||
user = await UserService.get_user_by_telegram_id(session, message.from_user.id)
|
||
lang = (user.language_interface if user else 'ru') or 'ru'
|
||
await message.answer(t(lang, 'add.prompt'))
|
||
await state.set_state(AddWordStates.waiting_for_word)
|
||
|
||
|
||
@router.message(_menu_match('menu.vocab'))
|
||
async def btn_vocab_pressed(message: Message):
|
||
from bot.handlers.vocabulary import cmd_vocabulary
|
||
await cmd_vocabulary(message)
|
||
|
||
|
||
@router.message(_menu_match('menu.task'))
|
||
async def btn_task_pressed(message: Message, state: FSMContext):
|
||
from bot.handlers.tasks import cmd_task
|
||
await cmd_task(message, state)
|
||
|
||
|
||
@router.message(_menu_match('menu.practice'))
|
||
async def btn_practice_pressed(message: Message, state: FSMContext):
|
||
from bot.handlers.practice import cmd_practice
|
||
await cmd_practice(message, state)
|
||
|
||
|
||
@router.message(_menu_match('menu.import'))
|
||
async def btn_import_pressed(message: Message, state: FSMContext):
|
||
from bot.handlers.import_text import cmd_import
|
||
await cmd_import(message, state)
|
||
|
||
|
||
@router.message(_menu_match('menu.stats'))
|
||
async def btn_stats_pressed(message: Message):
|
||
from bot.handlers.tasks import cmd_stats
|
||
await cmd_stats(message)
|
||
|
||
|
||
@router.message(_menu_match('menu.settings'))
|
||
async def btn_settings_pressed(message: Message):
|
||
from bot.handlers.settings import cmd_settings
|
||
await cmd_settings(message)
|
||
|
||
|
||
@router.message(_menu_match('menu.words'))
|
||
async def btn_words_pressed(message: Message, state: FSMContext):
|
||
"""Подсказать про тематические слова и показать быстрые темы."""
|
||
from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton
|
||
|
||
async with async_session_maker() as session:
|
||
user = await UserService.get_user_by_telegram_id(session, message.from_user.id)
|
||
lang = (user.language_interface if user else 'ru') or 'ru'
|
||
text = (
|
||
t(lang, 'words.help_title') + "\n\n" +
|
||
t(lang, 'words.help_usage') + "\n\n" +
|
||
t(lang, 'words.popular')
|
||
)
|
||
keyboard = InlineKeyboardMarkup(inline_keyboard=[
|
||
[InlineKeyboardButton(text=t(lang, 'words.topic_travel'), callback_data="menu_theme_travel")],
|
||
[InlineKeyboardButton(text=t(lang, 'words.topic_food'), callback_data="menu_theme_food")],
|
||
[InlineKeyboardButton(text=t(lang, 'words.topic_work'), callback_data="menu_theme_work")],
|
||
[InlineKeyboardButton(text=t(lang, 'words.topic_nature'), callback_data="menu_theme_nature")],
|
||
[InlineKeyboardButton(text=t(lang, 'words.topic_technology'), callback_data="menu_theme_technology")],
|
||
])
|
||
await message.answer(text, reply_markup=keyboard)
|
||
|
||
|
||
@router.callback_query(F.data.startswith("menu_theme_"))
|
||
async def pick_theme_from_menu(callback: CallbackQuery, state: FSMContext):
|
||
"""Сгенерировать слова по выбранной теме из меню и показать список."""
|
||
from database.db import async_session_maker
|
||
from services.user_service import UserService
|
||
from services.ai_service import ai_service
|
||
from bot.handlers.words import show_words_list, WordsStates
|
||
|
||
# Сразу отвечаем на callback, чтобы избежать таймаута
|
||
await callback.answer()
|
||
theme = callback.data.split("menu_theme_")[-1]
|
||
|
||
async with async_session_maker() as session:
|
||
user = await UserService.get_user_by_telegram_id(session, callback.from_user.id)
|
||
if not user:
|
||
await callback.answer(t('ru', 'common.start_first'), show_alert=True)
|
||
return
|
||
|
||
lang = (user.language_interface or 'ru')
|
||
generating = await callback.message.answer(t(lang, 'words.generating', theme=theme))
|
||
words = await ai_service.generate_thematic_words(
|
||
theme=theme,
|
||
level=user.level.value,
|
||
count=10,
|
||
learning_lang=user.learning_language,
|
||
translation_lang=user.language_interface,
|
||
)
|
||
await generating.delete()
|
||
|
||
if not words:
|
||
await callback.message.answer(t(lang, 'words.generate_failed'))
|
||
return
|
||
|
||
# Сохраняем в состояние как в /words
|
||
await state.update_data(theme=theme, words=words, user_id=user.id, level=user.level.name)
|
||
await state.set_state(WordsStates.viewing_words)
|
||
|
||
await show_words_list(callback.message, words, theme)
|
||
|
||
|
||
@router.message(Command("help"))
|
||
async def cmd_help(message: Message):
|
||
"""Обработчик команды /help"""
|
||
# Определяем язык пользователя
|
||
async with async_session_maker() as session:
|
||
user = await UserService.get_user_by_telegram_id(session, message.from_user.id)
|
||
lang = (user.language_interface if user else 'ru') or 'ru'
|
||
await message.answer(t(lang, "start.help"))
|
||
|
||
|
||
@router.callback_query(F.data == "offer_level_test")
|
||
async def offer_level_test_callback(callback: CallbackQuery, state: FSMContext):
|
||
"""Начать тест уровня из приветствия"""
|
||
from bot.handlers.level_test import start_level_test
|
||
await callback.message.delete()
|
||
await start_level_test(callback.message, state)
|
||
await callback.answer()
|
||
|
||
|
||
@router.callback_query(F.data == "skip_level_test")
|
||
async def skip_level_test_callback(callback: CallbackQuery):
|
||
"""Пропустить тест уровня"""
|
||
async with async_session_maker() as session:
|
||
user = await UserService.get_user_by_telegram_id(session, callback.from_user.id)
|
||
lang = (user.language_interface if user else 'ru') or 'ru'
|
||
await callback.message.edit_text(t(lang, 'start.skip_msg'))
|
||
await callback.answer()
|