factory_boy一對多關系

Question

我有一對多的關系，並想為其創建一個工廠，但是不知道如何使用factory_boy API來實現。 一對一的關系很簡單，只需使用SubFactory ，但是對於一對多的關系該如何做呢？

例：

class Order(object):
    def __init__(self, id, itens):
        self.id = id
        self.itens = itens

class Item(object):
    def __init__(self, id, itens):
        self.id = id


class OrderFactory(SQLAlchemyModelFactory):
    class Meta:
        model = Order

    id = lazy_attribute(lambda o: random.randint(1, 1000))
    itens = ????

Answer 1

您應該使用后期生成掛鈎：

@factory.post_generation
def items(self, create, extracted, **kwargs):
    if not create:
        return

    if extracted:
        for item in extracted:
            self.items.add(item)