first commit
This commit is contained in:
138
app/utils/email_sender.py
Normal file
138
app/utils/email_sender.py
Normal file
@@ -0,0 +1,138 @@
|
||||
import smtplib
|
||||
from email.mime.text import MIMEText
|
||||
from email.mime.multipart import MIMEMultipart
|
||||
from email.mime.application import MIMEApplication
|
||||
from typing import List, Optional
|
||||
from datetime import datetime
|
||||
from app.core.config import settings
|
||||
from app.core.logging import migration_logger
|
||||
|
||||
|
||||
class EmailSender:
|
||||
"""Класс для отправки email уведомлений"""
|
||||
|
||||
def __init__(self):
|
||||
self.smtp_server = settings.EMAIL_HOST
|
||||
self.smtp_port = settings.EMAIL_PORT
|
||||
self.username = settings.EMAIL_USER
|
||||
self.password = settings.EMAIL_PASSWORD
|
||||
self.from_addr = settings.EMAIL_FROM
|
||||
self.to_addrs = settings.EMAIL_TO
|
||||
|
||||
def send_email(self, subject: str, body: str, attachments: Optional[List[str]] = None) -> bool:
|
||||
"""Отправка email с вложениями"""
|
||||
if not all([self.smtp_server, self.username, self.password, self.from_addr]):
|
||||
migration_logger.warning("Настройки email не заполнены. Отправка email пропущена.")
|
||||
return False
|
||||
|
||||
try:
|
||||
# Создаем сообщение
|
||||
msg = MIMEMultipart()
|
||||
msg['From'] = self.from_addr
|
||||
msg['To'] = ', '.join(self.to_addrs)
|
||||
msg['Subject'] = subject
|
||||
msg['Date'] = datetime.now().strftime('%a, %d %b %Y %H:%M:%S %z')
|
||||
|
||||
# Добавляем текст
|
||||
msg.attach(MIMEText(body, 'plain', 'utf-8'))
|
||||
|
||||
# Добавляем вложения
|
||||
if attachments:
|
||||
for file_path in attachments:
|
||||
try:
|
||||
with open(file_path, 'rb') as f:
|
||||
part = MIMEApplication(f.read(), Name=file_path.split('/')[-1])
|
||||
part['Content-Disposition'] = f'attachment; filename="{file_path.split("/")[-1]}"'
|
||||
msg.attach(part)
|
||||
except Exception as e:
|
||||
migration_logger.error(f"Ошибка при добавлении вложения {file_path}: {e}")
|
||||
|
||||
# Отправляем
|
||||
with smtplib.SMTP_SSL(self.smtp_server, self.smtp_port) as server:
|
||||
server.login(self.username, self.password)
|
||||
server.send_message(msg)
|
||||
|
||||
migration_logger.info(f"Email успешно отправлен на {', '.join(self.to_addrs)}")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
migration_logger.error(f"Ошибка при отправке email: {e}")
|
||||
return False
|
||||
|
||||
def send_error_notification(self, error_message: str, traceback_str: str = None, table_name: str = None):
|
||||
"""Отправить уведомление об ошибке"""
|
||||
subject = f"ОШИБКА МИГРАЦИИ - {datetime.now().strftime('%Y-%m-%d %H:%M')}"
|
||||
|
||||
body = f"""
|
||||
🚨 ОШИБКА В ПРОЦЕССЕ МИГРАЦИИ ДАННЫХ
|
||||
{'='*60}
|
||||
|
||||
Время: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
|
||||
Сервис: Migration Service
|
||||
|
||||
"""
|
||||
if table_name:
|
||||
body += f"Таблица: {table_name}\n"
|
||||
|
||||
body += f"""
|
||||
Ошибка:
|
||||
{error_message}
|
||||
|
||||
"""
|
||||
if traceback_str:
|
||||
body += f"""
|
||||
📎 Детали:
|
||||
{traceback_str}
|
||||
|
||||
"""
|
||||
|
||||
body += """
|
||||
🔧 Необходимо проверить логи и устранить проблему.
|
||||
"""
|
||||
|
||||
return self.send_email(subject, body)
|
||||
|
||||
def send_success_notification(self, stats: dict, duration: float):
|
||||
"""Отправить уведомление об успешной миграции"""
|
||||
subject = f"УСПЕШНАЯ МИГРАЦИЯ - {datetime.now().strftime('%Y-%m-%d %H:%M')}"
|
||||
|
||||
body = f"""
|
||||
МИГРАЦИЯ ДАННЫХ УСПЕШНО ЗАВЕРШЕНА
|
||||
{'='*60}
|
||||
|
||||
Время: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
|
||||
Длительность: {duration:.1f} сек
|
||||
|
||||
СТАТИСТИКА:
|
||||
{'='*40}
|
||||
Всего строк в БД: {stats.get('total_rows', 0)}
|
||||
Таблиц обработано: {stats.get('total_tables', 0)}
|
||||
Последняя репликация: {stats.get('last_replication', 'Нет данных')}
|
||||
|
||||
Миграция выполнена успешно!
|
||||
"""
|
||||
|
||||
return self.send_email(subject, body)
|
||||
|
||||
def send_start_notification(self, tables: List[str], full_reload: bool):
|
||||
"""Отправить уведомление о начале миграции"""
|
||||
subject = f"НАЧАЛО МИГРАЦИИ - {datetime.now().strftime('%Y-%m-%d %H:%M')}"
|
||||
|
||||
body = f"""
|
||||
НАЧАЛО МИГРАЦИИ ДАННЫХ
|
||||
{'='*60}
|
||||
|
||||
Время старта: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
|
||||
Режим: {'ПОЛНАЯ ПЕРЕЗАГРУЗКА' if full_reload else 'ИНКРЕМЕНТАЛЬНАЯ'}
|
||||
Таблиц для обработки: {len(tables)}
|
||||
|
||||
СПИСОК ТАБЛИЦ:
|
||||
{chr(10).join([f' • {t}' for t in tables])}
|
||||
|
||||
Миграция запущена...
|
||||
"""
|
||||
|
||||
return self.send_email(subject, body)
|
||||
|
||||
|
||||
email_sender = EmailSender()
|
||||
Reference in New Issue
Block a user