Django Factory Boy新手：如何为数据库模型字段指定多测试数据模式？

嘿，刚上手Factory Boy的话，给模型字段设置多种测试数据模式其实挺灵活的，我来给你拆解几个实用的方法，正好适配你说的用户名（比如abc_123、abc123这类变体）场景！

Factory Boy本身集成了Faker库，你可以利用Faker的生成能力，再结合随机逻辑切换不同格式：

import factory
from faker import Faker
from myapp.models import User

fake = Faker()

class UserFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = User

    # 50%概率生成带下划线的用户名，50%概率生成去掉下划线的版本
    username = factory.LazyAttribute(
        lambda _: fake.user_name() 
        if fake.boolean(chance_of_getting_true=50) 
        else fake.user_name().replace("_", "")
    )

这样每次生成用户时，用户名会随机在带/不带下划线的变体里切换，完美匹配你要的abc_123和abc123格式。

如果需要更精准的格式（比如固定前缀+数字，带或不带下划线），可以写个自定义生成函数：

import factory
from faker import Faker
from myapp.models import User

fake = Faker()

def generate_username_variant():
    # 取一个短单词作为基础前缀
    base_prefix = fake.word()[:6].lower()
    # 生成3位随机数字
    num_suffix = fake.random_number(digits=3)
    # 随机决定是否加下划线
    if fake.boolean():
        return f"{base_prefix}_{num_suffix}"
    else:
        return f"{base_prefix}{num_suffix}"

class UserFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = User

    username = factory.LazyAttribute(lambda _: generate_username_variant())

这种方式完全由你控制变体的规则，想加更多格式（比如带点号的abc.123）也只需要在函数里加分支就行。

如果需要带编号的测试数据同时保留变体（比如user1_abc、user2abc交替），可以用factory.Sequence：

import factory
from myapp.models import User

class UserFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = User

    # 偶数编号用带下划线的格式，奇数用无下划线格式
    username = factory.Sequence(lambda n: f"user{n}_abc" if n % 2 == 0 else f"user{n}abc")

适合需要批量生成有序且有格式差异的测试数据场景。

如果需要精准控制不同变体的数量（比如7个带下划线，3个不带），可以直接在生成时指定：

# 生成7个带下划线的用户名
underscore_users = UserFactory.create_batch(7, username=factory.LazyAttribute(lambda _: fake.user_name()))
# 生成3个不带下划线的用户名
no_underscore_users = UserFactory.create_batch(3, username=factory.LazyAttribute(lambda _: fake.user_name().replace("_", "")))

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