Python 如何按列表对sqlalchemy中的数据进行排序
我有来自外部postgresql数据库的ID列表Python 如何按列表对sqlalchemy中的数据进行排序,python,sql,postgresql,sqlalchemy,Python,Sql,Postgresql,Sqlalchemy,我有来自外部postgresql数据库的ID列表 A = [1,2,3,4,5,6,7,98,0] 我希望使用SQLAlchemy对数据库进行查询,但我希望按照列表对postgresql中的数据进行排序 我已经阅读了很多文档,但找不到任何建议如何做到这一点 因此,在最后,我希望: results = session.query(user).limit(20).offset(10).order_by(A) 干杯 更新: 我找到了解决方案,虽然不如我想象的那么好,但效果很好。无论如何,如果你知道
A = [1,2,3,4,5,6,7,98,0]
我希望使用SQLAlchemy对数据库进行查询,但我希望按照列表对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)")
另一种方法是创建另一个助手表,其中包含每个user.id、join和order的订单位置:
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特定的。如果您不一定需要在SQL中这样做,您可以直接在python中对返回的对象列表进行排序 示例(使用python函数) 您可以尝试:
results = session.query(user).limit(20).offset(10).order_by(*A)
下面是使用SQLAlchemy case表达式进行排序的示例
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)
顺便说一句,我知道postgresql支持“连接值”,但不知道如何用SQLAlchem实现这一点。我很好奇这种排序的原因,你介意详细说明一下吗?我确实考虑过这种方式,但可能应该很慢,因为您必须在数据库中再执行一个操作和一个连接,而不是只执行一个查询。无需在数据库中执行操作,因为您在内存中执行此操作(如代码所示)。它会慢还是不慢?这在很大程度上取决于select语句的长度。
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)