diff --git a/app/routes/posts.py b/app/routes/posts.py index 7f695c8..caaa1d3 100644 --- a/app/routes/posts.py +++ b/app/routes/posts.py @@ -48,6 +48,9 @@ def update_post(post_id, new_content, markup_language='babycode'): @login_required def delete(post_id): post = Posts.find({'id': post_id}) + if not post: + return redirect(url_for('topics.all_topics')) + thread = Threads.find({'id': post.thread_id}) user = get_active_user() if not user: @@ -76,6 +79,10 @@ def delete(post_id): @bp.get("//edit") @login_required def edit(post_id): + post = Posts.find({'id': post_id}) + if not post: + return redirect(url_for('topics.all_topics')) + user = get_active_user() q = f"{Posts.FULL_POSTS_QUERY} WHERE posts.id = ?" editing_post = db.fetch_one(q, post_id) @@ -106,6 +113,8 @@ def edit(post_id): def edit_form(post_id): user = get_active_user() post = Posts.find({'id': post_id}) + if not post: + return redirect(url_for('topics.all_topics')) if post.user_id != user.id: return redirect(url_for('topics.all_topics'))