数据库
首页 > 数据库> > python – 为什么SQLAlchemy不将FactoryBoy SubFactory生成的这个对象转换为外键?

python – 为什么SQLAlchemy不将FactoryBoy SubFactory生成的这个对象转换为外键?

作者:互联网

我正在使用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工厂定义:

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)无法调整类型’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 = factory.SubFactory(GearCategoryFactory)

标签:python,sqlalchemy,flask-sqlalchemy,factory-boy
来源: https://codeday.me/bug/20190609/1203680.html