From 05bd034b2344c6f61facc808a8d2b111af77ead0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lera=20Elvo=C3=A9?= Date: Fri, 15 Aug 2025 22:45:19 +0300 Subject: [PATCH] redirect on post delete if target post cant be found --- app/routes/posts.py | 9 +++++++++ 1 file changed, 9 insertions(+) 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'))