Tolan/backend/send_email_code.py

"""
发送邮箱验证码脚本
扫描 verification_codes 表，查找未发送的邮箱验证码并发送

运行: python send_email_code.py
"""
import asyncio
import sys
import logging
from datetime import datetime
from typing import List, Optional

import aiomysql
from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession

from config import settings
from database import DatabaseManager
from models_db import VerificationCode

logger = logging.getLogger(__name__)


async def send_email_code(email: str, code: str) -> bool:
    """
    发送邮箱验证码
    
    Args:
        email: 邮箱地址
        code: 验证码
    
    Returns:
        是否发送成功
    """
    # TODO: 实现邮箱发送 API 调用
    # 示例：
    # async with httpx.AsyncClient() as client:
    #     response = await client.post(
    #         "https://api.email-provider.com/send",
    #         json={
    #             "to": email,
    #             "subject": "验证码",
    #             "template": "verification_code",
    #             "code": code
    #         }
    #     )
    #     return response.status_code == 200
    
    print(f"📧 [TODO] 发送邮箱验证码: email={email}, code={code}")
    return False


async def process_email_codes():
    """扫描并发送邮箱验证码"""
    db_manager = DatabaseManager()
    
    try:
        await db_manager.connect()
        print("✅ 数据库连接成功")
        
        async with db_manager.get_session() as session:
            # 查询未发送的邮箱验证码
            result = await session.execute(
                select(VerificationCode)
                .where(
                    VerificationCode.send_type == "email",
                    VerificationCode.is_sent == False,
                    VerificationCode.is_valid == True,
                    VerificationCode.email.isnot(None)
                )
                .order_by(VerificationCode.created_at.asc())
                .limit(100)  # 每次处理100条
            )
            codes = result.scalars().all()
            
            if not codes:
                print("ℹ️ 没有待发送的邮箱验证码")
                return
            
            print(f"📋 找到 {len(codes)} 条待发送的邮箱验证码")
            
            success_count = 0
            fail_count = 0
            
            for code_record in codes:
                try:
                    # 发送邮箱
                    success = await send_email_code(code_record.email, code_record.code)
                    
                    if success:
                        # 更新发送状态
                        code_record.is_sent = True
                        code_record.sent_at = datetime.now()
                        code_record.metadata_json = code_record.metadata_json or {}
                        code_record.metadata_json["send_status"] = "success"
                        code_record.metadata_json["send_at"] = datetime.now().isoformat()
                        success_count += 1
                        print(f"✅ 发送成功: code_id={code_record.code_id}, email={code_record.email}")
                    else:
                        # 记录失败信息
                        code_record.metadata_json = code_record.metadata_json or {}
                        code_record.metadata_json["send_status"] = "failed"
                        code_record.metadata_json["error"] = "API调用失败"
                        fail_count += 1
                        print(f"❌ 发送失败: code_id={code_record.code_id}, email={code_record.email}")
                    
                    await session.commit()
                    
                except Exception as e:
                    print(f"❌ 处理验证码失败: code_id={code_record.code_id}, error={e}")
                    code_record.metadata_json = code_record.metadata_json or {}
                    code_record.metadata_json["send_status"] = "error"
                    code_record.metadata_json["error"] = str(e)
                    await session.commit()
                    fail_count += 1
            
            print(f"\n📊 处理完成: 成功 {success_count} 条, 失败 {fail_count} 条")
            
    except Exception as e:
        print(f"❌ 处理失败: {e}")
        import traceback
        traceback.print_exc()
        sys.exit(1)
    finally:
        await db_manager.close()


if __name__ == "__main__":
    logging.basicConfig(
        level=logging.INFO,
        format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
        datefmt='%Y-%m-%d %H:%M:%S'
    )
    
    asyncio.run(process_email_codes())