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)