如何使用 SQLAlchemy 中的聚合 function 查询 @hybrid_property？

Question

这是我的理智：我有一个名为 Order 的 model：我有一些这样的字段：

@hybrid_property
    def final_total(self):
        total = self.subtotal

        if self.final_discount_amount is not None:
            total -= self.final_discount_amount

        if self.final_tax_cost is not None:
            total += self.final_tax_cost

        if self.final_tip is not None:
            total += self.final_tip

        if self.service_fee is not None:
            total += self.service_fee

        return total

    @final_total.expression
    def final_total(cls):
        return func.sum(cls.final_total).label("final_total")

我想使用查询 obj 从数据库中获取一条记录的 final_total。

这是我的查询：

session.query(Order.final_total).filter_by(order.id=1).first()

我收到如下错误：

python3.8/site-packages/sqlalchemy/ext/hybrid.py", line 1090, in _expr
    return ExprComparator(cls, expr(cls), self)
RecursionError: maximum recursion depth exceeded

Answer 1

可以使用 hybrid_property 来做到这一点，但您可能会面临巨大的性能问题，因为这个final_total计算必须在查询运行时完成......更好的选择是向您添加新列final_total model（和 db）并在任何元素（例如 final_tip 等）都被更改。

但是，如果混合属性性能对您来说足够了，那么这是应该有效的实现：

from sqlalchemy import func
    
    @hybrid_property
    def final_total(self):
        total = self.subtotal
        if self.final_discount_amount is not None:
            total -= self.final_discount_amount
        if self.final_tax_cost is not None:
            total += self.final_tax_cost
        if self.final_tip is not None:
            total += self.final_tip
        if self.service_fee is not None:
            total += self.service_fee
        return total

    @calc.expression
    def final_total(cls):
        return (
            cls.subtotal +
            func.coalesce(cls.final_discount_amount, 0) 
            func.coalesce(cls.service_fee, 0) + 
            func.coalesce(cls.final_tax_cost, 0) +
            func.coalesce(cls.final_tip, 0)
        )