43 lines
926 B
Python
43 lines
926 B
Python
"""
|
|
测试数据库连接脚本
|
|
"""
|
|
|
|
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() |