"""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 {}
input_data = task.input_data or {}
if task.type == "search":
count = len(result.get("sources", []))
query = 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 = 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)