Игорь Надеин
35f365a1b0
- Added departments ManyToMany field to Document model - Implemented department selection in document create/edit modals - Added automatic notification creation for all department employees - Fixed document filtering to include uploaded documents in 'My Documents' - Fixed acknowledgements API to include department recipients - Added bulk notification support with configurable threshold - Removed acknowledge button, added visual highlighting for unread documents - Implemented server-side acknowledgment flow with redirect to file - Fixed mobile responsive layout for document actions - Removed temporary test files and documentation
378 lines
14 KiB
Python
378 lines
14 KiB
Python
"""
|
||
Сервисный слой для работы с уведомлениями
|
||
"""
|
||
import json
|
||
import logging
|
||
from typing import Any, Dict, Optional
|
||
from django.contrib.auth import get_user_model
|
||
from django.utils import timezone
|
||
from channels.layers import get_channel_layer
|
||
from asgiref.sync import async_to_sync
|
||
from .models import (
|
||
Notification,
|
||
NotificationType,
|
||
UserNotificationSettings,
|
||
)
|
||
from .email_sender import EmailNotificationSender
|
||
from .telegram_sender import TelegramNotificationSender
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
User = get_user_model()
|
||
|
||
|
||
class NotificationService:
|
||
"""Сервис для создания и отправки уведомлений"""
|
||
|
||
@staticmethod
|
||
def create_notification(
|
||
recipient: User,
|
||
notification_type_code: str,
|
||
title: str,
|
||
message: str,
|
||
content_object=None,
|
||
action_url: str = '',
|
||
action_text: str = 'Посмотреть',
|
||
metadata: Optional[Dict[str, Any]] = None,
|
||
send_immediately: bool = True,
|
||
) -> Optional[Notification]:
|
||
"""
|
||
Создать и отправить уведомление
|
||
|
||
Args:
|
||
recipient: Получатель уведомления
|
||
notification_type_code: Код типа уведомления
|
||
title: Заголовок
|
||
message: Текст сообщения
|
||
content_object: Связанный объект (опционально)
|
||
action_url: URL для действия
|
||
action_text: Текст кнопки действия
|
||
metadata: Дополнительные данные
|
||
send_immediately: Отправить сразу или отложить
|
||
|
||
Returns:
|
||
Созданное уведомление или None если тип не найден
|
||
"""
|
||
logger.info(
|
||
f"[create_notification] Starting for recipient={recipient.id} "
|
||
f"type={notification_type_code} send_immediately={send_immediately}"
|
||
)
|
||
|
||
try:
|
||
notification_type = NotificationType.objects.get(
|
||
code=notification_type_code,
|
||
is_active=True
|
||
)
|
||
except NotificationType.DoesNotExist:
|
||
logger.warning(
|
||
f"[create_notification] Type {notification_type_code} not found"
|
||
)
|
||
return None
|
||
|
||
# Проверить настройки пользователя
|
||
settings = NotificationService.get_user_settings(
|
||
recipient, notification_type
|
||
)
|
||
|
||
if not settings.is_enabled:
|
||
logger.info(
|
||
f"[create_notification] Skipped - disabled for user {recipient.id}"
|
||
)
|
||
return None
|
||
|
||
# Создать краткое сообщение
|
||
short_message = message[:150] if len(message) > 150 else message
|
||
|
||
# Создать уведомление
|
||
notification = Notification.objects.create(
|
||
recipient=recipient,
|
||
notification_type=notification_type,
|
||
title=title,
|
||
message=message,
|
||
short_message=short_message,
|
||
content_object=content_object,
|
||
action_url=action_url,
|
||
action_text=action_text,
|
||
metadata=metadata or {},
|
||
)
|
||
|
||
# Отправить если нужно
|
||
if send_immediately:
|
||
NotificationService.send_notification(notification, settings)
|
||
|
||
return notification
|
||
|
||
@staticmethod
|
||
def send_notification(
|
||
notification: Notification,
|
||
settings: Optional[UserNotificationSettings] = None
|
||
):
|
||
"""Отправить уведомление по всем каналам"""
|
||
logger.info(
|
||
f"[send_notification] Starting for notification={notification.id}"
|
||
)
|
||
|
||
if settings is None:
|
||
settings = NotificationService.get_user_settings(
|
||
notification.recipient,
|
||
notification.notification_type
|
||
)
|
||
|
||
logger.info(
|
||
f"[send_notification] Channels: web={settings.send_web} "
|
||
f"email={settings.send_email} telegram={settings.send_telegram}"
|
||
)
|
||
|
||
# Отправить на веб (WebSocket)
|
||
if settings.send_web:
|
||
NotificationService.send_web_notification(notification)
|
||
notification.sent_web = True
|
||
|
||
# Отправить на email
|
||
if settings.send_email:
|
||
try:
|
||
recipient_email = notification.recipient.email
|
||
if recipient_email:
|
||
success = EmailNotificationSender.send_notification_email(
|
||
notification=notification,
|
||
recipient_email=recipient_email,
|
||
)
|
||
notification.sent_email = success
|
||
if success:
|
||
logger.info(
|
||
f"Email отправлен для уведомления {notification.id}"
|
||
)
|
||
else:
|
||
notification.sent_email = False
|
||
logger.warning(
|
||
f"У пользователя {notification.recipient} нет email"
|
||
)
|
||
except Exception as e:
|
||
notification.sent_email = False
|
||
logger.error(
|
||
f"Ошибка отправки email для {notification.id}: {e}",
|
||
exc_info=True
|
||
)
|
||
|
||
# Отправить в Telegram
|
||
if settings.send_telegram:
|
||
logger.info(
|
||
f"[Telegram] Attempting to send for notification {notification.id} "
|
||
f"to user {notification.recipient.id}"
|
||
)
|
||
try:
|
||
# Проверяем наличие Telegram Chat ID в профиле
|
||
telegram_chat_id = getattr(
|
||
notification.recipient, 'telegram', ''
|
||
)
|
||
if telegram_chat_id:
|
||
telegram_chat_id = telegram_chat_id.strip()
|
||
|
||
logger.info(
|
||
f"[Telegram] Chat ID for user {notification.recipient.id}: "
|
||
f"'{telegram_chat_id}'"
|
||
)
|
||
|
||
if telegram_chat_id:
|
||
# Получаем site_url для формирования полных ссылок
|
||
from django.conf import settings as django_settings
|
||
site_url = getattr(
|
||
django_settings,
|
||
'SITE_URL',
|
||
'http://localhost:9000'
|
||
)
|
||
|
||
logger.info(
|
||
f"[Telegram] Sending via TelegramNotificationSender "
|
||
f"to chat_id={telegram_chat_id}"
|
||
)
|
||
|
||
# Отправляем через Bot API по chat_id
|
||
from .telegram_sender import TelegramNotificationSender
|
||
success = TelegramNotificationSender.send_notification(
|
||
notification=notification,
|
||
chat_id=telegram_chat_id,
|
||
site_url=site_url
|
||
)
|
||
notification.sent_telegram = success
|
||
|
||
if success:
|
||
logger.info(
|
||
f"[Telegram] Successfully sent notification "
|
||
f"{notification.id}"
|
||
)
|
||
else:
|
||
logger.warning(
|
||
f"[Telegram] Failed to send notification "
|
||
f"{notification.id}"
|
||
)
|
||
else:
|
||
notification.sent_telegram = False
|
||
logger.debug(
|
||
f"[Telegram] Chat ID empty for user "
|
||
f"{notification.recipient.id}"
|
||
)
|
||
|
||
except Exception as e:
|
||
notification.sent_telegram = False
|
||
logger.error(
|
||
f"[Telegram] Error sending notification {notification.id}: {e}",
|
||
exc_info=True
|
||
)
|
||
|
||
# Обновить время отправки
|
||
notification.sent_at = timezone.now()
|
||
notification.save(update_fields=[
|
||
'sent_web',
|
||
'sent_email',
|
||
'sent_telegram',
|
||
'sent_whatsapp',
|
||
'sent_wechat',
|
||
'sent_at'
|
||
])
|
||
|
||
@staticmethod
|
||
def send_web_notification(notification: Notification):
|
||
"""Отправить уведомление через WebSocket"""
|
||
channel_layer = get_channel_layer()
|
||
notification_data = {
|
||
'id': notification.id,
|
||
'title': notification.title,
|
||
'message': notification.short_message,
|
||
'category': notification.notification_type.category.code,
|
||
'category_name': notification.notification_type.category.name,
|
||
'icon': notification.notification_type.category.icon,
|
||
'color': notification.notification_type.category.color,
|
||
'priority': notification.notification_type.priority,
|
||
'action_url': notification.action_url,
|
||
'action_text': notification.action_text,
|
||
'created_at': notification.created_at.isoformat(),
|
||
}
|
||
|
||
async_to_sync(channel_layer.group_send)(
|
||
f'notifications_{notification.recipient.id}',
|
||
{
|
||
'type': 'notification_new',
|
||
'notification': notification_data
|
||
}
|
||
)
|
||
|
||
# Обновить счетчик
|
||
unread_count = Notification.objects.filter(
|
||
recipient=notification.recipient,
|
||
is_read=False,
|
||
is_archived=False
|
||
).count()
|
||
|
||
async_to_sync(channel_layer.group_send)(
|
||
f'notifications_{notification.recipient.id}',
|
||
{
|
||
'type': 'notification_count_update',
|
||
'count': unread_count
|
||
}
|
||
)
|
||
|
||
@staticmethod
|
||
def get_user_settings(
|
||
user: User,
|
||
notification_type: NotificationType
|
||
) -> UserNotificationSettings:
|
||
"""Получить или создать настройки пользователя"""
|
||
settings, created = UserNotificationSettings.objects.get_or_create(
|
||
user=user,
|
||
notification_type=notification_type,
|
||
defaults={
|
||
'is_enabled': notification_type.default_enabled,
|
||
'send_web': notification_type.default_channels.get(
|
||
'web', True
|
||
),
|
||
'send_email': notification_type.default_channels.get(
|
||
'email', False
|
||
),
|
||
'send_telegram': notification_type.default_channels.get(
|
||
'telegram', False
|
||
),
|
||
'send_whatsapp': notification_type.default_channels.get(
|
||
'whatsapp', False
|
||
),
|
||
'send_wechat': notification_type.default_channels.get(
|
||
'wechat', False
|
||
),
|
||
}
|
||
)
|
||
|
||
logger.debug(
|
||
f"[Settings] User {user.id} type={notification_type.code} "
|
||
f"created={created} enabled={settings.is_enabled} "
|
||
f"web={settings.send_web} email={settings.send_email} "
|
||
f"telegram={settings.send_telegram}"
|
||
)
|
||
|
||
return settings
|
||
|
||
@staticmethod
|
||
def mark_as_read(notification_id: int, user: User) -> bool:
|
||
"""Отметить уведомление как прочитанное"""
|
||
try:
|
||
notification = Notification.objects.get(
|
||
id=notification_id,
|
||
recipient=user
|
||
)
|
||
notification.mark_as_read()
|
||
|
||
# Обновить счетчик через WebSocket
|
||
channel_layer = get_channel_layer()
|
||
unread_count = Notification.objects.filter(
|
||
recipient=user,
|
||
is_read=False,
|
||
is_archived=False
|
||
).count()
|
||
|
||
async_to_sync(channel_layer.group_send)(
|
||
f'notifications_{user.id}',
|
||
{
|
||
'type': 'notification_count_update',
|
||
'count': unread_count
|
||
}
|
||
)
|
||
|
||
return True
|
||
except Notification.DoesNotExist:
|
||
return False
|
||
|
||
@staticmethod
|
||
def mark_all_as_read(user: User, category: Optional[str] = None) -> int:
|
||
"""Отметить все уведомления как прочитанные"""
|
||
queryset = Notification.objects.filter(
|
||
recipient=user,
|
||
is_read=False
|
||
)
|
||
|
||
if category:
|
||
queryset = queryset.filter(
|
||
notification_type__category__code=category
|
||
)
|
||
|
||
count = queryset.update(
|
||
is_read=True,
|
||
read_at=timezone.now()
|
||
)
|
||
|
||
# Обновить счетчик
|
||
channel_layer = get_channel_layer()
|
||
unread_count = Notification.objects.filter(
|
||
recipient=user,
|
||
is_read=False,
|
||
is_archived=False
|
||
).count()
|
||
|
||
async_to_sync(channel_layer.group_send)(
|
||
f'notifications_{user.id}',
|
||
{
|
||
'type': 'notification_count_update',
|
||
'count': unread_count
|
||
}
|
||
)
|
||
|
||
return count
|