如何使用 SQLAlchemy 和 FastAPI 更新多对多关系？

Question

models.py

class Role(Base):
    __tablename__='role'
    role_id = Column(Integer,primary_key=True)
    role_name = Column(String(20),nullable=False)

class User(Base):
    __tablename__='user'
    user_id = Column(Integer,primary_key=True)
    user_name = Column(String(20),nullable=False)

class UserRoles(Base):
    __tablename__="user_roles"
    user_id = Column(Integer,ForeignKey("User.user_id"),primary_key=True)
    role_id = Column(Integer,ForeignKey("Role.role_id"),primary_key=True)
    status = Column(Integer,nullable=True)
    user = relationship("User")
    role = relationship("role")

表“user_roles”中的数据

user_id	role_id	status
1	1	1
1	2	1
2	1	1

当对 API 的请求对象为 {"user_id":1,"roles":[1,3]} 时，我如何编写 API 来更新 user_roles 表。如何在 FastAPI 中使用 sqlalchemy 来实现这一点？

Answer 1

首先你应该选择UserRoles进行更新：

results = await session.execute(
    select(UserRoles).where(UserRoles.id == user_id)
)  # where user_id = 1
user_roles = results.scalars().all()[0]

可能有更简单的方法来获得一个角色，你应该看看docs

然后选择所有要附加到 ManyToMany 字段的Roles：

roles = await session.execute(select(Role).where(Role.role_id.in_(roles)))  # Where roles = [1, 3]

然后像列表一样附加它们：

for role in roles.scalars().all():
    user_roles.role.append(role)

并提交会话：

await session.commit()