SQLAlchemy - 当 count() 表示还有更多结果时，仅返回一个结果

使用 SQLAlchemy 实现当 count() 表示还有更多结果时，仅返回一个结果

python
from sqlalchemy import create_engine, Column, Integer, String, text
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
# 创建 SQLite 内存数据库引擎
engine = create_engine('sqlite:///:memory:', echo=True)
# 声明基类
Base = declarative_base()
# 定义模型
class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    name = Column(String)
# 创建表
Base.metadata.create_all(engine)
# 创建会话
Session = sessionmaker(bind=engine)
session = Session()
# 添加一些数据
session.add_all([
    User(name='Alice'),
    User(name='Bob'),
    User(name='Charlie'),
    # 添加更多的数据...
])
# 提交事务
session.commit()
# 查询是否还有更多结果
has_more_results = session.query(User).filter(User.id > 2).count() > 1
if has_more_results:
    print("还有更多的结果需要处理")
else:
    print("没有更多的结果了")

python
# 优化查询，仅获取一个结果
user = session.query(User).filter(User.id > 2).limit(1).first()
if user:
    print("还有更多的结果需要处理")
else:
    print("没有更多的结果了")