使用外键创建多个对象

Question

我需要创建十个示例用户(User),每个用户必须有五十个文档(Doc).如何使用factoryboy在tests.py中执行此操作？

#factories.py

from app_name.models import *
import factory
from datetime import datetime, timedelta, time
from django.contrib.auth.models import User


class UserFactory(factory.Factory):
    FACTORY_FOR = User

    username = factory.Sequence(lambda n: 'User ' + n)
    email = 'demo@mail.com'
    password = '1234567'

class DocFactory(factory.Factory):
    FACTORY_FOR = Doc

    user = factory.SubFactory(UserFactory)
    kategories = '1'
    doc_number = '12345678'
    date_join = factory.Sequence(lambda n:(datetime.now() + timedelta(days=n)).date(), int)

在我的tests.py中:

from django.test import TestCase
from django_dynamic_fixture import G
from factories import *

Answer 1

users = UserFactory.create_batch(10)
for user in users:
    doc = DocFactory.create(user=user)

Answer 2

你可以使用post_generation装饰器:

class UserFactory(factory.Factory):

    ...

    @factory.post_generation
    def create_docs(self, create, extracted, **kwargs):
        if not create:
            return
        for i in range(50):
            doc = DocFactory.create(user=self)