SQLModel 教程：使用 Python ORM 更新数据库记录-优快云博客

SQLModel 教程：使用 Python ORM 更新数据库记录

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

前言

SQLModel 是一个强大的 Python ORM（对象关系映射）工具，它结合了 SQLAlchemy 和 Pydantic 的优点，为开发者提供了类型安全和简洁的 API。本教程将重点介绍如何使用 SQLModel 更新数据库中的记录。

环境准备

首先我们需要定义一个数据模型，这里以超级英雄为例：

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)

这个模型定义了英雄的四个属性：

id：主键，自动生成
name：英雄名称，建立索引
secret_name：秘密身份
age：年龄，可选字段，也建立了索引

数据库连接设置

我们使用 SQLite 作为数据库后端：

sqlite_file_name = "database.db"
sqlite_url = f"sqlite:///{sqlite_file_name}"
engine = create_engine(sqlite_url, echo=True)

echo=True 参数会输出执行的 SQL 语句，方便调试。

创建数据库和表

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

这个方法会创建所有定义的表结构。

初始化数据

我们创建了7个英雄实例并存入数据库：

def create_heroes():
    hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson")
    # ...其他英雄实例
    hero_7 = Hero(name="Captain North America", secret_name="Esteban Rogelios", age=93)

    with Session(engine) as session:
        session.add(hero_1)
        # ...添加其他英雄
        session.commit()

更新记录的核心部分

重点来了，我们来看如何更新数据库中的记录：

def update_heroes():
    with Session(engine) as session:
        # 1. 查询要更新的记录
        statement = select(Hero).where(Hero.name == "Spider-Boy")
        results = session.exec(statement)
        hero = results.one()
        print("Hero:", hero)

        # 2. 修改对象属性
        hero.age = 16
        
        # 3. 将修改后的对象添加回会话
        session.add(hero)
        
        # 4. 提交事务
        session.commit()
        
        # 5. 刷新对象以获取最新状态
        session.refresh(hero)
        print("Updated hero:", hero)

更新记录的完整流程包括：