Creating multiple objects with foreign key

Question

I need to create ten sample users (User) and each of them must have fifty documents (Doc). How to do this in tests.py using factoryboy?

#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 = '[email protected]'
    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)

in my tests.py:

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

Bernhard Vallant · Accepted Answer · 2013-01-29 22:48:19Z

17

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

answered Jan 29, 2013 at 22:48

Bernhard Vallant

50.9k21 gold badges126 silver badges149 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

guettli · Accepted Answer · 2014-08-12 14:31:47Z

8

You can use a post_generation decorator:

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)

answered Aug 12, 2014 at 14:31

guettli

27.7k109 gold badges423 silver badges779 bronze badges

Comments

User_1803 · Accepted Answer · 2021-09-23 09:54:46Z

1

You can simply do batch_create so that for each DocFactory object new UserFactory object will create

`DocFactory.create_batch(10)`

answered Sep 23, 2021 at 9:54

User_1803

462 bronze badges

Comments

The Aelfinn · Accepted Answer · 2017-05-09 16:25:41Z

For those of you working with SQLAlchemy, this can be done with the following recipe (notice that I'm using the Person/Address models instead of the User/Docs model example above).

from sqlalchemy import create_engine, Integer, Text, ForeignKey, Column
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import relationship, scoped_session, sessionmaker

import factory
from factory.alchemy import SQLAlchemyModelFactory as sqla_factory
import random

engine = create_engine("sqlite:////tmp/factory_boy.sql")
session = scoped_session(sessionmaker(bind=engine))
Base = declarative_base()

class Person(Base):
    id = Column(Integer, primary_key=True)
    name = Column(Text)
    addresses = relationship("Address", backref="person")

class Address(Base):
    id = Column(Integer, primary_key=True)
    city = Column(Text)
    street = Column(Text)
    street_number = Column(Integer)
    person_id = Column(Integer, ForeignKey('person.id'))

class AddressFactory(sqla_factory):
    class Meta:
        model = Address
        sqlalchemy_session = session
    street_number = random.randint(0, 10000)
    street = "Commonwealth Ave"
    city = "Boston"

class PersonFactory(sqla_factory):
    class Meta:
        model = Person
        sqlalchemy_session = session
    name = "John Doe"

Base.metadata.create_all(engine)
for i in range(100):
    person = PersonFactory(addresses=AddressFactory.create_batch(3))

This creates 3 workouts for each person created, where each workout references the person via the person_id FK.

Collectives™ on Stack Overflow

Creating multiple objects with foreign key

4 Answers 4

Comments

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

Comments

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related