如何按列表对sqlalchemy中的数据进行排序

Question

我有来自外部 postgresql 数据库的 ID 列表。

A = [1,2,3,4,5,6,7,98,0]

我会使用 SQLAlchemy 对数据库进行查询，但我会按 A 列表对 postgresql 中的数据进行排序。

我已阅读大量文档，但找不到任何建议。

所以，最后我会：

results = session.query(user).limit(20).offset(10).order_by(A)

干杯

更新：

我找到了解决方案，它没有我预期的那么好，但效果很好。无论如何，如果您知道更好的解决方案，请告诉我！

ids = ','.join([str(i) for i in A])
results = session.query(user).filter(user.id.in_(A)).\
limit(20).offset(10).\
order_by(" position(CONCAT(',',users.id::text,',') in ',{0},'.format(ids)")

Answer 1

如果您不一定需要在 SQL 中执行此操作，您可以直接在 python 中简单地对返回的对象列表进行排序。

示例（使用python的sorted函数）

results = session.query(user).filter(user.id.in_(A)).all()
results = sorted(results, key=lambda o: A.index(o.id))

Answer 2

这是使用 SQLAlchemy 大小写表达式进行排序的示例。

from sqlalchemy import case

from core.orm import session
from core.orm import models

query = session.query(models.Model)

ids_list = [1, 2, 4, 3, 5]

# order query by index of id in `id_list`

id_ordering = case(
    {_id: index for index, _id in enumerate(ids_list)},
    value=models.Model.id
)

# order 
query = query.order_by(id_ordering)

Answer 3

你可以试试：

results = session.query(user).limit(20).offset(10).order_by(*A)

Answer 4

另一种方法是创建另一个帮助表，其中包含每个 user.id 的订单位置、加入它和订单：

A = [1,2,3,4,5,6,7,98,0]
stmt = " UNION ALL ".join( 'SELECT {0} AS user_id, {1} AS order_id'.format(uid, row)
        for row, uid in enumerate(A))
ordq = text(stmt).columns(user_id=Integer, order_id=Integer).alias("t")
results = session.query(user).join(ordq, user.id == ordq.c.user_id).order_by(ordq.c.order_id).all()

我无法判断这是否比您的版本更好，但它至少应该是非 RDBMS 特定的。