Add comments processing

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
+        )