From 35cb3787586e88a2953640b7edbcfa43993ebc02 Mon Sep 17 00:00:00 2001 From: Aevann1 Date: Sat, 5 Feb 2022 23:51:00 +0200 Subject: [PATCH] fgd --- files/routes/front.py | 4 +++- files/routes/posts.py | 5 +++-- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/files/routes/front.py b/files/routes/front.py index a642a0d5c..4bcf205a1 100644 --- a/files/routes/front.py +++ b/files/routes/front.py @@ -133,7 +133,9 @@ def notifications(v): @limiter.limit("3/second;30/minute;400/hour;2000/day") @auth_desired def front_all(v, sub=None): - sub = g.db.query(Sub).filter_by(name=sub).one_or_none() + if sub: sub = g.db.query(Sub).filter_by(name=sub).one_or_none() + + if request.path.startswith('/s/') and not sub: abort(404) if g.webview and not session.get("session_id"): session.permanent = True diff --git a/files/routes/posts.py b/files/routes/posts.py index b0025c4d9..427448d62 100644 --- a/files/routes/posts.py +++ b/files/routes/posts.py @@ -91,9 +91,10 @@ def publish(pid, v): @app.get("/s//submit") @auth_required def submit_get(v, sub=None): - sub = g.db.query(Sub.name).filter_by(name=sub).one_or_none() + if sub: sub = g.db.query(Sub.name).filter_by(name=sub).one_or_none() + if sub: sub = sub[0] - else: sub = None + elif request.path.startswith('/s/'): abort(404) return render_template("submit.html", v=v, sub=sub)