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 工具，提供了简洁直观的 API 来处理数据库操作。本文将重点讲解如何使用 SQLModel 实现数据库记录的删除功能。

环境准备

首先需要确保已安装 SQLModel 和相关依赖：

pip install sqlmodel

数据模型定义

我们定义一个 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)

这个模型包含：

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

删除操作详解

删除操作是数据库 CRUD 中的重要一环，SQLModel 提供了简单直观的方式来实现：

def delete_heroes():
    with Session(engine) as session:
        # 1. 查询要删除的记录
        statement = select(Hero).where(Hero.name == "Spider-Youngster")
        results = session.exec(statement)
        hero = results.one()
        
        # 2. 打印待删除记录
        print("Hero: ", hero)
        
        # 3. 执行删除操作
        session.delete(hero)
        
        # 4. 提交事务
        session.commit()
        
        # 5. 打印删除后的记录
        print("Deleted hero:", hero)
        
        # 6. 验证删除结果
        statement = select(Hero).where(Hero.name == "Spider-Youngster")
        results = session.exec(statement)
        hero = results.first()
        
        if hero is None:
            print("There's no hero named Spider-Youngster")

关键步骤解析

1. 查询记录：使用 select() 和 where() 组合查询条件，找到要删除的记录
2. 获取记录：results.one() 确保只返回一条记录，如果不存在或多条会抛出异常
3. 执行删除：session.delete(hero) 将记录标记为待删除状态
4. 提交事务：session.commit() 将删除操作持久化到数据库
5. 验证结果：再次查询确认记录已被删除

完整流程示例

def main():
    create_db_and_tables()  # 创建数据库和表
    create_heroes()        # 插入初始数据
    update_heroes()        # 更新数据
    delete_heroes()        # 删除数据

这个完整示例展示了从创建数据库到最终删除记录的完整生命周期。

最佳实践

1. 事务管理：始终使用 with Session(engine) as session 确保会话正确关闭
2. 错误处理：实际应用中应添加 try-except 块处理可能的异常
3. 批量删除：可以使用 session.exec(delete(Hero).where(...)) 实现批量删除
4. 级联删除：定义模型关系时考虑设置级联删除规则

总结

通过 SQLModel 实现数据库删除操作非常简单直观。关键点在于：

• 先查询获取要删除的记录
• 使用 session.delete() 标记删除
• 通过 session.commit() 提交变更
• 最后验证删除结果

这种模式既保证了操作的原子性，又提供了良好的可读性，是处理数据库删除操作的理想选择。

掌握这些基础知识后，你可以进一步探索 SQLModel 的其他高级特性，如关系模型、复杂查询等，构建更加强大的数据访问层。

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