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

Joh*_*han 3 python many-to-many sqlalchemy insert

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

我阅读了 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'))
    )
Run Code Online (Sandbox Code Playgroud)

插入多对多数据

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=?
) ])
Run Code Online (Sandbox Code Playgroud)

ben*_*nvc 6

以下是一些使用 SQLAlchemy ORM 在多对多关联表中插入记录的示例,它们可能对您有所帮助(逐步显示,希望可以更轻松地了解每个不同的关联表中需要发生的事情场景)。

如果您尝试在关联表中创建将用户“sam”和“ashley”与“wildcats”团队关联的记录,您可以执行以下操作(假设您已经User为“sam”和“ashley”创建了记录)和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()
Run Code Online (Sandbox Code Playgroud)

或者,如果您尝试创建一个新的“wildcats”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()
Run Code Online (Sandbox Code Playgroud)

或者,如果您尝试创建多个新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()
Run Code Online (Sandbox Code Playgroud)