Отношения «один ко многим» в factory_boy

Я использую этот шаблон в своем проекте. Предполагая, что у вас уже есть AddressFactory.

https://factoryboy.readthedocs.io/en/latest/reference.html?highlight=post_generation#factory.post_generation

class PersonFactory(factory.alchemy.SQLAlchemyFactory):
    class Meta:
        model = Person

    @factory.post_generation
    def addresses(obj, create, extracted, **kwargs):
        if not create:
            return

        if extracted:
            assert isinstance(extracted, int)
            AddressFactory.create_batch(size=extracted, person_id=self.id, **kwargs)

Применение

PersonFactory(addresses=4)

Это создаст Person с 4 Addresses

Также это может принимать kwargs

PersonFactory(addresses=2, addresses__city='London')

Это создаст Person с 2 Addresses, у которых поле city установлено на 'London'

Вот сообщение в блоге, которое может помочь https://simpleit.rocks/python/django/setting-up-a-factory-for-one-to-many-relationships-in-factoryboy/

У меня был именно этот вопрос, и я был разочарован отсутствием хороших ответов здесь. Оказывается можно! Оставьте это здесь для тех, у кого есть тот же вопрос.

Во-первых, ваша модель должна определить отношение к модели, противоположной внешнему ключу, поэтому она должна выглядеть так:

class Person(...):
    id = Column(Integer, primary_key=True)
    name = Column(Text)
    addresses = relationship("Person", backref="person")
    [...]


class Address(...):
    id = Column(Integer, primary_key=True)
    city = Column(Text)
    [...]
    person_id = Column(Integer, ForeignKey('person.id'))

Then, on your PersonFactory, you can add a post_generation hook like this:

class PersonFactory(BaseFactory):
    [...attributes...]


    @factory.post_generation
    def addresses(self, create, extracted, **kwargs):
        return AddressFactory.create_batch(4)

В настоящее время нет способа реализовать "многие к одному RelatedFactory" таким образом, чтобы он "встраивался в вашу фабрику"...

Тем не менее, такое поведение можно реализовать с небольшим количеством хакерских действий при создании экземпляра PersonFactory.

Следующий рецепт даст вам то, что вы ищете:

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)
    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"

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))

@Kristen указала в правильном направлении, но AdderssFactory не имела отношения к Person. В Django мы можем использовать post_generation такой декоратор.

class PersonFactory(BaseFactory):
    @factory.post_generation
    def addresses(self, create, extracted, **kwargs):
        self.addresses_set.add(AddressFactory(person=self))

Вы можете использовать решение, описанное здесь: http://factoryboy.readthedocs.org/en/latest/recipes.html#reverse-dependencies-reverse-foreignkey

По сути, просто объявите несколько RelatedFactory на вашем PersonFactory:

class PersonFactory(factory.alchemy.SQLAlchemyFactory):
    class Meta:
        model = Person

    address_1 = factory.RelatedFactory(AddressFactory, 'person')
    address_2 = factory.RelatedFactory(AddressFactory, 'person')