SQLModel 多对多关系实战教程-优快云博客

SQLModel 多对多关系实战教程

sqlmodel SQL databases in Python, designed for simplicity, compatibility, and robustness. 项目地址: https://gitcode.com/gh_mirrors/sq/sqlmodel

理解多对多关系

在关系型数据库中，多对多关系是一种常见的数据关联方式。它描述的是一个实体可以与多个另一个实体相关联，反之亦然。例如，在超级英雄和团队的关系中：

一个英雄可以属于多个团队
一个团队可以包含多个英雄

SQLModel 提供了简洁而强大的方式来处理这种复杂关系。

模型定义解析

关联表 (HeroTeamLink)

class HeroTeamLink(SQLModel, table=True):
    team_id: Optional[int] = Field(
        default=None, foreign_key="team.id", primary_key=True
    )
    hero_id: Optional[int] = Field(
        default=None, foreign_key="hero.id", primary_key=True
    )

这个关联表是处理多对多关系的核心。它包含两个外键字段：

team_id 指向 Team 表
hero_id 指向 Hero 表

这两个字段共同组成复合主键，确保每个英雄-团队组合的唯一性。

团队模型 (Team)

class Team(SQLModel, table=True):
    id: Optional[int] = Field(default=None, primary_key=True)
    name: str = Field(index=True)
    headquarters: str

    heroes: List["Hero"] = Relationship(back_populates="teams", link_model=HeroTeamLink)

关键点在于 heroes 字段的 Relationship 配置：

back_populates="teams" 建立了双向关系
link_model=HeroTeamLink 指定了关联表

英雄模型 (Hero)

class Hero(SQLModel, table=True):
    id: Optional[int] = Field(default=None, primary_key=True)
    name: str = Field(index=True)
    secret_name: str
    age: Optional[int] = Field(default=None, index=True)

    teams: List[Team] = Relationship(back_populates="heroes", link_model=HeroTeamLink)

同样，teams 字段通过 Relationship 建立了反向关联。

数据库操作实战

创建数据库和表

def create_db_and_tables():
    SQLModel.metadata.create_all(engine)

这个函数会基于我们定义的模型自动创建所有表结构，包括关联表。

创建英雄和团队

def create_heroes():
    with Session(engine) as session:
        # 创建团队
        team_preventers = Team(name="Preventers", headquarters="Sharp Tower")
        team_z_force = Team(name="Z-Force", headquarters="Sister Margaret's Bar")
        
        # 创建英雄并关联团队
        hero_deadpond = Hero(
            name="Deadpond",
            secret_name="Dive Wilson",
            teams=[team_z_force, team_preventers],
        )
        # ...其他英雄创建代码
        session.commit()

关键点：

可以直接在创建英雄时通过 teams 参数指定所属团队
SQLModel 会自动处理关联表的记录插入

更新英雄团队关系

def update_heroes():
    with Session(engine) as session:
        # 查询英雄和团队
        hero_spider_boy = session.exec(
            select(Hero).where(Hero.name == "Spider-Boy")
        ).one()
        team_z_force = session.exec(select(Team).where(Team.name == "Z-Force")).one()
        
        # 添加关系
        team_z_force.heroes.append(hero_spider_boy)
        session.commit()
        
        # 移除关系
        hero_spider_boy.teams.remove(team_z_force)
        session.commit()

这个函数展示了如何：