SQLAlchemy-插入的自动查找外键关系

Question

我试图让一个SQLAlchemy ORM类自动：

查找字段的外键ID

要么

对于外键表中尚未包含该字段的条目，请将行添加到外键表中-并在原始表中使用自动生成的ID。

为了显示：

类定义

class EquityDB_Base(object):

    @declared_attr
    def __tablename__(cls):
        return cls.__name__.lower()

    __table_args__ = {'mysql_engine': 'InnoDB'}
    __mapper_args__= {'always_refresh': True}

    id =  Column(Integer, primary_key=True)

def fk(tablename, nullable=False):
    return Column("%s_id" % tablename, Integer,
               ForeignKey("%s.id" % tablename),
               nullable=nullable)

class Sector(EquityDB_Base, Base):
    name = Column(String(40))

class Industry(EquityDB_Base, Base):
    name = Column(String(40))
    sector_id = fk('sector')
    sector = relationship('Sector', backref='industries')

class Equity(EquityDB_Base, Base):
    symbol = Column(String(10), primary_key=True)
    name = Column(String(40))
    industry_id = fk('industry')
    industry = relationship('Industry', backref='industries')

使用班级设置行业和部门

for i in industry_record[]:
    industry = Industry(id=i.id, 
                    name=i.name,
                    sector=Sector(name=i.sector_name))
    session.merge(industry)

结果不幸的是，当我运行此命令时-数据库为'sector_name'的每一次重复使用将单独的行添加到扇区表中-例如，如果10个行业使用'Technology'作为其扇区名，则每一个获得10个唯一的行业ID 10个行业。

我想要的-每次显示数据库中已经存在的扇区名称时，它就会自动解析为相应的ector_id

我显然只是在学习SQLAlchemy，但似乎无法弄清楚如何启用此行为。

任何帮助，将不胜感激！

Answer 1

请参阅表中对create_or_get条目的类似问题的答案。
应用相同的逻辑，您将获得以下内容：

def create_or_get_sector(sector_name):
    obj = session.query(Sector).filter(Sector.name == sector_name).first()
    if not obj:
        obj = Sector(name = sector_name)
        session.add(obj)
    return obj

并如下所示使用它：

for i in industry_record[:]:
    industry = Industry(id=i.id, 
                    name=i.name,
                    sector=create_or_get_sector(sector_name=i.sector_name))
    session.merge(industry)

您应该注意的一件事是create_or_get_sector使用了哪个session实例。