python - Sqlalchemy 按列表中的字段过滤但保持原始顺序？

Question

我有一个像这样的鞋子模型:

class Shoe(db.Model):
id = db.Column(db.Integer, primary_key = True)
asin = db.Column(db.String(20), index = True)

我有一个像 ids = [2,1,3] 这样的 id 列表，当我查询 Shoe 模型时，结果在“ids”列表中有 id，我想返回: [{id:2, asin:"111"},{id:1, asin:"113"},{id:3, asin:"42"}] 但问题是使用下面的查询语句不会保留原始顺序，结果将随机返回。如何保持筛选所依据的列表的顺序？

不正确的一个:Shoe.query.filter(Shoe.id.in_(my_list_of_ids)).all()

Answer 1

如果您有一个合理的小型 ID 列表，您可以对每个 ID 单独执行 SQL 查询:

[Shoe.query.filter_by(id=id).one() for id in my_list_of_ids]

对于大量的id，SQL查询会耗时较长。然后，您最好使用单个查询并在第二步中以正确的顺序放置值(借用自 how to select an object from a list of objects by its attribute in python ):

shoes = Shoe.query.filter(Shoe.id.in_(my_list_of_ids)).all()
[next(s for s in shoes if s.id == id) for id in my_list_of_ids]

这是假设 id 是唯一的(在您的情况下它们应该是唯一的)。如果有多个具有相同 id 的元素，第一个方法将引发异常。