[英]SQLAlchemy Inserting Data in a Many-to-Many Relationship with Association Table
[英]SQLAlchemy relationship on many-to-many association table
我正在嘗試與另一個多對多關系建立關系,代碼如下所示:
from sqlalchemy import Column, Integer, ForeignKey, Table, ForeignKeyConstraint, create_engine
from sqlalchemy.orm import relationship, backref, scoped_session, sessionmaker
from sqlalchemy.ext.declarative import declarative_base
Base = declarative_base()
supervision_association_table = Table('supervision', Base.metadata,
Column('supervisor_id', Integer, ForeignKey('supervisor.id'), primary_key=True),
Column('client_id', Integer, ForeignKey('client.id'), primary_key=True)
)
class User(Base):
__tablename__ = 'user'
id = Column(Integer, primary_key=True)
class Supervisor(User):
__tablename__ = 'supervisor'
__mapper_args__ = {'polymorphic_identity': 'supervisor'}
id = Column(Integer, ForeignKey('user.id'), primary_key = True)
schedules = relationship("Schedule", backref='supervisor')
class Client(User):
__tablename__ = 'client'
__mapper_args__ = {'polymorphic_identity': 'client'}
id = Column(Integer, ForeignKey('user.id'), primary_key = True)
supervisor = relationship("Supervisor", secondary=supervision_association_table,
backref='clients')
schedules = relationship("Schedule", backref="client")
class Schedule(Base):
__tablename__ = 'schedule'
__table_args__ = (
ForeignKeyConstraint(['client_id', 'supervisor_id'], ['supervision.client_id', 'supervision.supervisor_id']),
)
id = Column(Integer, primary_key=True)
client_id = Column(Integer, nullable=False)
supervisor_id = Column(Integer, nullable=False)
engine = create_engine('sqlite:///temp.db')
db_session = scoped_session(sessionmaker(bind=engine))
Base.metadata.create_all(bind=engine)
我想要做的是將時間表與特定的客戶 - 主管關系聯系起來,盡管我還沒有找到如何做到這一點。 通過SQLAlchemy文檔,我發現了一些提示,導致Schedule-Table上的ForeignKeyConstraint。
如何指定關聯以使此關聯有效?
您需要映射supervision_association_table,以便您可以創建與之關系。
我可能會在這里討論一些事情,但似乎因為你在這里有多對多你真的不能有Client.schedules - 如果我說Client.schedules.append(some_schedule),那么“監督”中的哪一行是它指向? 因此,下面的示例為那些加入每個SupervisorAssociation的Schedule集合的訪問者提供了只讀“匯總”訪問器。 association_proxy擴展用於在方便時隱藏SupervisionAssociation對象的詳細信息。
from sqlalchemy import Column, Integer, ForeignKey, Table, ForeignKeyConstraint, create_engine
from sqlalchemy.orm import relationship, backref, scoped_session, sessionmaker
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.ext.associationproxy import association_proxy
from itertools import chain
Base = declarative_base()
class SupervisionAssociation(Base):
__tablename__ = 'supervision'
supervisor_id = Column(Integer, ForeignKey('supervisor.id'), primary_key=True)
client_id = Column(Integer, ForeignKey('client.id'), primary_key=True)
supervisor = relationship("Supervisor", backref="client_associations")
client = relationship("Client", backref="supervisor_associations")
schedules = relationship("Schedule")
class User(Base):
__tablename__ = 'user'
id = Column(Integer, primary_key=True)
class Supervisor(User):
__tablename__ = 'supervisor'
__mapper_args__ = {'polymorphic_identity': 'supervisor'}
id = Column(Integer, ForeignKey('user.id'), primary_key = True)
clients = association_proxy("client_associations", "client",
creator=lambda c: SupervisionAssociation(client=c))
@property
def schedules(self):
return list(chain(*[c.schedules for c in self.client_associations]))
class Client(User):
__tablename__ = 'client'
__mapper_args__ = {'polymorphic_identity': 'client'}
id = Column(Integer, ForeignKey('user.id'), primary_key = True)
supervisors = association_proxy("supervisor_associations", "supervisor",
creator=lambda s: SupervisionAssociation(supervisor=s))
@property
def schedules(self):
return list(chain(*[s.schedules for s in self.supervisor_associations]))
class Schedule(Base):
__tablename__ = 'schedule'
__table_args__ = (
ForeignKeyConstraint(['client_id', 'supervisor_id'],
['supervision.client_id', 'supervision.supervisor_id']),
)
id = Column(Integer, primary_key=True)
client_id = Column(Integer, nullable=False)
supervisor_id = Column(Integer, nullable=False)
client = association_proxy("supervisor_association", "client")
engine = create_engine('sqlite:///temp.db', echo=True)
db_session = scoped_session(sessionmaker(bind=engine))
Base.metadata.create_all(bind=engine)
c1, c2 = Client(), Client()
sp1, sp2 = Supervisor(), Supervisor()
sch1, sch2, sch3 = Schedule(), Schedule(), Schedule()
sp1.clients = [c1]
c2.supervisors = [sp2]
c2.supervisor_associations[0].schedules = [sch1, sch2]
c1.supervisor_associations[0].schedules = [sch3]
db_session.add_all([c1, c2, sp1, sp2, ])
db_session.commit()
print c1.schedules
print sp2.schedules
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.