创建具有外键的多个对象

2024-05-23 14:56:46 发布

您现在位置:Python中文网/ 问答频道 /正文

我需要创建10个示例用户(User),每个用户必须有50个文档(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)

在my tests.py中:

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

Tags: django用户frompyimportdatetimedocmodels