anti-plagiarism/services/worker-notifier/app/tasks/notify.py

"""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'Найдено <strong>{count} источников</strong> по запросу "{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'Проверка файла <strong>"{filename}"</strong> завершена.<br>'
            f'Уровень схожести: <strong style="color:{color};">{similarity:.1f}% ({label})</strong>.<br>'
            f'Проверено {total} фрагментов, выявлено совпадений: {flagged}.'
        )

    elif task.type == "gost":
        count = len(result.get("bibliography", []))
        style = result.get("style", "7.1")
        return (
            f'Библиография из <strong>{count} источников</strong>'
            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)