#!/usr/bin/env python3 # -*- coding: utf-8 -*- """ 邮件发送模块负责处理用户注册、登录验证邮件 """ import random import string import smtplib from datetime import datetime, timedelta from email.mime.text import MIMEText from email.header import Header from flask import current_app import logging import os # 验证码存储（生产环境建议使用Redis） verification_codes = {} # 初始化日志 def init_mail(app): """初始化邮件配置""" # 使用smtplib直接发送，不需要Flask-Mail pass # 生成验证码 def generate_verification_code(length=6): """生成验证码""" return ''.join(random.choices(string.digits, k=length)) # 发送验证邮件 def send_verification_email(email, verification_type='register'): """ 发送验证邮件 Args: email: 收件人邮箱 verification_type: 验证类型 ('register', 'login', 'reset_password') Returns: dict: 发送结果 """ try: # 验证QQ邮箱格式 if not is_qq_email(email): return { 'success': False, 'message': '仅支持QQ邮箱注册登录' } # 生成验证码 code = generate_verification_code() # 存储验证码（5分钟有效期） verification_codes[email] = { 'code': code, 'type': verification_type, 'expires_at': datetime.now() + timedelta(minutes=5), 'attempts': 0 } # 获取邮件配置 - 使用与QQEmailSendAPI相同的配置 sender_email = os.environ.get('MAIL_USERNAME', '3205788256@qq.com') sender_password = os.environ.get('MAIL_PASSWORD', 'szcaxvbftusqddhi') # 邮件模板 if verification_type == 'register': subject = '【InfoGenie】注册验证码' html_content = f'''

InfoGenie 神奇万事通

欢迎注册InfoGenie

验证码

{code}

请在5分钟内输入此验证码完成注册

如果您没有申请注册，请忽略此邮件
此验证码5分钟内有效，请勿泄露给他人

''' else: # login subject = '【InfoGenie】登录验证码' html_content = f'''

InfoGenie 神奇万事通

安全登录验证

登录验证码

{code}

请在5分钟内输入此验证码完成登录

如果不是您本人操作，请检查账户安全
此验证码5分钟内有效，请勿泄露给他人

''' # 创建邮件 - 使用与QQEmailSendAPI相同的方式 message = MIMEText(html_content, 'html', 'utf-8') message['From'] = sender_email # 直接使用邮箱地址，不使用Header包装 message['To'] = email message['Subject'] = Header(subject, 'utf-8') # 发送邮件 - 使用SSL端口465 try: # 使用与QQEmailSendAPI相同的连接方式 smtp_obj = smtplib.SMTP_SSL('smtp.qq.com', 465) smtp_obj.login(sender_email, sender_password) smtp_obj.sendmail(sender_email, [email], message.as_string()) smtp_obj.quit() print(f"验证码邮件发送成功: {email}") return { 'success': True, 'message': '验证码已发送到您的邮箱', 'email': email } except smtplib.SMTPAuthenticationError as auth_error: print(f"SMTP认证失败: {str(auth_error)}") return { 'success': False, 'message': 'SMTP认证失败，请检查邮箱配置' } except smtplib.SMTPConnectError as conn_error: print(f"SMTP连接失败: {str(conn_error)}") return { 'success': False, 'message': 'SMTP服务器连接失败' } except Exception as smtp_error: print(f"SMTP发送失败: {str(smtp_error)}") return { 'success': False, 'message': f'邮件发送失败: {str(smtp_error)}' } except Exception as e: print(f"邮件发送失败: {str(e)}") return { 'success': False, 'message': '邮件发送失败，请稍后重试' } # 验证验证码 def verify_code(email, code): """ 验证验证码 Args: email: 邮箱地址 code: 验证码 Returns: dict: 验证结果 """ if email not in verification_codes: return { 'success': False, 'message': '验证码不存在或已过期' } stored_info = verification_codes[email] # 检查过期时间 if datetime.now() > stored_info['expires_at']: del verification_codes[email] return { 'success': False, 'message': '验证码已过期，请重新获取' } # 检查尝试次数 if stored_info['attempts'] >= 3: del verification_codes[email] return { 'success': False, 'message': '验证码输入错误次数过多，请重新获取' } # 验证码校验 if stored_info['code'] != code: stored_info['attempts'] += 1 return { 'success': False, 'message': f'验证码错误，还可尝试{3 - stored_info["attempts"]}次' } # 验证成功，删除验证码 verification_type = stored_info['type'] del verification_codes[email] return { 'success': True, 'message': '验证码验证成功', 'type': verification_type } # 验证QQ邮箱格式 def is_qq_email(email): """ 验证是否为QQ邮箱 Args: email: 邮箱地址 Returns: bool: 是否为QQ邮箱 """ if not email or '@' not in email: return False domain = email.split('@')[1].lower() qq_domains = ['qq.com', 'vip.qq.com', 'foxmail.com'] return domain in qq_domains # 获取QQ头像URL def get_qq_avatar_url(email): """ 根据QQ邮箱获取QQ头像URL Args: email: QQ邮箱地址 Returns: str: QQ头像URL """ if not is_qq_email(email): return None # 提取QQ号码 qq_number = email.split('@')[0] # 验证是否为纯数字（QQ号） if not qq_number.isdigit(): return None # 返回QQ头像API URL return f"http://q1.qlogo.cn/g?b=qq&nk={qq_number}&s=100" # 清理过期验证码 def cleanup_expired_codes(): """清理过期的验证码""" current_time = datetime.now() expired_emails = [ email for email, info in verification_codes.items() if current_time > info['expires_at'] ] for email in expired_emails: del verification_codes[email] return len(expired_emails)