289 lines
11 KiB
Python
289 lines
11 KiB
Python
from aiogram import Router, F
|
||
from aiogram.filters import Command
|
||
from aiogram.types import Message, InlineKeyboardMarkup, InlineKeyboardButton, CallbackQuery
|
||
from aiogram.fsm.context import FSMContext
|
||
from aiogram.fsm.state import State, StatesGroup
|
||
|
||
from database.db import async_session_maker
|
||
from services.user_service import UserService
|
||
from services.task_service import TaskService
|
||
from services.ai_service import ai_service
|
||
from utils.i18n import t
|
||
|
||
router = Router()
|
||
|
||
|
||
class TaskStates(StatesGroup):
|
||
"""Состояния для прохождения заданий"""
|
||
doing_tasks = State()
|
||
waiting_for_answer = State()
|
||
|
||
|
||
@router.message(Command("task"))
|
||
async def cmd_task(message: Message, state: FSMContext):
|
||
"""Обработчик команды /task"""
|
||
async with async_session_maker() as session:
|
||
user = await UserService.get_user_by_telegram_id(session, message.from_user.id)
|
||
|
||
if not user:
|
||
await message.answer(t('ru', 'common.start_first'))
|
||
return
|
||
|
||
# Генерируем задания разных типов
|
||
tasks = await TaskService.generate_mixed_tasks(
|
||
session, user.id, count=5,
|
||
learning_lang=user.learning_language,
|
||
translation_lang=user.language_interface,
|
||
)
|
||
|
||
if not tasks:
|
||
await message.answer(t(user.level.value and (user.language_interface or 'ru') or 'ru', 'tasks.no_words'))
|
||
return
|
||
|
||
# Сохраняем задания в состоянии
|
||
await state.update_data(
|
||
tasks=tasks,
|
||
current_task_index=0,
|
||
correct_count=0,
|
||
user_id=user.id
|
||
)
|
||
await state.set_state(TaskStates.doing_tasks)
|
||
|
||
# Показываем первое задание
|
||
await show_current_task(message, state)
|
||
|
||
|
||
async def show_current_task(message: Message, state: FSMContext):
|
||
"""Показать текущее задание"""
|
||
data = await state.get_data()
|
||
tasks = data.get('tasks', [])
|
||
current_index = data.get('current_task_index', 0)
|
||
|
||
if current_index >= len(tasks):
|
||
# Все задания выполнены
|
||
await finish_tasks(message, state)
|
||
return
|
||
|
||
task = tasks[current_index]
|
||
|
||
# Определяем язык пользователя
|
||
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'
|
||
|
||
task_text = (
|
||
t(lang, 'tasks.header', i=current_index + 1, n=len(tasks)) + "\n\n" +
|
||
f"{task['question']}\n"
|
||
)
|
||
|
||
if task.get('transcription'):
|
||
task_text += f"🔊 [{task['transcription']}]\n"
|
||
|
||
task_text += t(lang, 'tasks.write_answer')
|
||
|
||
await state.set_state(TaskStates.waiting_for_answer)
|
||
await message.answer(task_text)
|
||
|
||
|
||
@router.message(TaskStates.waiting_for_answer)
|
||
async def process_answer(message: Message, state: FSMContext):
|
||
"""Обработка ответа пользователя"""
|
||
user_answer = message.text.strip()
|
||
data = await state.get_data()
|
||
|
||
tasks = data.get('tasks', [])
|
||
current_index = data.get('current_task_index', 0)
|
||
correct_count = data.get('correct_count', 0)
|
||
user_id = data.get('user_id')
|
||
|
||
task = tasks[current_index]
|
||
|
||
# Показываем индикатор проверки
|
||
# Язык пользователя
|
||
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'
|
||
|
||
checking_msg = await message.answer(t(lang, 'tasks.checking'))
|
||
|
||
# Проверяем ответ через AI
|
||
check_result = await ai_service.check_answer(
|
||
question=task['question'],
|
||
correct_answer=task['correct_answer'],
|
||
user_answer=user_answer
|
||
)
|
||
|
||
await checking_msg.delete()
|
||
|
||
is_correct = check_result.get('is_correct', False)
|
||
feedback = check_result.get('feedback', '')
|
||
|
||
# Формируем ответ
|
||
if is_correct:
|
||
result_text = t(lang, 'tasks.correct') + "\n\n"
|
||
correct_count += 1
|
||
else:
|
||
result_text = t(lang, 'tasks.incorrect') + "\n\n"
|
||
|
||
result_text += f"{t(lang, 'tasks.your_answer')}: <i>{user_answer}</i>\n"
|
||
result_text += f"{t(lang, 'tasks.right_answer')}: <b>{task['correct_answer']}</b>\n\n"
|
||
|
||
if feedback:
|
||
result_text += f"💬 {feedback}\n\n"
|
||
|
||
# Сохраняем результат в БД
|
||
async with async_session_maker() as session:
|
||
await TaskService.save_task_result(
|
||
session=session,
|
||
user_id=user_id,
|
||
task_type=task['type'],
|
||
content={
|
||
'question': task['question'],
|
||
'word': task['word']
|
||
},
|
||
user_answer=user_answer,
|
||
correct_answer=task['correct_answer'],
|
||
is_correct=is_correct,
|
||
ai_feedback=feedback
|
||
)
|
||
|
||
# Обновляем статистику слова
|
||
if 'word_id' in task:
|
||
await TaskService.update_word_statistics(
|
||
session=session,
|
||
word_id=task['word_id'],
|
||
is_correct=is_correct
|
||
)
|
||
|
||
# Обновляем счетчик
|
||
await state.update_data(
|
||
current_task_index=current_index + 1,
|
||
correct_count=correct_count
|
||
)
|
||
|
||
# Показываем результат и кнопку "Далее"
|
||
keyboard = InlineKeyboardMarkup(inline_keyboard=[
|
||
[InlineKeyboardButton(text=t(lang, 'tasks.next_btn'), callback_data="next_task")],
|
||
[InlineKeyboardButton(text=t(lang, 'tasks.stop_btn'), callback_data="stop_tasks")]
|
||
])
|
||
|
||
await message.answer(result_text, reply_markup=keyboard)
|
||
# После показа результата ждём нажатия кнопки – переключаемся в состояние doing_tasks
|
||
await state.set_state(TaskStates.doing_tasks)
|
||
|
||
|
||
@router.callback_query(F.data == "next_task", TaskStates.doing_tasks)
|
||
async def next_task(callback: CallbackQuery, state: FSMContext):
|
||
"""Переход к следующему заданию"""
|
||
await callback.message.delete()
|
||
await show_current_task(callback.message, state)
|
||
await callback.answer()
|
||
|
||
|
||
@router.callback_query(F.data == "stop_tasks", TaskStates.doing_tasks)
|
||
async def stop_tasks_callback(callback: CallbackQuery, state: FSMContext):
|
||
"""Остановить выполнение заданий через кнопку"""
|
||
await state.clear()
|
||
await callback.message.edit_reply_markup(reply_markup=None)
|
||
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.answer(t(lang, 'tasks.finished'))
|
||
await callback.answer()
|
||
|
||
|
||
@router.message(Command("stop"), TaskStates.doing_tasks)
|
||
@router.message(Command("stop"), TaskStates.waiting_for_answer)
|
||
async def stop_tasks(message: Message, state: FSMContext):
|
||
"""Остановить выполнение заданий командой /stop"""
|
||
await state.clear()
|
||
# Определяем язык пользователя
|
||
async with async_session_maker() as session:
|
||
user = await UserService.get_user_by_telegram_id(session, message.from_user.id)
|
||
await message.answer(t((user.language_interface if user else 'ru') or 'ru', 'tasks.stopped'))
|
||
|
||
|
||
@router.message(Command("cancel"), TaskStates.doing_tasks)
|
||
@router.message(Command("cancel"), TaskStates.waiting_for_answer)
|
||
async def cancel_tasks(message: Message, state: FSMContext):
|
||
"""Отмена выполнения заданий командой /cancel"""
|
||
await state.clear()
|
||
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, 'tasks.cancelled'))
|
||
|
||
|
||
async def finish_tasks(message: Message, state: FSMContext):
|
||
"""Завершение всех заданий"""
|
||
data = await state.get_data()
|
||
tasks = data.get('tasks', [])
|
||
correct_count = data.get('correct_count', 0)
|
||
total_count = len(tasks)
|
||
|
||
accuracy = int((correct_count / total_count) * 100) if total_count > 0 else 0
|
||
|
||
# Определяем эмодзи на основе результата
|
||
if accuracy >= 90:
|
||
emoji = "🏆"
|
||
comment_key = 'excellent'
|
||
elif accuracy >= 70:
|
||
emoji = "👍"
|
||
comment_key = 'good'
|
||
elif accuracy >= 50:
|
||
emoji = "📚"
|
||
comment_key = 'average'
|
||
else:
|
||
emoji = "💪"
|
||
comment_key = 'poor'
|
||
|
||
# Язык пользователя
|
||
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'
|
||
|
||
result_text = (
|
||
t(lang, 'tasks.finish_title', emoji=emoji) + "\n\n" +
|
||
t(lang, 'tasks.correct_of', correct=correct_count, total=total_count) + "\n" +
|
||
t(lang, 'tasks.accuracy', accuracy=accuracy) + "\n\n" +
|
||
t(lang, f"tasks.comment.{comment_key}") + "\n\n" +
|
||
t(lang, 'tasks.use_task') + "\n" +
|
||
t(lang, 'tasks.use_stats')
|
||
)
|
||
|
||
await state.clear()
|
||
await message.answer(result_text)
|
||
|
||
|
||
@router.message(Command("stats"))
|
||
async def cmd_stats(message: Message):
|
||
"""Обработчик команды /stats"""
|
||
async with async_session_maker() as session:
|
||
user = await UserService.get_user_by_telegram_id(session, message.from_user.id)
|
||
|
||
if not user:
|
||
await message.answer(t('ru', 'common.start_first'))
|
||
return
|
||
|
||
# Получаем статистику
|
||
stats = await TaskService.get_user_stats(session, user.id)
|
||
lang = (user.language_interface if user else 'ru') or 'ru'
|
||
|
||
stats_text = (
|
||
t(lang, 'stats.header') + "\n\n" +
|
||
t(lang, 'stats.total_words', n=stats['total_words']) + "\n" +
|
||
t(lang, 'stats.studied_words', n=stats['reviewed_words']) + "\n" +
|
||
t(lang, 'stats.total_tasks', n=stats['total_tasks']) + "\n" +
|
||
t(lang, 'stats.correct_tasks', n=stats['correct_tasks']) + "\n" +
|
||
t(lang, 'stats.accuracy', n=stats['accuracy']) + "\n\n"
|
||
)
|
||
|
||
if stats['total_words'] == 0:
|
||
stats_text += t(lang, 'stats.hint_add_words')
|
||
elif stats['total_tasks'] == 0:
|
||
stats_text += t(lang, 'stats.hint_first_task')
|
||
else:
|
||
stats_text += t(lang, 'stats.hint_keep_practice')
|
||
|
||
await message.answer(stats_text)
|