"""
测试数据库连接脚本
"""

import os
import sys
from dotenv import load_dotenv

# 手动加载.env文件
load_dotenv()

# 添加项目路径
sys.path.insert(0, os.path.dirname(__file__))
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src'))

from src.config.settings import settings
from src.storage.database import db_manager

def test_database_connection():
	"""测试数据库连接"""
	print("=== 测试数据库连接 ===")
	print(f"数据库URL: {settings.database.database_url}")
	
	try:
		# 测试获取会话
		session = db_manager.get_session()
		print("✅ 数据库连接成功")
		
		# 测试创建表
		db_manager.create_tables()
		print("✅ 数据库表创建成功")
		
		session.close()
		print("✅ 数据库会话关闭成功")
		
	except Exception as e:
		print(f"❌ 数据库连接失败: {e}")
		return False
	
	return True

if __name__ == "__main__":
	test_database_connection()