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

问题描述

我正在尝试使用 factory boy 为事件创建模型实例，但我需要查看是否可以在另一个字段中访问 Faker 声明的结果。在下面的代码中，我想访问 end_time 字段中的 start_time 字段，以便我可以创建一个距 start_time 1 小时的日期时间。它抱怨 start_time 是一个伪造对象，而不是日期时间对象。如何让 Faker 类呈现其信息，以便我可以在 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)

解决方法

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

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