我使用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())