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