diff options
Diffstat (limited to 'src/queries/comments_query.py')
-rw-r--r-- | src/queries/comments_query.py | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/src/queries/comments_query.py b/src/queries/comments_query.py new file mode 100644 index 0000000..834bd6a --- /dev/null +++ b/src/queries/comments_query.py @@ -0,0 +1,19 @@ +from src.database import db +from src.database.models import Comment +from src.queries.base_query import BaseQuery + + +class CommentsQuery(BaseQuery): + def __init__(self): + super().__init__(Comment) + + def all_comments(self, page=1, per_page=10, max_per_page=15): + return db.paginate(db.select(Comment), page=page, per_page=per_page, + max_per_page=max_per_page) + + def comments_of_post(self, post, page=1, per_page=10, max_per_page=15): + return db.paginate( + db.select(Comment).where(Comment.post == post), + page=page, per_page=per_page, + max_per_page=max_per_page + ) |