Django使用bulk_create批量创建含自引用外键实例的可行性咨询

核心结论 #

无法直接在单次bulk_create操作中完成同批次自引用外键的关联，必须分两步操作：先批量创建所有实例，再批量更新外键关联

原因说明 #

1. 你的示例代码存在字段名笔误：模型定义的自引用外键字段为parent_group，但构造实例时写的是group_parent，首先需要修正这个错误，否则会直接触发参数异常。
2. Django外键字段实际存储的是关联对象的主键值，你在构造实例时传入的Group(uid=record["parent_uid"])是未持久化的临时实例，不存在主键值。
3. bulk_create是一次性提交所有插入请求到数据库，插入执行前所有同批次实例都未生成主键，数据库无法匹配到对应的外键关联值，直接插入会导致外键字段为空或触发外键约束报错。

正确实现方案 #

步骤1：批量创建所有实例，暂不填充外键 #

构造实例时仅赋值非外键字段，执行bulk_create，支持RETURNING语法的数据库（PostgreSQL、SQLite 3.35+、MySQL 8.0.21+）会自动为实例填充生成的主键：

df_records = df.to_dict('records')
# 先构造所有实例，不设置parent_group
model_instances = [Group(
    uid = record["uid"],
    label = record["label"],
    ) for record in df_records]
# 批量插入，实例会自动获得主键
Group.objects.bulk_create(model_instances)

步骤2：建立uid到实例的映射 #

uid_to_group = {group.uid: group for group in model_instances}

步骤3：填充外键后批量更新 #

for record, group in zip(df_records, model_instances):
    parent_uid = record.get("parent_uid")
    # 仅当父组uid存在于本次批次中时赋值
    if parent_uid and parent_uid in uid_to_group:
        group.parent_group = uid_to_group[parent_uid]
# 批量更新仅修改parent_group字段，效率更高
Group.objects.bulk_update(model_instances, fields=["parent_group"])

如果你使用的数据库不支持bulk_create自动填充主键，执行完插入后可以再调用一次Group.objects.in_bulk([record["uid"] for record in df_records], field_name="uid")拿到所有实例，再做后续关联即可。

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