如何用lis对sqlalchemy中的数据进行排序

2024-09-29 06:32:35 发布

您现在位置:Python中文网/ 问答频道 /正文

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

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

我将使用SQLAlchemy对数据库进行查询,但我将在postgresql中按列表对数据进行排序。在

我读了很多文档,但找不到任何建议。在

所以,在决赛中我会:

^{pr2}$

干杯

更新:

我找到了解决办法,虽然没有我预期的那么好,但效果很好。不管怎样,如果你知道更好的解决办法,就让我知道!在

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)")

Tags: 数据in文档id数据库ids列表sqlalchemy
3条回答

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

示例(使用python的sorted函数)

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

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

你可以试试:

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

相关问题 更多 >