SQLModel 教程：使用 WHERE 子句进行条件查询

最新推荐文章于 2025-06-02 09:08:10 发布

原创最新推荐文章于 2025-06-02 09:08:10 发布 · 259 阅读

10 ·

CC 4.0 BY-SA版权

SQLModel 教程：使用 WHERE 子句进行条件查询

概述

本文介绍如何在 SQLModel 中使用 WHERE 子句进行条件查询。SQLModel 是一个强大的 Python ORM 工具，它结合了 SQLAlchemy 和 Pydantic 的优点，使得数据库操作更加简单直观。

数据模型定义

首先，我们需要定义一个数据模型来表示我们的英雄数据：

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

这个 Hero 类定义了英雄的基本属性：

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)

这个方法会创建所有定义的数据模型对应的数据库表。

插入测试数据

我们创建了一些英雄数据并插入到数据库中：

def create_heroes():
    hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson")
    hero_2 = Hero(name="Spider-Boy", secret_name="Pedro Parqueador")
    # 其他英雄...
    
    with Session(engine) as session:
        session.add(hero_1)
        session.add(hero_2)
        # 添加其他英雄...
        session.commit()

条件查询

核心部分是我们如何使用 WHERE 子句进行条件查询：

def select_heroes():
    with Session(engine) as session:
        statement = select(Hero).where(Hero.age <= 35)
        results = session.exec(statement)
        for hero in results:
            print(hero)

这段代码会查询所有年龄小于等于 35 岁的英雄。SQLModel 的查询语法非常直观：