83 lines
2.9 KiB
Python
83 lines
2.9 KiB
Python
|
import sqlite3
|
|||
|
|
import os
|
|||
|
|
import logging
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
# 设置日志
|
|||
|
|
logging.basicConfig(level=logging.INFO)
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
# 数据库路径
|
|||
|
|
DB_DIR = "data"
|
|||
|
|
DB_PATH = os.path.join(DB_DIR, "app.db")
|
|||
|
|
|
|||
|
|
def check_and_alter_table():
|
|||
|
|
# 检查数据库文件是否存在
|
|||
|
|
if not os.path.exists(DB_PATH):
|
|||
|
|
logger.error(f"数据库文件 {DB_PATH} 不存在")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
conn = sqlite3.connect(DB_PATH)
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
# 检查talents表是否存在idcode字段
|
|||
|
|
cursor.execute("PRAGMA table_info(talents)")
|
|||
|
|
columns = [column[1] for column in cursor.fetchall()]
|
|||
|
|
|
|||
|
|
if 'idcode' not in columns:
|
|||
|
|
logger.info("talents表中添加idcode字段")
|
|||
|
|
cursor.execute("ALTER TABLE talents ADD COLUMN idcode TEXT")
|
|||
|
|
conn.commit()
|
|||
|
|
else:
|
|||
|
|
logger.info("talents表已包含idcode字段，无需修改")
|
|||
|
|
|
|||
|
|
# 检查talents表是否存在educationBackground字段
|
|||
|
|
if 'educationBackground' not in columns:
|
|||
|
|
logger.info("talents表中添加educationBackground字段")
|
|||
|
|
cursor.execute("ALTER TABLE talents ADD COLUMN educationBackground TEXT")
|
|||
|
|
conn.commit()
|
|||
|
|
else:
|
|||
|
|
logger.info("talents表已包含educationBackground字段，无需修改")
|
|||
|
|
|
|||
|
|
# 检查labs表是否存在idcode字段
|
|||
|
|
cursor.execute("PRAGMA table_info(labs)")
|
|||
|
|
labs_columns = [column[1] for column in cursor.fetchall()]
|
|||
|
|
|
|||
|
|
if 'idcode' not in labs_columns:
|
|||
|
|
logger.info("labs表中添加idcode字段")
|
|||
|
|
cursor.execute("ALTER TABLE labs ADD COLUMN idcode TEXT")
|
|||
|
|
conn.commit()
|
|||
|
|
else:
|
|||
|
|
logger.info("labs表已包含idcode字段，无需修改")
|
|||
|
|
|
|||
|
|
# 检查dimensions表是否存在sub_dimensions字段
|
|||
|
|
cursor.execute("PRAGMA table_info(dimensions)")
|
|||
|
|
columns = [column[1] for column in cursor.fetchall()]
|
|||
|
|
|
|||
|
|
if 'sub_dimensions' not in columns:
|
|||
|
|
logger.info("dimensions表中添加sub_dimensions字段")
|
|||
|
|
cursor.execute("ALTER TABLE dimensions ADD COLUMN sub_dimensions JSON")
|
|||
|
|
conn.commit()
|
|||
|
|
else:
|
|||
|
|
logger.info("dimensions表已包含sub_dimensions字段，无需修改")
|
|||
|
|
|
|||
|
|
# 检查dimensions表是否存在parent_id字段
|
|||
|
|
if 'parent_id' not in columns:
|
|||
|
|
logger.info("dimensions表中添加parent_id字段")
|
|||
|
|
cursor.execute("ALTER TABLE dimensions ADD COLUMN parent_id INTEGER REFERENCES dimensions(id)")
|
|||
|
|
conn.commit()
|
|||
|
|
else:
|
|||
|
|
logger.info("dimensions表已包含parent_id字段，无需修改")
|
|||
|
|
|
|||
|
|
# 检查labs表是否存在sub_dimension_evaluations字段
|
|||
|
|
if 'sub_dimension_evaluations' not in labs_columns:
|
|||
|
|
logger.info("labs表中添加sub_dimension_evaluations字段")
|
|||
|
|
cursor.execute("ALTER TABLE labs ADD COLUMN sub_dimension_evaluations TEXT")
|
|||
|
|
conn.commit()
|
|||
|
|
else:
|
|||
|
|
logger.info("labs表已包含sub_dimension_evaluations字段，无需修改")
|
|||
|
|
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
check_and_alter_table()
|