Try to cleanup the clean_input function
This commit is contained in:
parent
49c686e54f
commit
bd53cb477e
30
src/main.py
30
src/main.py
|
@ -131,6 +131,20 @@ def empty(value):
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
def _clean_ttl(request):
|
||||||
|
if not request.form.get("ttl"):
|
||||||
|
return 604800
|
||||||
|
|
||||||
|
try:
|
||||||
|
time_period = int(request.form.get("ttl"))
|
||||||
|
except ValueError:
|
||||||
|
abort(400, "TTL must be an integer")
|
||||||
|
|
||||||
|
if time_period > 2419200:
|
||||||
|
abort(400, "TTL must be less than 2419200 seconds (4 weeks)")
|
||||||
|
return time_period
|
||||||
|
|
||||||
|
|
||||||
def clean_input():
|
def clean_input():
|
||||||
"""
|
"""
|
||||||
Make sure we're not getting bad data from the front end,
|
Make sure we're not getting bad data from the front end,
|
||||||
|
@ -139,20 +153,8 @@ def clean_input():
|
||||||
if empty(request.form.get("password", "")):
|
if empty(request.form.get("password", "")):
|
||||||
abort(400)
|
abort(400)
|
||||||
|
|
||||||
if not request.form.get("ttl"):
|
time_period = _clean_ttl(request)
|
||||||
time_period = 604800
|
return time_period, request.form["password"]
|
||||||
|
|
||||||
if request.form.get("ttl"):
|
|
||||||
if request.form.get("ttl").isdigit():
|
|
||||||
time_period = int(request.form.get("ttl"))
|
|
||||||
else:
|
|
||||||
abort(400, "TTL must be an integer")
|
|
||||||
|
|
||||||
if time_period > 2419200:
|
|
||||||
abort(400, "TTL must be less than 2419200 seconds (4 weeks)")
|
|
||||||
|
|
||||||
if request.form["password"]:
|
|
||||||
return time_period, request.form["password"]
|
|
||||||
|
|
||||||
|
|
||||||
@app.route("/", methods=["POST"])
|
@app.route("/", methods=["POST"])
|
||||||
|
|
Loading…
Reference in New Issue