如何在 Flask-SQLAlchemy 中对相关表的列求和

Posted 2023-03-16

【中文标题】如何在 Flask-SQLAlchemy 中对相关表的列求和

【英文标题】：How to Sum a column of a related table in Flask-SQLAlchemy

【发布时间】：2022-01-20 19:01:25

【问题描述】：

我有一个跟踪班次信息的烧瓶应用程序，并且在班次期间记录的订单，模型是这样设置的

class Shift(db.Model):
  # Columns
  orders = db.relationship('Orders', lazy='dynamic')


class Orders(db.Model):
  pay = db.Column(db.Integer)
  dash_id = db.Column(db.Integer, db.ForeignKey('dash.id'))

当用户正在轮班时，我想显示他们到目前为止所支付的总工资，稍后我也会将其提交到 Shift 表中。为了获得所有相关订单的总工资，我尝试查询类似

current_shift = Shift.query.filter_by(id=session['shiftID']).first()
orders = current_shift.orders
total_pay = func.sum(orders.pay)

但它总是返回 'AppenderBaseQuery' object has no attribute 'pay' 我知道我可以像这样循环遍历

total_pay = 0
for order in orders:
    total_pay += order

但这不能像查询中的聚合函数那样快速、高效或肯定可读。

我的问题是，对相关订单的 Orders.pay 列求和（或执行任何列的聚合函数）的正确方法是什么？

【参考方案1】：

您无需查看班次表，因为您已经在订单表中获得了所需的所有信息。

要获得单班的结果，您可以这样做

pay = db_session.query(func.sum(Orders.pay)).filter(Orders.shifts_id == shift_id).one()

或多个班次

pays = ( 
    s.query(Orders.shifts_id, sa.func.sum(Orders.pay))
    .filter(Orders.shifts_id.in_(list_of_shift_ids))
    .group_by(Orders.shifts_id)
    .all()
)  

请注意，这两个查询都以元组形式返回行，例如分别为 (50,)、[(25,), (50,)]。