如何使用 SQLAlchemy 插入多对多关系数据？

Question

我想弄清楚的是一个关于如何将数据插入多对多关系的简单示例。我正在尝试使用户可以成为许多团队的一部分，同时一个团队也可以拥有许多用户。

我阅读了 SQLAlchemy 文档，但它们没有说明如何将数据实际插入到多对多关系中。当我查看类似的问题时，他们不会过度插入多个元素——这正是我所需要的。它们主要展示如何通过关联表创建或进行更改。

class Teams(Base):
    __tablename__ = 'teams'
    id = Column(Integer, primary_key=True)
    name = Column(String(64))
    members = relationship('User', secondary=teams_association, back_populates='teams')

class User(Base):
    __tablename__ = 'user'
    id = Column(Integer, primary_key=True)
    name = Column(String(64))
    teams = relationship('Teams', secondary=teams_association, back_populates='members')

teams_association = Table(
    'teams_association',
    Base.metadata,
    Column('captain_id', Integer, ForeignKey('user.id')),
    Column('teams_id', Integer, ForeignKey('teams.id'))
    )

插入多对多数据

session.add_all([
Teams(
    name='wildcats',
    members=??      # <-- Members [1,3,4] How would I do a multiple insert here?
),
Teams(
    name='horseman',
    members=??
),
Teams(
    name='jokers',
    members=??
) ])

session.commit()

session.add_all([
User(
    name='sam',
    teams=?     # [1,3] ex. A Member of Teams 1(wildcats) & 3(jokers)
    ),
User(
    name='ashley',
    teams=?
),
User(
    name='jackson',
    teams=?
),
User(
    name='ethan',,
    teams=?
),
User(
    name='miles',
    teams=?
) ])

Answer 1

以下是一些使用 SQLAlchemy ORM 在多对多关联表中插入记录的示例，这些示例可能对您有所帮助（逐步显示，希望能够更容易地了解每个关联表中需要发生的事情不同的场景）。

如果您尝试在关联表中创建将用户“sam”和“ashley”与“wildcats”团队相关联的记录，您可以执行以下操作（假设您已经为“ sam”和“ashley”以及“wildcats”的Teams记录）：

members = session.query(User).filter(User.name.in_(['sam', 'ashley'])).all()
wildcats = session.query(Teams).filter(Teams.name == 'wildcats').first()
wildcats.members = [member for member in members]
session.commit()

或者，如果您尝试与关联用户“sam”和“ashley”创建新的“wildcats”Teams 记录，那么您可以修改上述内容以首先创建Teams 记录。例如：

wildcats = Teams()
wildcats.name = 'wildcats'
members = session.query(User).filter(User.name.in_(['sam', 'ashley'])).all()
wildcats.members = [member for member in members]
session.add(wildcats)
session.commit()

或者，如果您尝试创建多个新的 User 记录并将它们与多个 Teams 记录关联，那么您可以执行以下操作：

sam = User()
sam.name = 'sam'
sam_teams = session.query(Teams).filter(Teams.name.in_(['wildcats', 'jokers'])).all()
sam.teams = [team for team in sam_teams]
ashley = User()
ashley.name = 'ashley'
ashley_teams = session.query(Teams).filter(Teams.name.in_(['wildcats', 'horseman'])).all()
ashley.teams = [team for team in ashley_teams]
session.add_all([sam, ashley])
session.commit()