Создание объектов обратной связи с Model Mommy - PullRequest
Новая
       11

Создание объектов обратной связи с Model Mommy

0 голосов
/ 04 сентября 2018

Я использую Model Mommy для создания тестовых данных, и пока он работает нормально.

Теперь у меня есть модель Django с именем Invoice и связанная модель с именем 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())
...