From 250adf01cfeccb215bb2c07761df426eb391e6b8 Mon Sep 17 00:00:00 2001 From: "Dominik V. Salonen" Date: Fri, 23 Sep 2016 10:57:06 +0200 Subject: [PATCH] Move auth function to cut some lines --- QuadFile/application.py | 8 ++++++++ run.py | 12 ++---------- 2 files changed, 10 insertions(+), 10 deletions(-) create mode 100644 QuadFile/application.py diff --git a/QuadFile/application.py b/QuadFile/application.py new file mode 100644 index 0000000..ee1c231 --- /dev/null +++ b/QuadFile/application.py @@ -0,0 +1,8 @@ +# Very, very simple auth feature. +def basicauth(key, configured): + if configured == "": + return True + elif configured == key: + return True + else: + return False \ No newline at end of file diff --git a/run.py b/run.py index ff46dda..b32083f 100755 --- a/run.py +++ b/run.py @@ -15,6 +15,7 @@ from conf import config # Import QuadFile stuff from QuadFile import db from QuadFile.output import print_log, time_to_string +from QuadFile import application app = Flask(__name__) @@ -53,15 +54,6 @@ def delete_old(): db.delete_entry(file["file"]) -def auth(key): - if config["KEY"] == "": - return True - elif config["KEY"] == key: - return True - else: - return False - - def error_page(error, code): return render_template('error.html', page=config["SITE_DATA"], error=error, code=code) @@ -77,7 +69,7 @@ def allowed_file(filename): def upload_file(): if request.method == 'POST': print_log('Web', 'New file received') - if not auth(request.headers.get('X-Hyozan-Auth')): + if not application.auth(request.headers.get('X-Hyozan-Auth'), config["KEY"]): abort(403) data = dict() file = request.files['file']