Question

我正在使用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 (...

我做错了什么？

Answer 1

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

这是一条断线：

category_id = factory.SubFactory(GearCategoryFactory)

查看backref上的GearCategory如何命名为category而不是category_id？更新该行以使用category可以解决问题：

category = factory.SubFactory(GearCategoryFactory)

为什么SQLAlchemy不将FactoryBoy SubFactory生成的此对象转换为外键？

1 个答案: