import json
|
import pymysql
|
import logging
|
from database import DatabaseConfig
|
|
# 设置日志记录
|
logging.basicConfig(level=logging.INFO)
|
logger = logging.getLogger(__name__)
|
|
|
def update_database():
|
"""更新数据库，添加order字段到content表"""
|
try:
|
with open("config.json") as f:
|
config = json.load(f)
|
except FileNotFoundError:
|
logger.error("配置文件 config.json 未找到")
|
raise
|
except json.JSONDecodeError:
|
logger.error("配置文件 config.json 中JSON格式无效")
|
raise
|
|
if "database" not in config:
|
logger.error("配置文件中未找到数据库配置")
|
raise ValueError("数据库配置缺失")
|
|
db_config = config["database"]
|
|
# 连接数据库
|
try:
|
connection = pymysql.connect(
|
host=db_config.get("host", "localhost"),
|
port=int(db_config.get("port", 3306)),
|
user=db_config.get("user", "root"),
|
password=db_config.get("password", ""),
|
database=db_config.get("database_name", "admin_db"),
|
charset=db_config.get("charset", "utf8mb4"),
|
)
|
except Exception as e:
|
logger.error(f"建立数据库连接时出错: {e}")
|
raise
|
|
try:
|
with connection.cursor() as cursor:
|
# 检查字段是否已存在
|
check_column_sql = """
|
SELECT COLUMN_NAME
|
FROM INFORMATION_SCHEMA.COLUMNS
|
WHERE TABLE_SCHEMA = %s AND TABLE_NAME = 'content' AND COLUMN_NAME = 'order'
|
"""
|
cursor.execute(check_column_sql, (db_config.get('database_name', 'admin_db'),))
|
result = cursor.fetchone()
|
|
if result:
|
logger.info("order字段已存在，无需更新")
|
return
|
|
# 添加order字段
|
alter_table_sql = """
|
ALTER TABLE content ADD COLUMN `order` INT DEFAULT 0
|
"""
|
cursor.execute(alter_table_sql)
|
connection.commit()
|
|
logger.info("数据库更新成功! 已添加order字段到content表")
|
except Exception as e:
|
logger.error(f"更新数据库时出错: {e}")
|
raise
|
finally:
|
connection.close()
|
|
|
if __name__ == "__main__":
|
# 初始化数据库配置
|
db_config = DatabaseConfig()
|
update_database()
|
