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

本文链接：https://blog.youkuaiyun.com/gitblog_00244/article/details/148376236

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 的优点，为开发者提供了简洁高效的数据模型定义和数据库操作方式。本文将重点讲解如何使用 SQLModel 更新数据库中的记录。

数据模型定义

首先我们需要定义一个数据模型类 Hero，它继承自 SQLModel 并设置了 table=True 参数，表示这个类对应数据库中的一个表：

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 语句，便于调试。

创建数据库和表

create_db_and_tables() 函数会创建数据库和所有定义的表：

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

插入初始数据

create_heroes() 函数创建了多个英雄实例并插入数据库：

hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson")
hero_2 = Hero(name="Spider-Boy", secret_name="Pedro Parqueador")
# ...其他英雄实例

使用会话(Session)批量添加并提交：

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

更新记录详解

核心的更新操作在 update_heroes() 函数中实现，我们一步步分析：

查询记录：使用 select() 查询名为 "Spider-Boy" 的英雄
```
statement = select(Hero).where(Hero.name == "Spider-Boy")
```
执行查询：通过会话执行查询语句
```
results = session.exec(statement)
```
获取单个结果：使用 one() 方法确保只返回一条记录
```
hero = results.one()
```
修改属性：直接修改对象的属性
```
hero.age = 16
```
标记为待更新：将修改后的对象添加到会话
```
session.add(hero)
```
提交更改：将更改持久化到数据库
```
session.commit()
```
刷新对象：从数据库重新加载对象的最新状态
```
session.refresh(hero)
```