PostgreSQL
 sql >> Base de données >  >> RDS >> PostgreSQL

SQLAlchemy plusieurs clés étrangères dans une classe mappée à la même clé primaire

J'ai essayé de supprimer les guillemets des clés étrangères et d'en faire une liste. À partir de la documentation officielle sur Relationship Configuration: Handling Multiple Join Paths

Modifié dans la version 0.8 :relationship() peut résoudre l'ambiguïté entre les cibles de clés étrangères sur la base des foreign_keys argument seul ; le primaryjoin l'argument n'est plus nécessaire dans cette situation.

Le code autonome ci-dessous fonctionne avec 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()