Simon
25 lines
746 B
Python
25 lines
746 B
Python
import sqlite3
|
|
conn = sqlite3.connect('it_smart_desk.db')
|
|
cursor = conn.cursor()
|
|
|
|
# Employee 表新增字段
|
|
alterations = [
|
|
("it_level", "VARCHAR(20)", "'silver'"),
|
|
("it_level_source", "VARCHAR(20)", "'system'"),
|
|
("notes", "JSON", "'{}'"),
|
|
]
|
|
|
|
for col_name, col_type, default_val in alterations:
|
|
try:
|
|
cursor.execute(f"ALTER TABLE employees ADD COLUMN {col_name} {col_type} NOT NULL DEFAULT {default_val}")
|
|
print(f"ADDED: employees.{col_name}")
|
|
except sqlite3.OperationalError as e:
|
|
if "duplicate column" in str(e).lower():
|
|
print(f"SKIP: employees.{col_name} already exists")
|
|
else:
|
|
print(f"ERROR: employees.{col_name} — {e}")
|
|
|
|
conn.commit()
|
|
conn.close()
|
|
print("Done.")
|