5

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

我的GearItem模型有一个外键GearCategory。Factory_BoySubFactory通过创建在原始工厂中用作外键的对象的函数来处理这个问题。

这是我的模型定义：

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_boyFactory定义：

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 (...

我究竟做错了什么？

4

1 回答 1

7

问题在于该GearItemFactory定义为外键数据库 ID 指定了 Python 对象引用。sqlalchemy将 Python 对象转换为数据库外键 ID 没有问题。但是在我的工厂中，我指定了对象到数据库的列映射而不是对象到对象的映射，因此 SQLAlchemy（正确地）认为我想将 Python 对象直接传递给数据库。只需将工厂外键更改为对象到对象的映射，并sqlalchemy在后台处理实际的数据库 FK 列。

这是虚线：

category_id = factory.SubFactory(GearCategoryFactory)

看看backrefonGearCategory是怎么命名category的category_id？更新该行以使用category解决问题：

category = factory.SubFactory(GearCategoryFactory)
于 2015-11-02T18:16:48.630 回答