From fcd9465d564b08ae289417fb0616942b38fd0836 Mon Sep 17 00:00:00 2001 From: Mitsuo Tokumori Date: Sun, 12 Nov 2023 18:18:35 -0500 Subject: Move pagination functions to db.py --- ustayml/db.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) (limited to 'ustayml/db.py') diff --git a/ustayml/db.py b/ustayml/db.py index 8fb1ec9..25d0188 100644 --- a/ustayml/db.py +++ b/ustayml/db.py @@ -1,3 +1,4 @@ +import math import psycopg import click @@ -51,4 +52,17 @@ def init_db_command(): def init_app(app): app.teardown_appcontext(close_db) # callback after returning response - app.cli.add_command(init_db_command) \ No newline at end of file + app.cli.add_command(init_db_command) + + +# Helper functions + +def get_paginated_items(query: str, params: list=[], pagination: dict={}): + limit = pagination.get('pagesize', 100) + offset = pagination.get('page', 0) * limit + with get_db().cursor() as cursor: + cursor.execute(f"{query} LIMIT %s OFFSET %s;", params + [limit, offset]) + pagination['n_pages'] = math.ceil(cursor.rowcount / limit) + pagination['rowcount'] = cursor.rowcount + items = cursor.fetchall() + return items \ No newline at end of file -- cgit v1.2.3