如何自动为Django模型同时填充外键与多对多字段？

1. 使用Factory Boy（推荐） #

Factory Boy是Django生态里最流行的测试/填充数据工具，能自动处理外键、多对多关联，完全不用手动编写fixture。

安装依赖 #

pip install factory_boy faker

示例实现 #

假设你有以下模型：

# models.py
from django.db import models

class Author(models.Model):
    name = models.CharField(max_length=100)
    email = models.EmailField()

class Book(models.Model):
    title = models.CharField(max_length=200)
    author = models.ForeignKey(Author, on_delete=models.CASCADE, related_name='books')
    categories = models.ManyToManyField('Category')

class Category(models.Model):
    name = models.CharField(max_length=50)

创建对应的工厂类：

# factories.py
import factory
from faker import Faker
from .models import Author, Book, Category

fake = Faker()

class AuthorFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Author

    name = factory.LazyAttribute(lambda _: fake.name())
    email = factory.LazyAttribute(lambda _: fake.email())

class CategoryFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Category

    name = factory.LazyAttribute(lambda _: fake.word(ext_word_list=['科幻', '历史', '文学', '编程']))

class BookFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Book

    title = factory.LazyAttribute(lambda _: fake.sentence(nb_words=3))
    # 自动创建关联的Author实例
    author = factory.SubFactory(AuthorFactory)

    # 处理多对多字段：实例创建后自动关联
    @factory.post_generation
    def categories(self, create, extracted, **kwargs):
        if not create:
            return
        if extracted:
            # 如果调用时传入了categories参数，直接使用传入的分类
            for category in extracted:
                self.categories.add(category)
        else:
            # 自动生成2-3个分类并关联
            categories = CategoryFactory.create_batch(size=fake.random_int(min=2, max=3))
            self.categories.add(*categories)

填充数据 #

在Django Shell或自定义脚本中执行：

# 创建10个作者，每个作者自动关联1-3本书
for _ in range(10):
    author = AuthorFactory()
    BookFactory.create_batch(size=fake.random_int(min=1, max=3), author=author)

# 直接批量创建带完整关联的书籍
BookFactory.create_batch(20)

2. 自定义Django管理命令 #

如果不想依赖第三方库，可以自己编写管理命令实现自动填充。

创建命令文件 #

# myapp/management/commands/seed_data.py
from django.core.management.base import BaseCommand
from faker import Faker
from myapp.models import Author, Book, Category

fake = Faker()

class Command(BaseCommand):
    help = '为数据库填充测试数据'

    def add_arguments(self, parser):
        parser.add_argument('--authors', type=int, default=10, help='要创建的作者数量')
        parser.add_argument('--books-per-author', type=int, default=2, help='每个作者关联的书籍数量')

    def handle(self, *args, **options):
        author_count = options['authors']
        books_per_author = options['books_per_author']

        # 先初始化一批分类（避免重复创建）
        categories = [Category.objects.get_or_create(name=name)[0] for name in ['科幻', '历史', '文学', '编程']]

        for _ in range(author_count):
            author = Author.objects.create(
                name=fake.name(),
                email=fake.email()
            )
            for _ in range(books_per_author):
                book = Book.objects.create(
                    title=fake.sentence(nb_words=3),
                    author=author
                )
                # 随机选2-3个分类关联
                selected_cats = fake.random_elements(elements=categories, length=fake.random_int(min=2, max=3), unique=True)
                book.categories.add(*selected_cats)

        self.stdout.write(self.style.SUCCESS(f'成功填充{author_count}位作者，共{author_count * books_per_author}本书'))

运行命令 #

python manage.py seed_data --authors 15 --books-per-author 3

3. 使用Django Seed轻量库 #

django-seed是一个轻量工具，能根据模型字段类型自动生成随机数据，同时支持关联字段的自动处理。

安装与配置 #

pip install django-seed

在settings.py的INSTALLED_APPS中添加：

INSTALLED_APPS = [
    ...
    'django_seed',
]

快速填充 #

# 为指定APP生成20条数据，自动处理外键和多对多关联
python manage.py seed myapp --number=20

# 只为指定模型生成数据
python manage.py seed myapp --model=Author --number=10

它会自动根据字段类型生成对应随机值，外键会关联已存在的对象（无数据时自动创建），多对多字段也会自动完成关联。

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