133 lines
5.2 KiB
Python
133 lines
5.2 KiB
Python
import datetime
|
|
|
|
# 导入数据库工具
|
|
from db_utils import get_db_connection, close_db_connection
|
|
|
|
# 连接到数据库并确保users表结构正确
|
|
def update_users_table():
|
|
try:
|
|
# 连接数据库
|
|
conn = get_db_connection()
|
|
cursor = conn.cursor()
|
|
|
|
# 检查users表的当前结构
|
|
cursor.execute("PRAGMA table_info(users)")
|
|
columns = [column[1] for column in cursor.fetchall()]
|
|
print(f"当前users表的字段: {columns}")
|
|
|
|
# 使用不同的方法来添加字段，避免NOT NULL约束可能导致的问题
|
|
if 'name' not in columns:
|
|
# 先添加可为空的字段
|
|
cursor.execute("ALTER TABLE users ADD COLUMN name TEXT")
|
|
# 然后设置默认值并更新现有数据
|
|
cursor.execute("UPDATE users SET name = '未知用户' WHERE name IS NULL")
|
|
print("已添加name字段并设置默认值")
|
|
|
|
if 'phone' not in columns:
|
|
# 先添加可为空的字段
|
|
cursor.execute("ALTER TABLE users ADD COLUMN phone TEXT")
|
|
# 然后设置默认值并更新现有数据
|
|
cursor.execute("UPDATE users SET phone = '00000000000' WHERE phone IS NULL")
|
|
print("已添加phone字段并设置默认值")
|
|
|
|
if 'department' not in columns:
|
|
# 先添加可为空的字段
|
|
cursor.execute("ALTER TABLE users ADD COLUMN department TEXT")
|
|
# 然后设置默认值并更新现有数据
|
|
cursor.execute("UPDATE users SET department = '未知部门' WHERE department IS NULL")
|
|
print("已添加department字段并设置默认值")
|
|
|
|
# 无条件更新admin用户的信息，确保所有字段都有值
|
|
cursor.execute(
|
|
"UPDATE users SET name = ?, phone = ?, department = ? WHERE username = ?",
|
|
('管理员', '13800138000', 'IT部', 'admin')
|
|
)
|
|
print("已更新admin用户信息")
|
|
|
|
# 提交更改
|
|
try:
|
|
conn.commit()
|
|
except Exception as e:
|
|
print(f"提交更改时出错: {e}")
|
|
|
|
# 再次验证表结构
|
|
cursor.execute("PRAGMA table_info(users)")
|
|
updated_columns = cursor.fetchall()
|
|
print("\n更新后的users表字段:")
|
|
for col in updated_columns:
|
|
print(f"字段名: {col[1]}, 类型: {col[2]}")
|
|
|
|
# 检查admin用户的信息是否正确更新
|
|
cursor.execute("SELECT id, username, name, phone, department FROM users WHERE username = ?", ('admin',))
|
|
admin_user = cursor.fetchone()
|
|
if admin_user:
|
|
print("\n更新后的admin用户信息:")
|
|
print(f"ID: {admin_user[0]}, 用户名: {admin_user[1]}, 姓名: {admin_user[2]}, 电话: {admin_user[3]}, 部门: {admin_user[4]}")
|
|
|
|
# 关闭连接
|
|
close_db_connection(conn)
|
|
print("\n数据库更新完成")
|
|
|
|
except Exception as e:
|
|
print(f"更新数据库时出错: {str(e)}")
|
|
# 确保在异常情况下也关闭连接
|
|
if 'conn' in locals() and conn:
|
|
close_db_connection(conn)
|
|
|
|
# 更新数据库中的数据
|
|
def update_database():
|
|
try:
|
|
# 连接数据库
|
|
conn = get_db_connection()
|
|
cursor = conn.cursor()
|
|
|
|
# 检查是否存在测试数据表
|
|
cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='test_data'")
|
|
if not cursor.fetchone():
|
|
# 如果不存在，创建测试数据表
|
|
cursor.execute('''
|
|
CREATE TABLE test_data (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
name TEXT NOT NULL,
|
|
value TEXT,
|
|
create_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
|
update_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
|
)
|
|
''')
|
|
conn.commit()
|
|
print("测试数据表创建成功")
|
|
|
|
# 检查是否有测试数据，如果没有则插入
|
|
cursor.execute("SELECT COUNT(*) FROM test_data")
|
|
count = cursor.fetchone()[0]
|
|
|
|
if count == 0:
|
|
# 插入测试数据
|
|
test_data = [
|
|
('测试数据1', 'value1', datetime.datetime.now()),
|
|
('测试数据2', 'value2', datetime.datetime.now()),
|
|
('测试数据3', 'value3', datetime.datetime.now())
|
|
]
|
|
|
|
for data in test_data:
|
|
cursor.execute(
|
|
"INSERT INTO test_data (name, value, create_time, update_time) VALUES (?, ?, ?, ?)",
|
|
data
|
|
)
|
|
conn.commit()
|
|
print("测试数据插入成功")
|
|
|
|
# 关闭数据库连接
|
|
close_db_connection(conn)
|
|
return True, "数据库更新成功"
|
|
|
|
except Exception as e:
|
|
# 确保在异常情况下也关闭连接
|
|
if 'conn' in locals() and conn:
|
|
close_db_connection(conn)
|
|
return False, f"数据库更新失败：{str(e)}"
|
|
|
|
# 如果直接运行此脚本，则执行更新操作
|
|
if __name__ == "__main__":
|
|
success, message = update_database()
|
|
print(message) |