"""测试 SQLAlchemy 连接"""
from sqlalchemy import text
from lover.db import engine, SessionLocal
from lover.models import Lover

try:
    # 测试连接
    with engine.connect() as conn:
        result = conn.execute(text("SELECT 1"))
        print("✓ Engine 连接成功")
    
    # 测试 Session
    db = SessionLocal()
    try:
        lovers = db.query(Lover).limit(1).all()
        print(f"✓ Session 查询成功，找到 {len(lovers)} 条记录")
    finally:
        db.close()
        
except Exception as e:
    print(f"✗ 失败: {e}")
    import traceback
    traceback.print_exc()