BREAKING CHANGES: - Удалена старая система (6 моделей → 2 модели) - models_new.py → models.py (Notification, UserChannelPreferences) - signals_new.py → signals.py (notify.send API) - admin_new.py → admin.py Новая архитектура: - Все отправки через Celery (полностью асинхронно) - Модульная структура tasks/ (base, email, push, websocket) - Каждый канал - отдельный класс задачи наследующий BaseNotificationTask - Rate limiting, retry, приоритеты из коробки Обратная совместимость: - Создан services.py facade для старого API - NotificationService.create_notification() работает через notify.send() - Старый код продолжает работать без изменений Удалено: - Старые models.py, signals.py, admin.py, tasks.py (→ .old) - telegram_bot.py, telegram_models.py (Telegram отключен) - test_views.py, views.py, urls.py (устаревшие) Создано: - tasks/ модуль с классовой архитектурой (как senders/) - tasks/base.py - BaseNotificationTask - tasks/email.py - EmailNotificationTask + DigestEmailTask - tasks/push.py - PushNotificationTask - tasks/websocket.py - WebSocketNotificationTask Обновлено: - apps.py - загружает signals, channels, tasks - channels.py - все отправки через Celery задачи - Все импорты models_new → models
112 lines
4.5 KiB
Python
112 lines
4.5 KiB
Python
"""
|
||
Celery задачи для отправки email уведомлений
|
||
"""
|
||
from .base import BaseNotificationTask
|
||
|
||
|
||
class EmailNotificationTask(BaseNotificationTask):
|
||
"""
|
||
Celery задача для асинхронной отправки email уведомлений.
|
||
|
||
Особенности:
|
||
- Rate limiting: 10 email/минуту (защита от SMTP бана)
|
||
- Retry: 3 попытки с интервалом 5 минут
|
||
- Поддержка кастомных тем и получателей
|
||
"""
|
||
|
||
task_name = "notifications.send_email"
|
||
max_retries = 3
|
||
retry_delay = 300 # 5 минут между попытками
|
||
rate_limit = '10/m' # Не более 10 email в минуту
|
||
|
||
def send_notification(self, notification, **kwargs) -> bool:
|
||
"""
|
||
Отправляет email через EmailNotificationSender.
|
||
|
||
Args:
|
||
notification: Объект Notification
|
||
**kwargs:
|
||
- custom_subject: кастомная тема письма
|
||
- recipient_email: явный email получателя
|
||
|
||
Returns:
|
||
True если успешно, False иначе
|
||
"""
|
||
from notifications.senders.email import EmailNotificationSender
|
||
|
||
sender = EmailNotificationSender()
|
||
return sender.send(notification, **kwargs)
|
||
|
||
|
||
class DigestEmailTask(BaseNotificationTask):
|
||
"""
|
||
Celery задача для отправки email дайджестов.
|
||
|
||
Особенности:
|
||
- Отправка сводки непрочитанных уведомлений
|
||
- Поддержка daily/weekly частоты
|
||
- Максимум 50 уведомлений в одном дайджесте
|
||
"""
|
||
|
||
task_name = "notifications.send_digest_email"
|
||
max_retries = 2
|
||
retry_delay = 600 # 10 минут между попытками
|
||
rate_limit = '5/h' # Не более 5 дайджестов в час
|
||
|
||
def execute(self, celery_task, user_id: int, frequency: str = 'daily', **kwargs):
|
||
"""Переопределяем execute для работы с user_id вместо notification_id"""
|
||
try:
|
||
from django.contrib.auth import get_user_model
|
||
from notifications.models import Notification
|
||
from notifications.senders.email import EmailNotificationSender
|
||
from django.utils import timezone
|
||
from datetime import timedelta
|
||
|
||
User = get_user_model()
|
||
user = User.objects.get(id=user_id)
|
||
|
||
# Определяем период для дайджеста
|
||
if frequency == 'weekly':
|
||
cutoff = timezone.now() - timedelta(days=7)
|
||
else: # daily
|
||
cutoff = timezone.now() - timedelta(days=1)
|
||
|
||
# Получаем непрочитанные уведомления за период
|
||
notifications = Notification.objects.filter(
|
||
recipient=user,
|
||
unread=True,
|
||
emailed=False, # Еще не отправлены по email
|
||
timestamp__gte=cutoff,
|
||
).order_by('-timestamp')[:50] # Максимум 50 в дайджесте
|
||
|
||
if not notifications.exists():
|
||
self.logger.info(f"📭 No notifications for digest: user={user_id}, frequency={frequency}")
|
||
return False
|
||
|
||
# Отправляем дайджест
|
||
sender = EmailNotificationSender()
|
||
success = sender.send_digest(user, list(notifications), frequency=frequency)
|
||
|
||
if success:
|
||
self.logger.info(
|
||
f"✅ Digest sent: user={user_id}, frequency={frequency}, "
|
||
f"count={notifications.count()}"
|
||
)
|
||
else:
|
||
self.logger.warning(f"⚠️ Digest failed: user={user_id}, frequency={frequency}")
|
||
|
||
return success
|
||
|
||
except Exception as exc:
|
||
self.logger.exception(f"❌ Digest task error: user={user_id}, frequency={frequency}")
|
||
raise celery_task.retry(exc=exc)
|
||
|
||
def send_notification(self, notification, **kwargs) -> bool:
|
||
"""Не используется для дайджестов"""
|
||
pass
|
||
|
||
|
||
# Регистрируем задачи в Celery
|
||
send_email_notification = EmailNotificationTask.register_task()
|
||
send_digest_email = DigestEmailTask.register_task()
|