SQLModel教程：使用Python ORM实现数据库删除操作详解-优快云博客

本文链接：https://blog.youkuaiyun.com/gitblog_00611/article/details/148376060

SQLModel教程：使用Python ORM实现数据库删除操作详解

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

概述

本文将深入讲解如何使用SQLModel进行数据库删除操作。SQLModel是一个强大的Python ORM工具，它结合了SQLAlchemy和Pydantic的优点，为开发者提供了简单直观的数据库操作接口。

环境准备

首先我们需要定义一个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、姓名、秘密身份和年龄，其中ID是主键，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)

这个方法会创建数据库文件（如果不存在）并根据模型定义创建对应的表结构。

数据初始化

我们先创建一些英雄数据作为测试数据：

def create_heroes():
    hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson")
    # ...其他英雄实例
    with Session(engine) as session:
        session.add(hero_1)
        # ...添加其他英雄
        session.commit()

数据更新操作

在删除前，我们先看看如何更新数据：

def update_heroes():
    with Session(engine) as session:
        statement = select(Hero).where(Hero.name == "Spider-Boy")
        hero_1 = session.exec(statement).one()
        hero_1.age = 16
        hero_1.name = "Spider-Youngster"
        session.add(hero_1)
        session.commit()
        session.refresh(hero_1)

这里我们查询名为"Spider-Boy"的英雄，修改其年龄和名称后提交更改。

核心删除操作

重点来看删除功能的实现：

def delete_heroes():
    with Session(engine) as session:
        # 查询要删除的英雄
        statement = select(Hero).where(Hero.name == "Spider-Youngster")
        results = session.exec(statement)
        hero = results.one()
        print("Hero: ", hero)

        # 执行删除
        session.delete(hero)
        session.commit()

        print("Deleted hero:", hero)

        # 验证删除结果
        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")

删除操作的关键步骤：

首先查询要删除的记录
使用session.delete()方法标记要删除的对象
调用session.commit()提交删除操作
最后验证删除是否成功

完整流程

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