SQLAlchemy：优化查询，仅获取模型的部分字段-Python教程-PHP中文网

SQLAlchemy：优化查询，仅获取模型的部分字段

碧海醫心

发布： 2025-08-12 17:48:01

原创

590人浏览过

sqlalchemy：优化查询，仅获取模型的部分字段

本文旨在解决 SQLAlchemy 中查询数据库时，如何只获取模型的部分字段，避免加载不必要的数据，从而优化查询性能的问题。通过对比不同的查询方式，深入探讨 load_only 选项的使用，并强调缓存可能带来的影响，帮助开发者更高效地使用 SQLAlchemy。

SQLAlchemy 中选择特定列的两种方法

在使用 SQLAlchemy 进行数据库查询时，有时我们只需要获取表中的部分列，而不是全部列。这可以减少数据传输量，提高查询效率。 SQLAlchemy 提供了多种方法来实现这个目标。

方法一：直接选择特定列

最直接的方法是在 select 语句中指定需要查询的列。

from sqlalchemy import select
from sqlalchemy.orm import Session

# 假设 PostSQL 是你的模型类
# query = select(PostSQL.id, PostSQL.title)
# result = session.execute(query)
# posts = result.all()
# print(posts) # Output: [(1, 'hello'), (2, 'hello')]

登录后复制

这种方法返回的结果是一个元组的列表，每个元组包含所选列的值。这种方式简单直接，避免了创建完整的模型对象，适用于只需要部分字段的场景。

示例

百度文心一格

百度推出的AI绘画作图工具

112

查看详情

from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class PostSQL(Base):
    __tablename__ = 'posts'

    id = Column(Integer, primary_key=True)
    title = Column(String)
    content = Column(String)
    # 假设还有其他列，例如 author, created_at, updated_at

engine = create_engine('sqlite:///:memory:')  # 使用内存数据库进行演示
Base.metadata.create_all(engine)

Session = sessionmaker(bind=engine)
session = Session()

# 插入一些示例数据
post1 = PostSQL(title='Hello World', content='This is the first post.')
post2 = PostSQL(title='Another Post', content='This is the second post.')
session.add_all([post1, post2])
session.commit()

# 查询 id 和 title 列
from sqlalchemy import select
query = select(PostSQL.id, PostSQL.title)
result = session.execute(query)
posts = result.all()
print(posts)  # Output: [(1, 'Hello World'), (2, 'Another Post')]

session.close()

登录后复制

方法二：使用 load_only 选项

另一种方法是使用 load_only 选项。这种方法会返回完整的模型对象，但只从数据库中加载指定的列。

from sqlalchemy.orm import load_only

# query = select(PostSQL).options(load_only(PostSQL.id, PostSQL.title))
# result = session.execute(query)
# posts = result.scalars().all()
# print(posts) # Output: [<PostSQL object at ...>, <PostSQL object at ...>]
# for post in posts:
#     print(post.id, post.title)

登录后复制

注意事项

缓存问题： 需要注意的是，如果之前已经加载过完整的模型对象， SQLAlchemy 可能会从缓存中读取数据，而不是从数据库中重新加载。这可能导致 load_only 选项失效。解决方法是清除缓存，或者确保在第一次查询时就使用 load_only 选项。
Pycharm/Python 缓存： 某些 IDE 或 Python 环境可能会缓存查询结果，导致 load_only 选项看似没有生效。尝试清除 IDE 或 Python 的缓存，重新运行代码。
异步环境： 在异步环境中使用 SQLAlchemy 时，确保正确使用 await 关键字，并检查异步数据库连接池的配置。

示例

from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.orm import sessionmaker, load_only
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class PostSQL(Base):
    __tablename__ = 'posts'

    id = Column(Integer, primary_key=True)
    title = Column(String)
    content = Column(String)
    # 假设还有其他列，例如 author, created_at, updated_at

engine = create_engine('sqlite:///:memory:')  # 使用内存数据库进行演示
Base.metadata.create_all(engine)

Session = sessionmaker(bind=engine)
session = Session()

# 插入一些示例数据
post1 = PostSQL(title='Hello World', content='This is the first post.')
post2 = PostSQL(title='Another Post', content='This is the second post.')
session.add_all([post1, post2])
session.commit()

# 使用 load_only 查询 id 和 title 列
from sqlalchemy import select
query = select(PostSQL).options(load_only(PostSQL.id, PostSQL.title))
result = session.execute(query)
posts = result.scalars().all()

for post in posts:
    print(f"Post ID: {post.id}, Title: {post.title}")
    # 尝试访问 post.content 会导致延迟加载或错误，取决于配置
    # print(post.content)

session.close()

登录后复制