"""测试数据库连接"""
import os
import sys

# 添加项目根目录到路径
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from app.repositories.database import init_db, engine
from sqlalchemy import text

print("=" * 50)
print("数据库连接测试")
print("=" * 50)

try:
    # 测试连接
    with engine.connect() as conn:
        result = conn.execute(text("SELECT version()"))
        version = result.scalar()
        print(f"✅ 数据库连接成功")
        print(f"   PostgreSQL 版本: {version}")
    
    # 初始化数据库（创建表）
    print("\n正在初始化数据库表...")
    init_db()
    print("✅ 数据库表创建成功")
    
    # 显示所有表
    from sqlalchemy import inspect
    inspector = inspect(engine)
    tables = inspector.get_table_names()
    print(f"\n已创建的表 ({len(tables)} 个):")
    for table in sorted(tables):
        print(f"   - {table}")
    
    print("\n" + "=" * 50)
    print("数据库初始化完成！")
    print("=" * 50)
    
except Exception as e:
    print(f"❌ 错误: {e}")
    print("\n请检查:")
    print("1. Docker 是否已启动: docker ps")
    print("2. 数据库端口是否正确: netstat -ano | findstr 5432")
    print("3. 数据库密码是否正确")
    sys.exit(1)