if語句在hybrid_property表達式中

Question

我有一個hybrid_property執行一些條件並返回一個響應。 我希望能夠查詢這個字段，所以我創建了一個expression 。 但是，我不確定如何為這種用例編寫條件。

如何根據條件的結果編寫一個表達式，將兩個字段中的一個與另一個字段相乘？

@hybrid_property
def func(self):
    if self.some_column is True:
        return self.true_column * self.amount
    return self.false_column * self.amount

@func.expression
def func(cls):
    # ???
    return

Answer 1

你可以使用case表達式

from sqlalchemy import case

@func.expression
def func(cls):
    return case(
        [
            (cls.some_column == True, cls.true_column * cls.amount),
        ],
        else_=cls.false_column * cls.amount
    )