84 lines
3.4 KiB
Python
84 lines
3.4 KiB
Python
|
import os
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
# 添加项目根目录到Python路径
|
|||
|
|
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
|||
|
|
|
|||
|
|
# 在导入应用之前加载环境变量
|
|||
|
|
try:
|
|||
|
|
from dotenv import load_dotenv
|
|||
|
|
if load_dotenv():
|
|||
|
|
print("成功加载.env文件")
|
|||
|
|
except ImportError:
|
|||
|
|
print("python-dotenv未安装，跳过.env文件加载")
|
|||
|
|
|
|||
|
|
from app import create_app, db
|
|||
|
|
from sqlalchemy import text
|
|||
|
|
|
|||
|
|
# 创建应用实例
|
|||
|
|
app = create_app()
|
|||
|
|
|
|||
|
|
with app.app_context():
|
|||
|
|
try:
|
|||
|
|
# 检查是否需要添加 is_deleted 字段
|
|||
|
|
result = db.session.execute(text("SHOW COLUMNS FROM admin LIKE 'is_deleted'"))
|
|||
|
|
if result.fetchone() is None:
|
|||
|
|
print("Adding is_deleted column to admin table...")
|
|||
|
|
db.session.execute(text("ALTER TABLE admin ADD COLUMN is_deleted INTEGER NOT NULL DEFAULT 0"))
|
|||
|
|
print("✓ Added is_deleted column")
|
|||
|
|
else:
|
|||
|
|
print("is_deleted column already exists")
|
|||
|
|
|
|||
|
|
# 检查是否需要添加 delete_time 字段
|
|||
|
|
result = db.session.execute(text("SHOW COLUMNS FROM admin LIKE 'delete_time'"))
|
|||
|
|
if result.fetchone() is None:
|
|||
|
|
print("Adding delete_time column to admin table...")
|
|||
|
|
db.session.execute(text("ALTER TABLE admin ADD COLUMN delete_time DATETIME NULL"))
|
|||
|
|
print("✓ Added delete_time column")
|
|||
|
|
else:
|
|||
|
|
print("delete_time column already exists")
|
|||
|
|
|
|||
|
|
# 检查是否需要添加 phone 字段（如果不存在）
|
|||
|
|
result = db.session.execute(text("SHOW COLUMNS FROM admin LIKE 'phone'"))
|
|||
|
|
if result.fetchone() is None:
|
|||
|
|
print("Adding phone column to admin table...")
|
|||
|
|
db.session.execute(text("ALTER TABLE admin ADD COLUMN phone VARCHAR(16) NULL"))
|
|||
|
|
print("✓ Added phone column")
|
|||
|
|
else:
|
|||
|
|
print("phone column already exists")
|
|||
|
|
|
|||
|
|
# 创建 audit_log 表（如果不存在）
|
|||
|
|
result = db.session.execute(text("SHOW TABLES LIKE 'audit_log'"))
|
|||
|
|
if result.fetchone() is None:
|
|||
|
|
print("Creating audit_log table...")
|
|||
|
|
db.session.execute(text("""
|
|||
|
|
CREATE TABLE audit_log (
|
|||
|
|
log_id INTEGER NOT NULL AUTO_INCREMENT,
|
|||
|
|
admin_id INTEGER NOT NULL,
|
|||
|
|
action VARCHAR(32) NOT NULL,
|
|||
|
|
target_type VARCHAR(32) NOT NULL,
|
|||
|
|
target_id INTEGER,
|
|||
|
|
details TEXT,
|
|||
|
|
ip_address VARCHAR(32),
|
|||
|
|
user_agent VARCHAR(256),
|
|||
|
|
create_time DATETIME NOT NULL,
|
|||
|
|
PRIMARY KEY (log_id),
|
|||
|
|
FOREIGN KEY (admin_id) REFERENCES admin (admin_id)
|
|||
|
|
)
|
|||
|
|
"""))
|
|||
|
|
# 创建索引
|
|||
|
|
db.session.execute(text("CREATE INDEX ix_audit_log_admin_id ON audit_log (admin_id)"))
|
|||
|
|
db.session.execute(text("CREATE INDEX ix_audit_log_action ON audit_log (action)"))
|
|||
|
|
db.session.execute(text("CREATE INDEX ix_audit_log_create_time ON audit_log (create_time)"))
|
|||
|
|
print("✓ Created audit_log table")
|
|||
|
|
else:
|
|||
|
|
print("audit_log table already exists")
|
|||
|
|
|
|||
|
|
# 提交所有更改
|
|||
|
|
db.session.commit()
|
|||
|
|
print("\n✓ Database schema update completed!")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"Database schema update failed: {e}")
|
|||
|
|
import traceback
|
|||
|
|
traceback.print_exc()
|