在制作 DjangoModelFactory 时是否可以获取 Factory boy Faker 对象的值以用于另一个领域

Question

我正在尝试使用 factory boy 为事件创建模型实例，但我需要查看是否可以在另一个字段中访问 Faker 声明的结果。在下面的代码中，我想访问 end_time 字段中的 start_time 字段，这样我就可以创建一个距离 start_time 1 小时的日期时间。它抱怨 start_time 是一个 faker 对象，而不是 datetime 对象。如何让 Faker class 呈现其信息以便我可以在 end_time 字段中使用它？

class EventFactory(DjangoModelFactory):
    class Meta:
        model = Event

    client = factory.SubFactory(
        ClientFactory,
    )
    customer = factory.SubFactory(
        CustomerFactory,
    )
    resource = factory.SubFactory(
        ResourceFactory,
    )
    title = Faker("sentence", nb_words=5)
    start_time = Faker(
        "date_time_between_dates",
        datetime_start=datetime.datetime.now(),
        datetime_end=datetime.datetime.now() + datetime.timedelta(days=30),
    )
    end_time = start_time + datetime.timedelta(hours=1)
    all_day = False
    background_color = None
    border_color = None
    text_color = None
    duration = datetime.timedelta(hours=1)

Answer 1

LazyAttribute 为您提供了一种访问正在创建的对象中的其他属性的方法。尝试像这样定义你的 end_time：

    end_time = factory.LazyAttribute(
        lambda o: o.start_time + datetime.timedelta(hours=1)
    )

在制作 DjangoModelFactory 时是否可以获取 Factory boy Faker 对象的值以用于另一个领域

Is it possible to get the value of a Factory boy Faker object for use in another field while making a DjangoModelFactory

testing

django

unit-testing

factory-boy