Jef*_*man 4 python sqlalchemy flask-sqlalchemy factory-boy
我正在使用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)
Run Code Online (Sandbox Code Playgroud)
这是我的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)
Run Code Online (Sandbox Code Playgroud)
我可以毫无问题地调用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)
Run Code Online (Sandbox Code Playgroud)
请参阅如何backref在GearCategory名为category不是category_id?更新要使用的那一行可以category解决问题:
category = factory.SubFactory(GearCategoryFactory)
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
386 次 |
| 最近记录: |