使用 nosql 排名的 SQLalchemy 中的排序和分页

Question

我有一个 python 算法，它返回给定用户的数据库行的排名顺序。该算法输出一个主键 ID 列表（这些键可以与 post.id 连接）。如下所示，除了可能有数千个匹配项：

result_rank = [1286, 1956, 6759, 3485, 2309, 3409, 9023, 912, 13098, 23489, 19023, 1239]

我想指示 sqlalchemy 选择这些行，并按照它们在列表中的顺序对它们进行排序。问题是我想对此执行pagination

results = posts.query().filter(posts.id.in_(
    resultIds)).order_by(??? how can I order by post.id = result_rank ???).paginate(page_num, posts_per_page)

我使用 Postgresql 作为数据库。

Answer 1

除非有好的解决方案，否则我将自己破解分页对象：

class paginate_obj:

    """ Pagination dummy object. Takes a list and paginates it similar to sqlalchemy paginate() """
    def __init__(self, paginatable, page, per_page):
        self.has_next = (len(paginatable)/per_page) > page
        self.has_prev = bool(page - 1)
        self.next = page + self.has_next
        self.prev = page - self.has_prev
        self.items = paginatable[(page-1)*(per_page):(page)*(per_page)]

我认为进行排序的唯一方法是创建一个所有结果的列表，并根据一些 lambda 函数在 python 中对其进行排序：

results = my_table.query().all()
results.sort(key=lamba x: distance(x.lat, x.long, user_lat, user_long)
paginated_results = paginate_obj(results, 1, 10) #returns the first page of 10 elements

Answer 2

我认为排序更重要，因为没有它，数据库级别的分页完全没有用。注意到这一点后，我的答案根本不涉及分页方面，但我认为即使@mgoldwasser 提供的答案也可以用于此。

这是我想出的，以便能够选择一些对象并根据初始过滤器列表保留它们的顺序。代码不言自明：

# input
post_ids = [3, 4, 1]

# create helper (temporary in-query table with two columns: post_id, sort_order)
# this table looks like this:
# key | sort_order
#   3 |          0
#   4 |          1
#   1 |          2
q_subq = "\nUNION ALL\n".join(
    "SELECT {} AS key, {} AS sort_order".format(_id, i)
    for i, _id in enumerate(post_ids)
)

# wrap it in a `Selectable` so that we can use JOINs
s = (select([literal_column("key", Integer),
             literal_column("sort_order", Integer)])
     .select_from(text("({}) AS helper".format(text(q_subq))))
     ).alias("helper")

# actual query which is both the filter and sorter
q = (session.query(Post)
     .join(s, Post.id == s.c.key)  # INNER JOIN will filter implicitly
     .order_by(s.c.sort_order)  # apply sort order
     )

它适用于postgresql 和sqlite。