From 9e38502edec81c47a54641186d888f191ed999f0 Mon Sep 17 00:00:00 2001 From: HombreLaser Date: Mon, 26 Feb 2024 15:40:55 -0600 Subject: Add pagination handling --- src/queries/comments_query.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) (limited to 'src/queries/comments_query.py') diff --git a/src/queries/comments_query.py b/src/queries/comments_query.py index e343667..5312cca 100644 --- a/src/queries/comments_query.py +++ b/src/queries/comments_query.py @@ -10,9 +10,16 @@ class CommentsQuery(BaseQuery): def all_comments(self): return db.paginate(db.select(Comment)) - def comments_of_post(self, post): + def comments_of_post(self, post, page=1): + try: + if page is not None and isinstance(page, str): + page = int(page) + except ValueError: + page = 1 + return db.paginate( - db.select(Comment).where(Comment.post == post).order_by(Comment.created_at.desc()) + db.select(Comment).where(Comment.post == post).order_by(Comment.created_at.desc()), + page=page, max_per_page=5 ) def replies_of(self, comment_id): -- cgit v1.2.3