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

Question

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

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

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”团队相关联的记录，您可以执行类似以下的操作(假设您已经为“创建了 User 记录” 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()

或者，如果您正在尝试创建一个新的“野猫”Teams记录关联用户“sam”和“ashley”，然后您可以修改上面的内容以首先创建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()