34 lines
1,021 B
Python
34 lines
1,021 B
Python
from peewee import SQL
|
|
|
|
def paginate(query, model, descending=False, page_token=None, limit=50, id_alias=None):
|
|
""" Paginates the given query using an ID range, starting at the optional page_token.
|
|
Returns a *list* of matching results along with an unencrypted page_token for the
|
|
next page, if any. If descending is set to True, orders by the ID descending rather
|
|
than ascending.
|
|
"""
|
|
id_field = model.id
|
|
if id_alias is not None:
|
|
id_field = SQL(id_alias)
|
|
|
|
if descending:
|
|
query = query.order_by(id_field.desc())
|
|
else:
|
|
query = query.order_by(id_field)
|
|
|
|
if page_token is not None:
|
|
start_id = page_token.get('start_id')
|
|
if start_id is not None:
|
|
if descending:
|
|
query = query.where(model.id <= start_id)
|
|
else:
|
|
query = query.where(model.id >= start_id)
|
|
|
|
results = list(query)
|
|
page_token = None
|
|
if len(results) > limit:
|
|
start_id = results[limit].id
|
|
page_token = {
|
|
'start_id': start_id
|
|
}
|
|
|
|
return results[0:limit], page_token
|