#!/usr/bin/env python3 # -*- coding: utf-8 -*- """ 通知服務 Author: PANJIT IT Team Created: 2024-01-28 Modified: 2024-01-28 """ import os import smtplib from email.mime.text import MIMEText from email.mime.multipart import MIMEMultipart from datetime import datetime from typing import Optional, List from flask import current_app, url_for from app.utils.logger import get_logger from app.models.job import TranslationJob from app.models.user import User logger = get_logger(__name__) class NotificationService: """通知服務""" def __init__(self): self.smtp_server = current_app.config.get('SMTP_SERVER') self.smtp_port = current_app.config.get('SMTP_PORT', 587) self.use_tls = current_app.config.get('SMTP_USE_TLS', False) self.use_ssl = current_app.config.get('SMTP_USE_SSL', False) self.auth_required = current_app.config.get('SMTP_AUTH_REQUIRED', False) self.sender_email = current_app.config.get('SMTP_SENDER_EMAIL') self.sender_password = current_app.config.get('SMTP_SENDER_PASSWORD', '') self.app_name = current_app.config.get('APP_NAME', 'PANJIT Document Translator') def _create_smtp_connection(self): """建立 SMTP 連線""" try: if self.use_ssl: server = smtplib.SMTP_SSL(self.smtp_server, self.smtp_port) else: server = smtplib.SMTP(self.smtp_server, self.smtp_port) if self.use_tls: server.starttls() if self.auth_required and self.sender_password: server.login(self.sender_email, self.sender_password) return server except Exception as e: logger.error(f"SMTP connection failed: {str(e)}") return None def _send_email(self, to_email: str, subject: str, html_content: str, text_content: str = None) -> bool: """發送郵件的基礎方法""" try: if not self.smtp_server or not self.sender_email: logger.error("SMTP configuration incomplete") return False # 建立郵件 msg = MIMEMultipart('alternative') msg['From'] = f"{self.app_name} <{self.sender_email}>" msg['To'] = to_email msg['Subject'] = subject # 添加文本內容 if text_content: text_part = MIMEText(text_content, 'plain', 'utf-8') msg.attach(text_part) # 添加 HTML 內容 html_part = MIMEText(html_content, 'html', 'utf-8') msg.attach(html_part) # 發送郵件 server = self._create_smtp_connection() if not server: return False server.send_message(msg) server.quit() logger.info(f"Email sent successfully to {to_email}") return True except Exception as e: logger.error(f"Failed to send email to {to_email}: {str(e)}") return False def send_job_completion_notification(self, job: TranslationJob) -> bool: """發送任務完成通知""" try: if not job.user or not job.user.email: logger.warning(f"No email address for job {job.job_uuid}") return False # 準備郵件內容 subject = f"📄 翻譯完成通知 - {job.original_filename}" # 計算處理時間 processing_time = "" if job.processing_started_at and job.completed_at: duration = job.completed_at - job.processing_started_at total_seconds = int(duration.total_seconds()) if total_seconds < 60: processing_time = f"{total_seconds}秒" elif total_seconds < 3600: minutes = total_seconds // 60 seconds = total_seconds % 60 processing_time = f"{minutes}分{seconds}秒" else: hours = total_seconds // 3600 minutes = (total_seconds % 3600) // 60 processing_time = f"{hours}小時{minutes}分" # 生成下載連結（簡化版本） download_links = [] for lang in job.target_languages: download_links.append(f"• {lang}: [下載翻譯檔案]") html_content = f"""

🎉 翻譯任務完成

親愛的 {job.user.display_name}，

您的文件翻譯任務已成功完成！

📋 任務詳細資訊

檔案名稱: {job.original_filename}

任務編號: {job.job_uuid}

來源語言: {job.source_language}

目標語言: {', '.join(job.target_languages)}

處理時間: {processing_time}

完成時間: {job.completed_at.strftime('%Y-%m-%d %H:%M:%S') if job.completed_at else '未知'}

