如何按列表对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的排序函数）

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

Answer 2

另一种方法是创建另一个辅助表，其中包含每个 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 特定的。