diff --git a/code/app.py b/code/app.py index 0ee8d21..63e92ad 100644 --- a/code/app.py +++ b/code/app.py @@ -15,7 +15,7 @@ from flask import ( ) from werkzeug.utils import secure_filename from revprox import ReverseProxied -from utils.misc import random_token, hash_password, verify_password,file_date_human +from utils.misc import random_token, hash_password, verify_password, file_date_human from utils.files import ( db_store_file, file_details, @@ -48,12 +48,14 @@ app.wsgi_app = ReverseProxied(app.wsgi_app) def log_request_info(): pass + @app.after_request def log_the_status_code(response): ip = request.environ.get("HTTP_X_FORWARDED_FOR", request.remote_addr) - app.logger.info(f'{request.method}:{response.status_code} {request.url} {ip}') + app.logger.info(f"{request.method}:{response.status_code} {request.url} {ip}") return response + @app.route("/") def index(): """Returns Nothing""" @@ -138,7 +140,7 @@ def upload(): db_store_file(token, safe_filename, expires, max_dl, password) download_url = file_full_url(token, safe_filename) app.logger.info( - f'Upload: {download_url} MaxDL:{max_dl} Exp:{file_date_human(expires)}' + f"Upload: {download_url} MaxDL:{max_dl} Exp:{file_date_human(expires)}" ) return "File uploaded\n%s\n" % (download_url,), 200 @@ -242,14 +244,15 @@ def login(): session[request.form["token"]] = request.form["password"] return redirect(request.form["redirect"]) - if set(('name','redirect','token')).issubset(session.keys()): + if set(("name", "redirect", "token")).issubset(session.keys()): return render_template( "login.html", filename=session["name"], redirect=session["redirect"], token=session["token"], ) - return "",400 + return "", 400 + @app.route("/logout", methods=["GET"]) def logout():