123 lines
3.9 KiB
Python
123 lines
3.9 KiB
Python
import logging
|
||
import re
|
||
import threading
|
||
import time
|
||
from queue import Queue
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
class EmailProcessor:
|
||
"""邮件处理器，负责处理邮件并提取验证信息"""
|
||
|
||
def __init__(self, mail_store):
|
||
"""
|
||
初始化邮件处理器
|
||
|
||
参数:
|
||
mail_store: 邮件存储服务实例
|
||
"""
|
||
self.mail_store = mail_store
|
||
self.processing_queue = Queue()
|
||
self.is_running = False
|
||
self.worker_thread = None
|
||
|
||
def start(self):
|
||
"""启动邮件处理器"""
|
||
if self.is_running:
|
||
logger.warning("邮件处理器已在运行")
|
||
return False
|
||
|
||
self.is_running = True
|
||
self.worker_thread = threading.Thread(
|
||
target=self._processing_worker,
|
||
daemon=True
|
||
)
|
||
self.worker_thread.start()
|
||
logger.info("邮件处理器已启动")
|
||
return True
|
||
|
||
def stop(self):
|
||
"""停止邮件处理器"""
|
||
if not self.is_running:
|
||
logger.warning("邮件处理器未在运行")
|
||
return False
|
||
|
||
self.is_running = False
|
||
if self.worker_thread:
|
||
self.worker_thread.join(timeout=5.0)
|
||
self.worker_thread = None
|
||
|
||
logger.info("邮件处理器已停止")
|
||
return True
|
||
|
||
def queue_email_for_processing(self, email_id):
|
||
"""将邮件添加到处理队列"""
|
||
self.processing_queue.put(email_id)
|
||
return True
|
||
|
||
def _processing_worker(self):
|
||
"""处理队列中的邮件的工作线程"""
|
||
while self.is_running:
|
||
try:
|
||
# 获取队列中的邮件，最多等待1秒
|
||
try:
|
||
email_id = self.processing_queue.get(timeout=1.0)
|
||
except:
|
||
continue
|
||
|
||
# 处理邮件
|
||
self._process_email(email_id)
|
||
|
||
# 标记任务完成
|
||
self.processing_queue.task_done()
|
||
|
||
except Exception as e:
|
||
logger.error(f"处理邮件时出错: {str(e)}")
|
||
|
||
def _process_email(self, email_id):
|
||
"""处理单个邮件，提取验证码和链接"""
|
||
# 从邮件存储获取邮件
|
||
email_data = self.mail_store.get_email_by_id(email_id, mark_as_read=False)
|
||
if not email_data:
|
||
logger.warning(f"找不到ID为 {email_id} 的邮件")
|
||
return False
|
||
|
||
# 提取验证码和链接已经在Email模型的extract_verification_data方法中实现
|
||
# 这里可以添加更复杂的提取逻辑或后处理
|
||
|
||
logger.info(f"邮件 {email_id} 处理完成")
|
||
return True
|
||
|
||
@staticmethod
|
||
def extract_verification_code(content):
|
||
"""从内容中提取验证码"""
|
||
code_patterns = [
|
||
r'\b[A-Z0-9]{4,8}\b', # 基本验证码格式
|
||
r'验证码[：:]\s*([A-Z0-9]{4,8})',
|
||
r'验证码是[：:]\s*([A-Z0-9]{4,8})',
|
||
r'code[：:]\s*([A-Z0-9]{4,8})',
|
||
r'码[：:]\s*(\d{4,8})' # 纯数字验证码
|
||
]
|
||
|
||
for pattern in code_patterns:
|
||
matches = re.findall(pattern, content, re.IGNORECASE)
|
||
if matches:
|
||
return matches[0]
|
||
|
||
return None
|
||
|
||
@staticmethod
|
||
def extract_verification_link(content):
|
||
"""从内容中提取验证链接"""
|
||
link_patterns = [
|
||
r'(https?://\S+(?:verify|confirm|activate)\S+)',
|
||
r'(https?://\S+(?:token|auth|account)\S+)',
|
||
r'href\s*=\s*["\']([^"\']+(?:verify|confirm|activate)[^"\']*)["\']'
|
||
]
|
||
|
||
for pattern in link_patterns:
|
||
matches = re.findall(pattern, content, re.IGNORECASE)
|
||
if matches:
|
||
return matches[0]
|
||
|
||
return None |