如何在 SQLite 中为已有表添加主键（Python 实战解决方案）-Python教程-PHP中文网

如何在 SQLite 中为已有表添加主键（Python 实战解决方案）

SQLite 不支持通过 ALTER TABLE ... ADD PRIMARY KEY 直接添加主键；正确做法是重建表——先创建新表并定义主键，再迁移数据，最后重命名替换原表。

sqlite 不支持通过 `alter table ... add primary key` 直接添加主键；正确做法是重建表——先创建新表并定义主键，再迁移数据，最后重命名替换原表。

在使用 Python 的 sqlite3 模块操作内存数据库（如 ':memory:'）或磁盘数据库时，开发者常误以为 SQLite 支持标准 SQL 的 ALTER TABLE ... ADD PRIMARY KEY 语法。但事实上，SQLite 的 ALTER TABLE 语句功能极为有限：仅支持重命名表（RENAME TO）和添加新列（ADD COLUMN），完全不支持添加、删除或修改主键约束。

因此，以下代码必然报错：

import sqlite3
conn = sqlite3.connect(':memory:')  # 注意：应为 connect()，非 connection()
c = conn.cursor()
c.execute("CREATE TABLE Persons (ID INTEGER, Name TEXT)")  # 先建表
c.execute("ALTER TABLE Persons ADD PRIMARY KEY (ID)")  # ❌ 语法错误：near "PRIMARY"

执行时将抛出 OperationalError: near "PRIMARY": syntax error。

Python 3.14.2

Python 3.14.2是Python编程语言在2025年12月5日发布的稳定版本，属于3.14系列的第二个维护更新。该版本包含了18项修复，重点解决了多进程、数据类及正则表达式等模块的回归问题，并修复了CVE-2025-12084等安全漏洞。此版本标志着自由线程模式（移除GIL）正式获得官方支持，是Python发展的重要里程碑。

下载

✅ 正确解决方案：原子化重建表（Table Reconstruction）
需分四步完成（推荐封装为可复用函数）：

启用写入模式（确保能修改系统表）
获取原表结构（用 PRAGMA table_info()）
构造新表 DDL（显式声明 PRIMARY KEY(ID)）
迁移数据 + 重命名（INSERT INTO ... SELECT + DROP + ALTER TABLE ... RENAME TO）

完整可运行示例：

import sqlite3

def add_primary_key(conn, table_name, pk_column):
    cursor = conn.cursor()

    # 1. 获取原表列信息
    cursor.execute(f"PRAGMA table_info({table_name})")
    columns = cursor.fetchall()
    col_defs = []
    for col in columns:
        name, type_, not_null, default, pk = col[1], col[2], col[3], col[4], col[5]
        # 将目标列设为 PRIMARY KEY，其余保持原约束
        if name == pk_column:
            col_defs.append(f"{name} {type_} PRIMARY KEY")
        else:
            nullable = "NOT NULL" if not_null else ""
            default_part = f"DEFAULT {default}" if default is not None else ""
            col_defs.append(f"{name} {type_} {nullable} {default_part}")

    new_table_name = f"{table_name}_new"
    create_sql = f"CREATE TABLE {new_table_name} ({', '.join(col_defs)});"
    cursor.execute(create_sql)

    # 2. 迁移数据（注意：列名需显式列出，避免顺序错位）
    all_cols = [col[1] for col in columns]
    cursor.execute(f"INSERT INTO {new_table_name} ({', '.join(all_cols)}) SELECT {', '.join(all_cols)} FROM {table_name};")

    # 3. 替换原表
    cursor.execute(f"DROP TABLE {table_name};")
    cursor.execute(f"ALTER TABLE {new_table_name} RENAME TO {table_name};")
    conn.commit()

# 使用示例
conn = sqlite3.connect(':memory:')
c = conn.cursor()
c.execute("CREATE TABLE Persons (ID INTEGER, Name TEXT);")
c.execute("INSERT INTO Persons VALUES (1, 'Alice'), (2, 'Bob');")

add_primary_key(conn, "Persons", "ID")  # ✅ 成功添加主键

# 验证
c.execute("PRAGMA table_info(Persons);")
print(c.fetchall())  # 输出中 ID 行的 pk 字段为 1

⚠️ 注意事项：