avoid duplicating err and msg code

pull/102/head
Aevann 2023-01-27 11:41:32 +02:00
parent fd343424ec
commit 2c332c5fa7
3 changed files with 14 additions and 12 deletions

View File

@ -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

View File

@ -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")

View File

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