"""Celery задачи отправки email уведомлений.""" import logging from celery.utils.log import get_task_logger from app.celery_app import celery_app from app.config import settings from app.db import db_session from app.email_sender import EmailSender logger = get_task_logger(__name__) @celery_app.task( name="notify.send_task_done", bind=True, max_retries=3, default_retry_delay=30, ) def send_task_done(self, task_id: str) -> dict: """ Отправить email уведомление о завершении задачи. Args: task_id: ID завершённой задачи Returns: dict со статусом отправки """ from app.models import Task, User logger.info(f"Отправка уведомления для задачи {task_id!r}") try: with db_session() as session: task = session.get(Task, task_id) if task is None: logger.error(f"Задача {task_id!r} не найдена") return {"status": "task_not_found"} user = session.get(User, task.user_id) if user is None: logger.error(f"Пользователь {task.user_id} не найден") return {"status": "user_not_found"} # Формируем краткое описание результата summary = _build_summary(task) sender = EmailSender() sender.send_task_done( user_email=user.email, user_name=user.name, task_id=task_id, task_type=task.type, summary=summary, app_url=settings.APP_URL, ) logger.info(f"Уведомление отправлено пользователю {user.email!r} для задачи {task_id!r}") return {"status": "sent", "email": user.email} except Exception as exc: logger.error(f"Ошибка отправки уведомления для задачи {task_id!r}: {exc}", exc_info=True) raise self.retry(exc=exc, countdown=30) def _build_summary(task) -> str: """Сформировать краткое описание результата для email.""" result = task.result or {} if task.type == "search": count = len(result.get("sources", [])) query = task.input_data.get("query", "") return ( f'Найдено {count} источников по запросу "{query[:80]}".' f' Откройте результат, чтобы просмотреть список с ГОСТ-цитатами.' ) elif task.type == "plagiarism": similarity = result.get("overall_similarity", 0) total = result.get("total_fragments", 0) flagged = result.get("flagged_fragments", 0) filename = task.input_data.get("filename", "") color = "#dc2626" if similarity > 30 else "#d97706" if similarity > 10 else "#16a34a" label = "Высокий" if similarity > 30 else "Средний" if similarity > 10 else "Низкий" return ( f'Проверка файла "{filename}" завершена.
' f'Уровень схожести: {similarity:.1f}% ({label}).
' f'Проверено {total} фрагментов, выявлено совпадений: {flagged}.' ) elif task.type == "gost": count = len(result.get("bibliography", [])) style = result.get("style", "7.1") return ( f'Библиография из {count} источников' f' отформатирована по ГОСТ {style}-2003.' ) elif task.type == "summarize": return "Краткое изложение успешно создано." return "Ваша задача выполнена." @celery_app.task( name="notify.send_verification", bind=True, max_retries=3, default_retry_delay=60, ) def send_verification(self, user_email: str, user_name: str, token: str) -> dict: """ Отправить письмо с подтверждением email. Args: user_email: Email для верификации user_name: Имя пользователя token: Токен верификации Returns: dict со статусом отправки """ logger.info(f"Отправка письма верификации на {user_email!r}") try: sender = EmailSender() sender.send_verification( user_email=user_email, user_name=user_name, token=token, app_url=settings.APP_URL, ) logger.info(f"Письмо верификации отправлено на {user_email!r}") return {"status": "sent"} except Exception as exc: logger.error(f"Ошибка отправки верификации на {user_email!r}: {exc}", exc_info=True) raise self.retry(exc=exc, countdown=60)