如何按列表在sqlalchemy中订购数据

时间:2014-04-30 07:00:42

标签: python sql postgresql sqlalchemy

我有来自外部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)")

4 个答案:

答案 0 :(得分:2)

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

示例(使用python' s sorted函数)

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

答案 1 :(得分:0)

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

答案 2 :(得分:0)

您可以尝试:

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

答案 3 :(得分:0)

这里是使用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)