加入收藏 | 设为首页 | 会员中心 | 我要投稿 李大同 (https://www.lidatong.com.cn/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 编程开发 > Python > 正文

python – factory-boy subfactory传递对象而不是key

发布时间:2020-12-20 13:15:06 所属栏目:Python 来源:网络整理
导读:我正在使用SQLAlchemy的工厂男孩. 我正在尝试创建Fact对象,我希望工厂男孩生成PatientDim对象,这些对象是Fact的外键.但是Subfactory没有传递密钥,而是将整个对象传递给foreignkey字段. 我怎样才能通过Subfactory传递PatientDim键? Factories.py class FactF
我正在使用SQLAlchemy的工厂男孩.

我正在尝试创建Fact对象,我希望工厂男孩生成PatientDim对象,这些对象是Fact的外键.但是Subfactory没有传递密钥,而是将整个对象传递给foreignkey字段.

我怎样才能通过Subfactory传递PatientDim键?

Factories.py

class FactFactory(SQLAlchemyModelFactory):
    class Meta:
        model = models.Fact
        sqlalchemy_session = common.Session

    patient_id = factory.SubFactory(PatientDimFactory)

class PatientDimFactory(SQLAlchemyModelFactory):
    class Meta:
        model = models.PatientDim
        sqlalchemy_session = common.Session

Models.py

class Fact(TimeStampedModel):
    class Meta:
        db_table = 'fact'

    id = Column(Integer,primary_key=True)
    patient_id = Column('patient_id',Integer,ForeignKey(PatientDim.__table__.c.id),nullable=False)
    patient = relationship(PatientDim,foreign_keys='Fact.patient_id' )

Base = declarative_base()
class PatientDim(TimestampMixin,Base):
    __tablename__ = 'patient_dim'

    id = Column('id',primary_key=True)

解决方法

我今天的解决方法:

从Factory生成dict数据

data = factory.build(dict,FACTORY_CLASS=FactFactory)

创建ForeignKey / subFactory对象

patient_dim = PatientDimFactory()

更改已创建数据中的foreignKey对象

data['patient_id '] = patient_dim.pk

在create Form中使用crated数据

form = MyCrateForm(data=data)

(编辑:李大同)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    推荐文章
      热点阅读