列表子工厂元素内的随机模型值

Question

我找不到我的问题的答案。问题是我想生成一个用户工厂模型，其中内部将是一个带有地址的子工厂列表。每个地址元素必须具有不同/随机值（我的意思是每个元素都有不可重复的唯一值）。

在我当前的实现中，所有元素都具有相同的值（也许需要播种？）

实际代码：

from pydantic import BaseModel
from factory import Factory, List, Subfactory


class Address(BaseModel):
    Name: str


class User(BaseModel):
    Addresses: list[Address]


class AddressFactory(Factory):
    Name = fake.name()

    class Meta:
        model = Address


class UserFactory(Factory):
    Addresses = List([SubFactory(AddressFactory) for _ in range(3)])

    class Meta:
        model = User

实际输出：

> UserFactory()
> User(Addresses=[Address(Name='Isa Merkx'), Address(Name='Isa Merkx'), Address(Name='Isa Merkx')])

所需输出：

> UserFactory()
> User(Addresses=[Address(Name='Isa Merkx'), Address(Name='John Smith'), Address(Name='Elon Musk')])

Answer 1

您应该使用 LazyAttribute 每次获取不同的值。

from typing import List
from pydantic import BaseModel
import factory
from faker import Faker

fake = Faker('en_GB') # <-- missing from the original example


class Address(BaseModel):
    Street: str
    HouseNumber: str
    City: str
    Postcode: str


class AddressFactory(factory.Factory):
    Street = factory.LazyAttribute(lambda _: fake.street_name()) # <-- Lazy load the attribute values
    HouseNumber = factory.LazyAttribute(lambda _: fake.building_number())
    City = factory.LazyAttribute(lambda _: fake.city())
    Postcode = factory.LazyAttribute(lambda _: fake.postcode())

    class Meta:
        model = Address


class User(BaseModel):
    Addresses: List[Address]


class UserFactory(factory.Factory):
    Addresses = factory.List([factory.SubFactory(AddressFactory) for _ in range(3)])

    class Meta:
        model = User


user = UserFactory()
user

输出：

User(Addresses=[Address(Street='Jade rapids', HouseNumber='3', City='Vanessaville', Postcode='B6H 2XA'), Address(Street='Wendy grove', HouseNumber='76', City='West Patricia', Postcode='WR5 0GL'), Address(Street='Smith ramp', HouseNumber='3', City='New Leslie', Postcode='L6 6JF')])

Answer 2

在您提供的代码中，您必须将 fake.name() 替换为 Faker("name")，其中 Faker 是从 < code>factory 在文件的开头。

事实上，fake.name() 仅在定义类时调用一次，而您希望每次实例化工厂时都调用一次。

您会找到问题的更详细答案 此处。