SQLAlchemy映射类中多个外键指向同一主键

88 投票
2 回答
103099 浏览
提问于 2025-04-17 21:57

我正在尝试设置一个PostgreSQL表,这个表有两个外键,它们都指向另一个表中的同一个主键。

当我运行这个脚本时,出现了一个错误:

sqlalchemy.exc.AmbiguousForeignKeysError: 无法确定父表和子表之间的连接条件,因为有多个外键路径连接这些表。请指定'foreign_keys'参数,提供一个包含那些应该被视为指向父表的外键引用的列的列表。

这个错误在SQLAlchemy文档中有详细说明,但当我尝试他们提供的解决方案时,错误并没有消失。我可能做错了什么呢?

#The business case here is that a company can be a stakeholder in another company.
class Company(Base):
    __tablename__ = 'company'
    id = Column(Integer, primary_key=True)
    name = Column(String(50), nullable=False)

class Stakeholder(Base):
    __tablename__ = 'stakeholder'
    id = Column(Integer, primary_key=True)
    company_id = Column(Integer, ForeignKey('company.id'), nullable=False)
    stakeholder_id = Column(Integer, ForeignKey('company.id'), nullable=False)
    company = relationship("Company", foreign_keys='company_id')
    stakeholder = relationship("Company", foreign_keys='stakeholder_id')

我在这里看到过类似的问题,但一些答案建议使用primaryjoin,而文档中却说在这种情况下不需要primaryjoin

2 个回答

11

最新的文档:

文档中提到的 foreign_keys= 这个写法会导致一个名称错误,我不太明白在类还没创建的情况下,应该怎么用。经过一些尝试,我终于成功了,代码如下:

company_id = Column(Integer, ForeignKey('company.id'), nullable=False)
company = relationship("Company", foreign_keys='Stakeholder.company_id')

stakeholder_id = Column(Integer, ForeignKey('company.id'), nullable=False)
stakeholder = relationship("Company",
                            foreign_keys='Stakeholder.stakeholder_id')

换句话说:

… foreign_keys='CurrentClass.thing_id')
109

我试着把外键的引号去掉,并把它们做成一个列表。这个内容来自官方文档,讲的是如何处理多个连接路径的关系配置。

在0.8版本中有了变化:现在使用relationship()可以仅通过foreign_keys这个参数来解决外键目标之间的模糊性;在这种情况下,不再需要primaryjoin参数。


下面的自包含代码适用于sqlalchemy>=0.9

from sqlalchemy import create_engine, Column, Integer, String, ForeignKey
from sqlalchemy.orm import relationship, scoped_session, sessionmaker
from sqlalchemy.ext.declarative import declarative_base

engine = create_engine(u'sqlite:///:memory:', echo=True)
session = scoped_session(sessionmaker(bind=engine))
Base = declarative_base()

#The business case here is that a company can be a stakeholder in another company.
class Company(Base):
    __tablename__ = 'company'
    id = Column(Integer, primary_key=True)
    name = Column(String(50), nullable=False)

class Stakeholder(Base):
    __tablename__ = 'stakeholder'
    id = Column(Integer, primary_key=True)
    company_id = Column(Integer, ForeignKey('company.id'), nullable=False)
    stakeholder_id = Column(Integer, ForeignKey('company.id'), nullable=False)
    company = relationship("Company", foreign_keys=[company_id])
    stakeholder = relationship("Company", foreign_keys=[stakeholder_id])

Base.metadata.create_all(engine)

# simple query test
q1 = session.query(Company).all()
q2 = session.query(Stakeholder).all()

撰写回答