sqlalchemy的with_entities用法

sqlalchemy的with_entities用法
with_entities是SQLAlchemy 中用于查询时选择特定字段的方法。

它在查询中允许你指定你感兴趣的列，以减小返回结果集的大小，提高性能。

以下是with_entities的基本用法：
from sqlalchemy import create_engine, Column, Integer, String, select
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import Session
# 创建数据库连接
engine = create_engine('sqlite:///:memory:')
# 创建映射基类
Base = declarative_base()
# 定义模型
class User(Base):
__tablename__ = 'users'
id = Column(Integer, primary_key=True)
name = Column(String)
age = Column(Integer)
# 创建表
Base.metadata.create_all(engine)
# 插入一些数据
session = Session(engine)
session.add_all([
User(name='Alice', age=25),
User(name='Bob', age=30),
User(name='Charlie', age=22)
])
mit()
# 使用with_entities选择特定列
result = session.query(User).with_entities(, User.age).all()
for name, age in result:
print(f'Name: {name}, Age: {age}')
# 输出：
# Name: Alice, Age: 25
# Name: Bob, Age: 30
# Name: Charlie, Age: 22
在上述示例中，with_entities(, User.age)指定了我们感兴趣的列，这样查询结果中只包含这两列的数据，而不是整个模型的所有列。

你还可以使用.distinct()来去重结果集，.order_by()来排序结果集等。

总体而言，with_entities提供了在查询中精确选择所需列的便利方法。