生成满足外键约束的Django Fixtures最小子集失败，求助解决

问题场景 #

为Django项目搭建测试数据库时，因全量数据规模大、关联复杂，导入全量Fixture导致测试启动过慢，尝试生成数据子集Fixture却遇到外键约束错误：

• 使用NestedObjects收集EntryTable(pk=1)的关联对象时，生成的Fixture包含了A和ALinkB，但遗漏了ALinkB关联的B表对象，加载Fixture时触发外键约束违反。

用户使用的脚本 #

from itertools import chain
from django.db.utils import DEFAULT_DB_ALIAS
from django.core import serializers
from django.contrib.admin.utils import NestedObjects

from models import EntryTable

collector = NestedObjects(using=DEFAULT_DB_ALIAS)  # database name
collector.collect([EntryTable.objects.get(pk=1)])

objects = list(chain.from_iterable(collector.data.values()))
objects.reverse()
with open("search/fixtures/TestSkeleton.json", "w") as f:
    f.write(serializers.serialize("json", objects))

示例模型 #

from django.db import models
from django.db.models.fields.related import ForeignKey

class EntryTable(models.Model):
    pass

class A(models.Model):
    et = models.ForeignKey(EntryTable)

class B(models.Model):
    pass

class ALinkB(models.Model):
    a = models.ForeignKey(A)
    b = models.ForeignKey(B)

问题原因 #

NestedObjects的核心逻辑是追踪反向关联关系（即从起始对象出发，所有引用它的对象，以及引用这些对象的对象），但不会主动追踪正向关联的对象：

• 从EntryTable(pk=1)出发，能找到反向关联的A对象；
• 从A对象出发，能找到反向关联的ALinkB对象；
• 但ALinkB中的b是正向关联到B，这个关联不在NestedObjects的默认追踪范围内，所以B对象不会被收集。

解决方法 #

需要手动扩展收集逻辑，遍历已收集的所有对象，找出它们的正向外键关联对象，加入收集器中，同时避免重复收集。

修改后的脚本如下：

from itertools import chain
from django.db.utils import DEFAULT_DB_ALIAS
from django.core import serializers
from django.contrib.admin.utils import NestedObjects
from django.db.models import ForeignKey

from models import EntryTable

collector = NestedObjects(using=DEFAULT_DB_ALIAS)
# 先收集起始对象的反向关联
collector.collect([EntryTable.objects.get(pk=1)])

# 遍历已收集的对象，收集正向外键关联的对象
collected_pks = {(obj._meta.label, obj.pk) for obj in chain.from_iterable(collector.data.values())}
new_objects = []

for obj in chain.from_iterable(collector.data.values()):
    # 检查对象的所有正向外键字段
    for field in obj._meta.get_fields():
        if isinstance(field, ForeignKey) and not field.auto_created:
            related_obj = getattr(obj, field.name, None)
            if related_obj:
                obj_key = (related_obj._meta.label, related_obj.pk)
                if obj_key not in collected_pks:
                    collected_pks.add(obj_key)
                    new_objects.append(related_obj)

# 把新收集的对象加入collector，确保它们的关联也被处理
if new_objects:
    collector.collect(new_objects)

# 整理对象并序列化，反转顺序保证依赖对象先创建
objects = list(chain.from_iterable(collector.data.values()))
objects.reverse()

with open("search/fixtures/TestSkeleton.json", "w") as f:
    f.write(serializers.serialize("json", objects))

代码说明 #

1. 先通过NestedObjects收集起始对象的反向关联；
2. 遍历已收集的对象，检查每个对象的正向外键字段，获取关联对象并去重；
3. 将新找到的关联对象再次传入collector.collect()，确保它们的反向关联（如果有的话）也被收集；
4. 最后反转对象顺序，保证依赖对象（如B）先被序列化，加载Fixture时不会出现外键约束错误。

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