Revamp comments style like sql statements

1 files changed, 7 insertions, 6 deletions

diff --git a/src/db_handler.py b/src/db_handler.py
index 6789105..471154a 100644
--- a/src/db_handler.py
+++ b/src/db_handler.py
@@ -15,14 +15,15 @@ class DBHandler:
     def get_comments(self, url: str) -> list[dict]:
         with self.con_factory() as con:
             cur = con.cursor()
-            cur.execute(
-                '''
-                SELECT id, nickname, comment, page_url, visitor_url, (CASE WHEN show_contact THEN contact ELSE NULL END) as contact_info, created_at
-                FROM comment WHERE approved AND public AND page_url = ? ORDER BY created_at DESC;
-                ''', (url,))
+            comments_sql = '''
+SELECT id, nickname, comment, page_url, visitor_url, (CASE WHEN show_contact THEN contact ELSE NULL END) as contact_info, created_at
+FROM comment WHERE approved AND public AND page_url = ? ORDER BY created_at DESC;
+'''
+            cur.execute(comments_sql, (url,))
             comments = cur.fetchall()
             headers = list(map(lambda attr : attr[0], cur.description))
-            return comments, headers
+            comments_sql = comments_sql.replace('?', f"'{url}'")
+            return comments, headers, comments_sql
 
 
     def post_comment(self, comment: str, page_url: str, visitor_url: str=None, nickname: str=None, contact: str=None,