Django FactoryBoy：填充含选项的模型字段时抛出错误

你遇到的这个报错核心原因很明确：choice并不是factory_boy工厂字段定义里支持的关键字参数，你肯定是把Django模型字段的参数和工厂的参数搞混了，才触发了TypeError: 'choice' is an invalid keyword argument for this function。

下面我给你梳理几种正确的实现方式，帮你从固定选项列表里随机取值填充字段：

先明确场景前提 #

假设你的Django模型是这样定义带选项的字段：

from django.db import models

class Task(models.Model):
    PRIORITY_CHOICES = [
        ('low', 'Low'),
        ('medium', 'Medium'),
        ('high', 'High'),
    ]
    priority = models.CharField(max_length=10, choices=PRIORITY_CHOICES)

正确的工厂实现方式 #

方式1：用Faker的random_element生成随机值 #

这是最常用的方案，借助factory_boy集成的Faker工具，直接从选项列表里随机选取值：

import factory
from .models import Task

class TaskFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Task

    # 注意要取选项元组里的第一个值（也就是存在数据库中的实际存储值）
    priority = factory.Faker(
        'random_element',
        elements=[choice[0] for choice in Task.PRIORITY_CHOICES]
    )

方式2：结合Python标准库random.choice和LazyFunction #

如果你不想依赖Faker，直接用Python自带的随机方法也可以，配合LazyFunction延迟执行随机选择逻辑：

import random
import factory
from .models import Task

class TaskFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Task

    priority = factory.LazyFunction(
        lambda: random.choice([choice[0] for choice in Task.PRIORITY_CHOICES])
    )

方式3：用Iterator循环遍历选项（非随机场景） #

如果你不需要随机取值，而是希望每次创建对象时按顺序循环使用选项，就用Iterator：

import factory
from .models import Task

class TaskFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Task

    priority = factory.Iterator([choice[0] for choice in Task.PRIORITY_CHOICES])

为什么你之前会报错？ #

你大概率写了类似下面的错误代码：

# 错误示例！不要这么写！
priority = factory.CharField(choice=Task.PRIORITY_CHOICES)

choice是Django模型字段的专属参数，并不是factory_boy的factory.CharField（或其他工厂字段类）支持的参数，自然会触发类型错误。

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