{f'

總成本: ${job.total_cost:.4f}

' if job.total_cost else ''}

📥 下載翻譯檔案

請登入系統下載您的翻譯檔案：

{'
'.join(download_links)}

注意: 翻譯檔案將在系統中保留 7 天，請及時下載。

感謝您使用 {self.app_name}！

如有任何問題，請聯繫系統管理員。

此郵件由 {self.app_name} 系統自動發送，請勿回覆。

發送時間: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}

""" # 純文字版本 text_content = f""" 翻譯任務完成通知 親愛的 {job.user.display_name}， 您的文件翻譯任務已成功完成！ 任務詳細資訊: - 檔案名稱: {job.original_filename} - 任務編號: {job.job_uuid} - 來源語言: {job.source_language} - 目標語言: {', '.join(job.target_languages)} - 處理時間: {processing_time} - 完成時間: {job.completed_at.strftime('%Y-%m-%d %H:%M:%S') if job.completed_at else '未知'} 請登入系統下載您的翻譯檔案。翻譯檔案將在系統中保留 7 天。 感謝您使用 {self.app_name}！ ---- 此郵件由系統自動發送，請勿回覆。 """ return self._send_email(job.user.email, subject, html_content, text_content) except Exception as e: logger.error(f"Failed to send completion notification for job {job.job_uuid}: {str(e)}") return False def send_job_failure_notification(self, job: TranslationJob) -> bool: """發送任務失敗通知""" try: if not job.user or not job.user.email: logger.warning(f"No email address for job {job.job_uuid}") return False subject = f"⚠️ 翻譯失敗通知 - {job.original_filename}" html_content = f"""

❌ 翻譯任務失敗

親愛的 {job.user.display_name}，

很抱歉，您的文件翻譯任務處理失敗。

📋 任務資訊

檔案名稱: {job.original_filename}

任務編號: {job.job_uuid}

重試次數: {job.retry_count}

錯誤訊息: {job.error_message or '未知錯誤'}

失敗時間: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}

建議處理方式:

  • 檢查檔案格式是否正確
  • 確認檔案沒有損壞
  • 稍後再次嘗試上傳
  • 如問題持續，請聯繫系統管理員

如需協助，請聯繫系統管理員。

此郵件由 {self.app_name} 系統自動發送，請勿回覆。

發送時間: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}

""" text_content = f""" 翻譯任務失敗通知 親愛的 {job.user.display_name}， 很抱歉，您的文件翻譯任務處理失敗。 任務資訊: - 檔案名稱: {job.original_filename} - 任務編號: {job.job_uuid} - 重試次數: {job.retry_count} - 錯誤訊息: {job.error_message or '未知錯誤'} 建議處理方式: 1. 檢查檔案格式是否正確 2. 確認檔案沒有損壞 3. 稍後再次嘗試上傳 4. 如問題持續，請聯繫系統管理員 如需協助，請聯繫系統管理員。 ---- 此郵件由 {self.app_name} 系統自動發送，請勿回覆。 """ return self._send_email(job.user.email, subject, html_content, text_content) except Exception as e: logger.error(f"Failed to send failure notification for job {job.job_uuid}: {str(e)}") return False def send_admin_notification(self, subject: str, message: str, admin_emails: List[str] = None) -> bool: """發送管理員通知""" try: if not admin_emails: # 取得所有管理員郵件地址 admin_users = User.get_admin_users() admin_emails = [user.email for user in admin_users if user.email] if not admin_emails: logger.warning("No admin email addresses found") return False html_content = f"""

🔔 系統管理通知

系統管理員您好，

{subject}

{message}

發送時間: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}

此郵件由 {self.app_name} 系統自動發送，請勿回覆。

""" success_count = 0 for email in admin_emails: if self._send_email(email, f"[管理通知] {subject}", html_content): success_count += 1 return success_count > 0 except Exception as e: logger.error(f"Failed to send admin notification: {str(e)}") return False def test_smtp_connection(self) -> bool: """測試 SMTP 連線""" try: server = self._create_smtp_connection() if server: server.quit() return True return False except Exception as e: logger.error(f"SMTP connection test failed: {str(e)}") return False