87 lines
3.8 KiB
Python
87 lines
3.8 KiB
Python
from django.contrib.auth.models import User
|
||
from django.core.mail import send_mail
|
||
|
||
from Main.models import Solution, Progress
|
||
from Sprint.settings import EMAIL_HOST_USER
|
||
from SprintLib.queue import MessagingSupport, send_to_queue
|
||
from daemons.management.commands.bot import bot
|
||
|
||
|
||
class Command(MessagingSupport):
|
||
help = "starts notification manager"
|
||
queue_name = "notification"
|
||
|
||
def send_email(self, payload):
|
||
"""С email потом разберемся"""
|
||
return
|
||
subject = payload['subject']
|
||
message = payload['message']
|
||
email = payload['email']
|
||
send_mail(
|
||
subject,
|
||
message,
|
||
EMAIL_HOST_USER,
|
||
[email]
|
||
)
|
||
|
||
def send_telegram(self, payload):
|
||
chat_id = payload['chat_id']
|
||
text = payload['text']
|
||
bot.send_message(
|
||
int(chat_id),
|
||
text,
|
||
parse_mode="html",
|
||
)
|
||
|
||
def handle_send_code(self, payload):
|
||
yield "telegram", payload
|
||
|
||
def handle_solution(self, payload):
|
||
solution = Solution.objects.get(id=payload["solution_id"])
|
||
user = solution.user
|
||
if user.userinfo.notification_solution_result:
|
||
message = f"Задача: {solution.task.name}\n" \
|
||
f"Результат: {solution.result}\n" \
|
||
f"Очки решения: {Progress.by_solution(solution).score}\n" \
|
||
f"Текущий рейтинг: {solution.user.userinfo.rating}"
|
||
if user.userinfo.notification_telegram is not None:
|
||
yield "telegram", {"chat_id": user.userinfo.telegram_chat_id, "text": message}
|
||
if user.userinfo.notification_email:
|
||
yield "email", {"subject": "Тестирование завершено", "message": message, "email": user.email}
|
||
|
||
def handle_friends_add(self, payload):
|
||
user = User.objects.get(id=payload['to_user'])
|
||
from_user = User.objects.get(id=payload['from_user'])
|
||
if user.userinfo.notification_friends:
|
||
message = f"Пользователь {from_user.username} хочет добавить тебя в друзья"
|
||
if user.userinfo.notification_telegram:
|
||
yield "telegram", {"chat_id": user.userinfo.telegram_chat_id, "text": message}
|
||
if user.userinfo.notification_email:
|
||
yield "email", {"subject": "Новая заявка в друзья", "message": message, "email": user.email}
|
||
|
||
def handle_friends_accept(self, payload):
|
||
user = User.objects.get(id=payload['to_user'])
|
||
from_user = User.objects.get(id=payload['from_user'])
|
||
if user.userinfo.notification_friends:
|
||
if payload['accepted']:
|
||
message = f"Пользователь {from_user} одобрил заявку в друзья"
|
||
else:
|
||
message = f"Пользователь {from_user} отклонил заявку в друзья"
|
||
if user.userinfo.notification_telegram:
|
||
yield "telegram", {"chat_id": user.userinfo.telegram_chat_id, "text": message}
|
||
if user.userinfo.notification_email:
|
||
yield "email", {"subject": "Новая заявка в друзья", "message": message, "email": user.email}
|
||
|
||
def process(self, payload: dict):
|
||
notification_type = payload["type"]
|
||
handler = getattr(self, "handle_" + notification_type, None)
|
||
if handler is None:
|
||
raise ValueError(f"Unknown type: {notification_type}")
|
||
for queue, payload in handler(payload):
|
||
if queue == 'telegram':
|
||
self.send_telegram(payload)
|
||
elif queue == 'email':
|
||
self.send_email(payload)
|
||
else:
|
||
raise ValueError(f"Unknown type {queue}")
|