Ai_GirlFriend/test_db_connection.py
2026-01-31 19:15:41 +08:00

24 lines
605 B
Python

"""测试 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()