moonq/mini-flees
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

format

Browse Source
This commit is contained in:
Ville Rantanen
2023-08-21 09:24:57 +03:00
parent e670161908
commit 5b95c415c6
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
code/app.py
View File

@@ -15,7 +15,7 @@ from flask import (
) )
from werkzeug.utils import secure_filename from werkzeug.utils import secure_filename
from revprox import ReverseProxied 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 ( from utils.files import (
db_store_file, db_store_file,
file_details, file_details,
@@ -48,12 +48,14 @@ app.wsgi_app = ReverseProxied(app.wsgi_app)
def log_request_info(): def log_request_info():
pass pass
@app.after_request @app.after_request
def log_the_status_code(response): def log_the_status_code(response):
ip = request.environ.get("HTTP_X_FORWARDED_FOR", request.remote_addr) 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 return response
@app.route("/") @app.route("/")
def index(): def index():
"""Returns Nothing""" """Returns Nothing"""
@@ -138,7 +140,7 @@ def upload():
db_store_file(token, safe_filename, expires, max_dl, password) db_store_file(token, safe_filename, expires, max_dl, password)
download_url = file_full_url(token, safe_filename) download_url = file_full_url(token, safe_filename)
app.logger.info( 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 return "File uploaded\n%s\n" % (download_url,), 200
@@ -242,14 +244,15 @@ def login():
session[request.form["token"]] = request.form["password"] session[request.form["token"]] = request.form["password"]
return redirect(request.form["redirect"]) return redirect(request.form["redirect"])
if set(('name','redirect','token')).issubset(session.keys()): if set(("name", "redirect", "token")).issubset(session.keys()):
return render_template( return render_template(
"login.html", "login.html",
filename=session["name"], filename=session["name"],
redirect=session["redirect"], redirect=session["redirect"],
token=session["token"], token=session["token"],
) )
return "",400 return "", 400
@app.route("/logout", methods=["GET"]) @app.route("/logout", methods=["GET"])
def logout(): def logout():