Peewee未指定id列时自动创建主键的机制探究

Hey there! Let's break down how Peewee handles primary key creation when you don't explicitly define an id column in your model, plus key points to keep in mind while building your SQL Server connector with SQL Relay.

1. 核心流程：SchemaManager的_create_table方法 #

Peewee的表创建逻辑核心就在SchemaManager._create_table里，不过自动主键的生成逻辑其实在模型初始化阶段就已经启动了：

• 当你定义的模型里没有任何标记了primary_key=True的字段时，Peewee会在模型元数据初始化时自动注入一个名为id的AutoField作为主键。
• 当_create_table方法执行时，它会收集模型的所有排序字段（包括这个自动生成的id列），再用这些字段构建CREATE TABLE语句。

你提到的方法开头片段可以帮你理解上下文：

def _create_table(self, safe=True, **options):
    is_temp = options.pop('temporary', False)
    # 收集字段，包括自动生成的主键（如果需要）
    fields = self.model._meta.sorted_fields
    # 基于这些字段生成CREATE TABLE语句
    # ...后续逻辑会处理主键约束的生成

2. 适配不同数据库的自动主键类型 #

Peewee会根据目标数据库自动调整自动主键的类型：

• 对于SQL Server，自动生成的id列会是INT IDENTITY(1,1)（这是SQL Server的自增列语法）。
• 对于PostgreSQL会用SERIAL/BIGSERIAL，MySQL则是INT AUTO_INCREMENT。

如果你要基于SQL Relay编写自定义SQL Server连接器，要确保数据库类里正确映射AutoField对应的SQL语法：

class SQLRelaySQLServerDatabase(Database):
    field_types = {
        AutoField: 'INT IDENTITY(1,1)',
        # 在这里添加其他字段类型的映射
    }

3. 自定义主键的覆盖逻辑 #

如果你的模型里显式定义了带primary_key=True的字段，Peewee就不会自动生成id列了。比如：

class User(Model):
    user_uuid = UUIDField(primary_key=True)  # 自定义主键
    username = CharField(max_length=50)

这种情况下，_create_table会直接用user_uuid作为主键生成表结构，不会额外添加id列。

4. 针对SQL Relay连接器的注意事项 #

在编写自定义连接器时，这些要点要记牢：

• 除非有特殊需求（比如强制使用其他默认主键名），否则不要覆盖Peewee默认的自动主键注入逻辑。
• 测试生成的CREATE TABLE语句，确保自动主键的语法符合SQL Server要求（重点检查IDENTITY子句）。
• 如果SQL Relay对表创建有特殊语法要求，你可以重写_create_table的部分逻辑来调整SQL生成流程，但要保留主键处理的核心逻辑。

内容的提问来源于stack exchange，提问作者Nicholas Tulach