diff --git a/files/helpers/get.py b/files/helpers/get.py index 39e9f1ccb..9f8b21fbe 100644 --- a/files/helpers/get.py +++ b/files/helpers/get.py @@ -361,3 +361,15 @@ def get_profile_picture(identifier:Union[int, str]) -> str: x = get_user(identifier, graceful=True) return x.profile_url if x else 'not_found' + +def get_err(): + if request.referrer and request.referrer.split('?')[0] == request.base_url: + return request.values.get("err") + else: + return None + +def get_msg(): + if request.referrer and request.referrer.split('?')[0] == request.base_url: + return request.values.get("msg") + else: + return None diff --git a/files/routes/static.py b/files/routes/static.py index c3356b6dd..021fbccdb 100644 --- a/files/routes/static.py +++ b/files/routes/static.py @@ -213,12 +213,7 @@ def api(v): @app.get("/media") @auth_desired def contact(v:Optional[User]): - if request.referrer and request.referrer.split('?')[0] == request.base_url: - msg = request.values.get("msg") - else: - msg = None - - return render_template("contact.html", v=v, msg=msg) + return render_template("contact.html", v=v, msg=get_msg()) @app.post("/send_admin") @limiter.limit("1/second;1/2 minutes;10/day") diff --git a/files/routes/subs.py b/files/routes/subs.py index 54c4f197f..07155f6c9 100644 --- a/files/routes/subs.py +++ b/files/routes/subs.py @@ -321,12 +321,7 @@ def create_sub(v): if not v.can_create_hole: abort(403) - if request.referrer and request.referrer.split('?')[0] == request.base_url: - err = request.values.get("err") - else: - err = None - - return render_template("sub/create_hole.html", v=v, cost=HOLE_COST, error=err) + return render_template("sub/create_hole.html", v=v, cost=HOLE_COST, error=get_err()) @app.post("/create_hole") @limiter.limit(DEFAULT_RATELIMIT, key_func=get_ID)