first commit

old/app/services/mail_store.py

@@ -0,0 +1,140 @@
+import logging
+import os
+import email
+from email.policy import default
+from sqlalchemy.orm import Session
+from datetime import datetime
+import re
+import redis
+
+from ..models.domain import Domain
+from ..models.mailbox import Mailbox
+from ..models.email import Email
+from ..models.attachment import Attachment
+
+logging.basicConfig(
+    level=logging.DEBUG,  # 设置日志级别为DEBUG
+    format='%(asctime)s - %(levelname)s - %(message)s',  # 日志格式
+    filename='app.log',  # 日志文件名
+    filemode='a'  # 追加模式
+)
+
+logger = logging.getLogger(__name__)
+
+# 连接到 Redis
+redis_client = redis.StrictRedis(host='localhost', port=6379, db=0)
+
+class MailStore:
+    """邮件存储服务，负责保存和检索邮件"""
+    
+    def __init__(self, db_session_factory, storage_path=None):
+        """
+        初始化邮件存储服务
+        
+        参数:
+            db_session_factory: 数据库会话工厂函数
+            storage_path: 附件存储路径
+        """
+        self.db_session_factory = db_session_factory
+        self.storage_path = storage_path or os.path.join(os.getcwd(), 'email_data')
+        
+        # 确保存储目录存在
+        if not os.path.exists(self.storage_path):
+            os.makedirs(self.storage_path)
+            
+    async def save_email(self, message, sender, recipients, raw_data=None):
+        logging.info(f"开始保存邮件: 发件人={sender}, 收件人={recipients}")
+        
+        # 处理收件人列表
+        if recipients is None:
+            logging.error("收件人列表为None，无法保存邮件")
+            return False, "收件人列表为None"
+        elif isinstance(recipients, list):
+            recipients_list = recipients  # 如果是列表，直接使用
+        else:
+            recipients_list = recipients.split(",")  # 假设是以逗号分隔的字符串
+
+        # 确保收件人列表不为空
+        if not recipients_list:
+            logging.error("收件人列表为空，无法保存邮件")
+            return False, "收件人列表为空"
+
+        # 解析邮件内容
+        email_subject = message.subject if message.subject else "无主题"
+        body_text = message.get_body(preferencelist=('plain')).get_content()
+        received_at = datetime.now().isoformat()
+
+        # 存储邮件到 Redis
+        for recipient in recipients_list:
+            email_id = f"email:{recipient}:{received_at}"
+            redis_client.hset(email_id, mapping={
+                "subject": email_subject,
+                "sender": sender,
+                "recipients": recipients,
+                "body": body_text,
+                "received_at": received_at
+            })
+            logging.info(f"邮件已保存到 Redis: {email_id}")
+
+        return True, "邮件已保存到 Redis"
+            
+    def get_emails_for_mailbox(self, mailbox_id, limit=50, offset=0, unread_only=False):
+        """获取指定邮箱的邮件列表"""
+        try:
+            # 从 Redis 获取邮件
+            keys = redis_client.keys(f"email:*")
+            emails = []
+            for key in keys:
+                email_data = redis_client.hgetall(key)
+                emails.append(email_data)
+
+            total = len(emails)
+            return {
+                'total': total,
+                'items': emails[offset:offset + limit]
+            }
+        except Exception as e:
+            logger.error(f"获取邮件列表时出错: {str(e)}")
+            return {'total': 0, 'items': []}
+            
+    def get_email_by_id(self, email_id):
+        """获取指定ID的邮件详情"""
+        try:
+            email_data = redis_client.hgetall(f"email:{email_id}")
+            if not email_data:
+                return None
+            return email_data
+        except Exception as e:
+            logger.error(f"获取邮件详情时出错: {str(e)}")
+            return None
+            
+    def delete_email(self, email_id):
+        """删除指定ID的邮件"""
+        try:
+            redis_client.delete(f"email:{email_id}")
+            return True
+        except Exception as e:
+            logger.error(f"删除邮件时出错: {str(e)}")
+            return False
+            
+    def get_attachment_content(self, attachment_id):
+        """获取附件内容"""
+        db = self.db_session_factory()
+        try:
+            attachment = db.query(Attachment).filter(Attachment.id == attachment_id).first()
+            
+            if not attachment:
+                return None
+                
+            content = attachment.get_content()
+            
+            return {
+                'content': content,
+                'filename': attachment.filename,
+                'content_type': attachment.content_type
+            }
+        except Exception as e:
+            logger.error(f"获取附件内容时出错: {str(e)}")
+            return None
+        finally:
+            db.close()