postgresql - 从 SQLAlchemy 中的查询返回 "computed"列

Question

假设我有以下类(class):

class Basket(db.Model):
    id = Column('id', Integer, primary_key=True)

class Fruit(db.Model):
    id = Column('id', Integer, primary_key=True)
    basket_id = Column(Integer, ForeignKey('basket.id'))

我想对 fruit 表进行查询，在其结果中返回一个名为 in_basket 的“计算”列，该列为 True如果水果的 basket_id 在篮子 ID [1, 2, 3] 列表中，否则为 False。

在 SQLAlchemy 中有没有办法做到这一点？我正在使用 flask-sqlalchemy 和 Postgres 9。

谢谢。

Answer 1

使用 case 应该很简单表达式:

class Fruit(...):
    ...
    def is_in_basket(self, busket_list):
        return self.basket_id in busket_list

def get_fruits(basket_list):
    from sqlalchemy.sql.expression import case
    xpr = case([(Fruit.basket_id.in_(basket_list), True),], else_ = False).label("in_basket")
    qry = session.query(Fruit, xpr)
    fruits = []
    for fruit, in_basket in qry:
        fruit.in_basket = in_basket # assign value to a Fruit property (as per comment below)
        fruits.append(fruit)
    return fruits

fruits = get_fruits(my_basket_list)
for f in fruits:
    print f, f.in_basket, f.is_in_basket(my_basket_list)