37 lines
1.0 KiB
Python
37 lines
1.0 KiB
Python
"""
|
|
数据库工具模块
|
|
"""
|
|
from flask_sqlalchemy import SQLAlchemy
|
|
import sys
|
|
|
|
# 创建数据库实例
|
|
db = SQLAlchemy()
|
|
|
|
def init_db(app):
|
|
"""初始化数据库"""
|
|
db.init_app(app)
|
|
|
|
try:
|
|
with app.app_context():
|
|
# 测试数据库连接
|
|
result = db.session.execute(db.text('SELECT 1'))
|
|
print("✅ 数据库连接成功")
|
|
|
|
# 由于表已存在，我们只需要确保模型与数据库同步
|
|
# 不需要重新创建表
|
|
print("✅ 数据库初始化完成")
|
|
|
|
except Exception as e:
|
|
print(f"❌ 数据库初始化失败: {e}")
|
|
print("请检查数据库配置和网络连接")
|
|
# 在开发环境中不退出，允许继续运行
|
|
print("⚠️ 继续运行，但可能会有数据库相关问题")
|
|
|
|
def test_connection():
|
|
"""测试数据库连接"""
|
|
try:
|
|
result = db.session.execute(db.text('SELECT 1'))
|
|
return True, "数据库连接正常"
|
|
except Exception as e:
|
|
return False, f"数据库连接失败: {str(e)}"
|