在当前查询中使用 COUNT(*) OVER() 和 SQLAlchemy over PostgreSQL

Question

在将 Python 和 SQLAlchemy 与 PostgreSQL 数据库一起使用的原型应用程序中，我有以下架构（摘录）：

class Guest(Base):
    __tablename__ = 'guest'
    id = Column(Integer, primary_key=True)
    name = Column(String(50))
    surname = Column(String(50))
    email = Column(String(255))
    [..]
    deleted = Column(Date, default=None)

我想使用 SQLAlchemy 构建一个查询来检索来宾列表，以显示在后台。

为了实现分页，我将使用 LIMIT 和 OFFSET，以及 COUNT(*) OVER() 来获取执行查询时的记录总数（而不是使用不同的查询）。

SQL 查询的示例可能是：

  SELECT id, name, surname, email,
       COUNT(*) OVER() AS total
    FROM guest
   WHERE (deleted IS NULL)
ORDER BY id ASC
   LIMIT 50
  OFFSET 0

如果我要使用 SQLAlchemy 构建查询，我可以执行以下操作：

query = session.query(Guest)
query = query.filter(Login.deleted == None)
query = query.order_by(Guest.id.asc())
query = query.offset(0)
query = query.limit(50)
result = query.all()

如果我想计算来宾表中的所有行，我可以这样做：

from sqlalchemy import func
query = session.query(func.count(Guest.id))
query = query.filter(Login.deleted == None)
result = query.scalar()

现在我要问的问题是如何使用 SQLAlchemy 执行一个查询，类似于上面的查询，用一颗石头杀死两只鸟（返回前 50 行和总行数以构建分页链接，所有在一个查询中）。

有趣的一点是在 PostgreSQL 中使用了窗口函数，它允许上述行为，从而使您不必查询两次而只查询一次。

是否可以？

提前致谢。

Answer 1

所以我在 SQLAlchemy 文档中找不到任何示例，但我找到了这些函数：

• count()
• over()
• label()

我设法将它们结合起来以产生我正在寻找的结果：

from sqlalchemy import func
query = session.query(Guest, func.count(Guest.id).over().label('total'))
query = query.filter(Guest.deleted == None)
query = query.order_by(Guest.id.asc())
query = query.offset(0)
query = query.limit(50)
result = query.all()

干杯!

PS 我也在Stack Overflow 上发现了这个问题，没有回答。