为什么 SQLAlchemy 不将 FactoryBoy SubFactory 生成的这个对象翻译成外键?

Why isn't SQLAlchemy translating this object generated by a FactoryBoy SubFactory into a foreign key?

我将 Flask 和 SQLAlchemy(通过 Flask-SQLAlchemy 扩展)与 Factory_Boy 一起使用。

我的 GearItem 模型有一个指向 GearCategory 的外键。 Factory_Boy 通过 SubFactory 函数处理此问题,该函数创建对象以用作原始工厂中的外键。

这是我的模型定义:

class GearCategory(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.Text, unique=True, nullable=False)
    gear_items = db.relationship('GearItem', backref='category',
            lazy='dynamic', order_by='GearItem.name')

class GearItem(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.Text, nullable=False, index=True)
    category_id = db.Column(db.Integer, db.ForeignKey('gear_category.id'), index=True)

这是我的 factory_boy Factory 定义:

class GearCategoryFactory(BaseFactory): # Based on factory.alchemy.SQLAlchemyModelFactory
    class Meta:
        model = gear_models.GearCategory
    name = factory.LazyAttribute(lambda n: faker.word())

class GearItemFactory(BaseFactory):
    class Meta:
        model = gear_models.GearItem
    name = factory.LazyAttribute(lambda n: faker.word())
    category_id = factory.SubFactory(GearCategoryFactory)

我可以毫无问题地调用 GearItemFactory(),它显然会生成一个 GearItem 和一个旨在用作外键的父 GearCategory

但是,当我调用 db.session.flush() 时,SQLAlchemy 不会将 SubFactory 创建的对象转换为可用作外键的整数。相反,它试图将对象本身传递给底层数据库驱动程序,然后数据库驱动程序抱怨不知道如何处理 GearCategory 类型的对象。

我得到的错误是 sqlalchemy.exc.ProgrammingError: (db_driver) can't adapt type 'GearCategory' [SQL: 'INSERT INTO gear_item (...

我做错了什么?

问题是 GearItemFactory 定义为外键数据库 ID 指定了一个 Python 对象引用。 sqlalchemy 将 Python 对象转换为数据库外键 ID 没有问题。但是,在我的工厂中,我指定了对象到数据库的列映射而不是对象到对象的映射,因此 SQLAlchemy(理所当然地)认为我想将 Python 对象直接传递给数据库。只需要将工厂外键更改为对象到对象的映射,sqlalchemy 将在后台处理实际的数据库 FK 列。

这是虚线:

category_id = factory.SubFactory(GearCategoryFactory)

看看 GearCategory 上的 backref 是如何命名为 category 而不是 category_id 的? 更新该行以使用 category 修复了问题:

category = factory.SubFactory(GearCategoryFactory)