summaryrefslogtreecommitdiff
path: root/src/queries/comments_query.py
blob: 834bd6a22f93e0401ab43221a00298d55cd85af9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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
        )