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):
        return db.paginate(db.select(Comment))

    def comments_of_post(self, post):
        return db.paginate(
            db.select(Comment).where(Comment.post == post),
        )