diff --git a/raincloud.py b/raincloud.py index 44d6585..858ff3c 100755 --- a/raincloud.py +++ b/raincloud.py @@ -7,6 +7,7 @@ import werkzeug from datetime import datetime, timedelta from hmac import compare_digest as compare_hash from flask import ( + abort, Flask, render_template, redirect, @@ -22,10 +23,6 @@ base_path = Path("public") cloud_name = "raincloud" sessions = [] -# mkpass = "$6$7lTWfEYgx.nZdM.C$jb2cQt30FzpnEibp2sN2juGL0sGT2Y9dGlVTQvqxBB579Yy5lfbt3tIHjKYnt/MIcff6I6AFp8Q5k1xjN9C8a0" -# print(compare_hash(mkpass, crypt.crypt("test", mkpass))) -# exit(1) - class RaincloudIOException(Exception): pass @@ -156,7 +153,7 @@ def directory(directory, filename=None): if config["download"] and filename != "rc.toml": return send_from_directory(base_path / directory, filename) else: - return "Not allowed" + abort(403) # Upload elif request.method == "POST": @@ -169,11 +166,11 @@ def directory(directory, filename=None): # Reload return redirect(url_for("directory", directory=directory)) else: - return "No upload allowed" + abort(403) except RaincloudIOException as e: print(e) - return "No 404 file" + abort(404) app.secret_key = "raincloud"