使用Model Mommy

时间:2018-09-04 08:15:03

标签: python django model-mommy

我使用Model Mommy模型创建测试数据,到目前为止效果很好。

现在我有一个名为Invoice的Django模型和一个名为InvoiceItem的相关模型。

class Invoice(models.Model):
    created_by = models.ForeignKey(users.User)

class InvoiceItem(models.Model):
    invoice = models.ForeignKey(Invoice, related_name='items')

我为这些模型创建了两个简单的妈妈食谱:

InvoiceRecipe = Recipe(
    Invoice,
    created_by=foreign_key(UserRecipe),
)

InvoiceItemRecipe = Recipe(
    InvoiceItem,
    invoice=foreign_key(InvoiceRecipe),
)

现在,我希望InvoiceRecipe.make()自动向创建的发票中添加不同数量的InvoiceItem对象。

目前,我正在这样做,但是我希望它能够自动发生:

invoice = InvoiceRecipe.make(created_by=contractor)
invoice.items.add(InvoiceItemRecipe.make())
invoice.items.add(InvoiceItemRecipe.make())

0 个答案:

没有答案