From 3a433e69cd82a756fa5f0f7dffb7d6d84d706d7f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?El=C4=ABna=20Lukjanovi=C4=8Da?= Date: Mon, 29 Apr 2024 09:26:32 +0000 Subject: [PATCH] Update main.py --- main.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 9ab4e49..b115373 100644 --- a/main.py +++ b/main.py @@ -12,6 +12,7 @@ from functools import wraps from datetime import datetime import os import logging +from sqlalchemy import func # import csv # from sqlalchemy import create_engine, Table, MetaData @@ -278,14 +279,16 @@ def create_post(category_id): def view_post(category_name, post_id): post = ForumPost.query.get_or_404(post_id) comments = ForumComment.query.filter_by(post_id=post_id).all() + comment_count = len(comments) if request.method == 'POST': created_by = request.form['created_by'] text = request.form['text'] - new_comment = ForumComment(post_id=post_id, created_by=created_by, text=text) # type: ignore + creation_date = datetime.now() + new_comment = ForumComment(post_id=post_id, created_by=created_by, text=text, creation_date=creation_date) db.session.add(new_comment) db.session.commit() return redirect(url_for('view_post', category_name=category_name, post_id=post_id)) - return render_template('post.html', post=post, comments=comments) + return render_template('post.html', post=post, comment_count=comment_count, comments=comments) @app.route('/forums/upvote_post/') def upvote_post(post_id):