skorotkiewicz/rDrama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

under attack

Browse source
This commit is contained in:
Aevann1 2022-01-09 17:15:02 +02:00
parent 93ce52b766
commit 1bea9edf79
32 changed files with 106 additions and 185 deletions
Download patch file Download diff file Expand all files Collapse all files

42
files/routes/errors.py
View file

@ -1,5 +1,3 @@
import jinja2.exceptions
from files.helpers.wrappers import *
from flask import *
from urllib.parse import quote, urlencode
@ -8,10 +6,9 @@ from files.__main__ import app, limiter
@app.errorhandler(400)
@auth_desired
def error_400(e, v):
def error_400(e):
if request.headers.get("Authorization"): return {"error": "400 Bad Request"}, 400
else: return render_template('errors/400.html', error=True, v=v), 400
else: return render_template('errors/400.html', error=True), 400
@app.errorhandler(401)
def error_401(e):
@ -26,40 +23,35 @@ def error_401(e):
@app.errorhandler(403)
@auth_desired
def error_403(e, v):
def error_403(e):
if request.headers.get("Authorization"): return {"error": "403 Forbidden"}, 403
else: return render_template('errors/403.html', error=True, v=v), 403
else: return render_template('errors/403.html', error=True), 403
@app.errorhandler(404)
@auth_desired
def error_404(e, v):
def error_404(e):
if request.headers.get("Authorization"): return {"error": "404 Not Found"}, 404
else: return render_template('errors/404.html', error=True, v=v), 404
else: return render_template('errors/404.html', error=True), 404
@app.errorhandler(405)
@auth_desired
def error_405(e, v):
def error_405(e):
if request.headers.get("Authorization"): return {"error": "405 Method Not Allowed"}, 405
else: return render_template('errors/405.html', error=True, v=v), 405
else: return render_template('errors/405.html', error=True), 405
@app.errorhandler(429)
@auth_desired
def error_429(e, v):
def error_429(e):
if request.headers.get("Authorization"): return {"error": "429 Too Many Requests"}, 429
else: return render_template('errors/429.html', error=True, v=v), 429
else: return render_template('errors/429.html', error=True), 429
@app.errorhandler(500)
@auth_desired
def error_500(e, v):
def error_500(e):
g.db.rollback()
if request.headers.get("Authorization"): return {"error": "500 Internal Server Error"}, 500
else: return render_template('errors/500.html', error=True, v=v), 500
else: return render_template('errors/500.html', error=True), 500
@app.post("/allow_nsfw")
@ -69,11 +61,7 @@ def allow_nsfw():
@app.get("/error/<error>")
@auth_desired
def error_all_preview(error, v):
try:
return render_template(f"errors/{error}.html", error=True, v=v)
except jinja2.exceptions.TemplateNotFound:
abort(400)
def error_all_preview(error):
try: return render_template(f"errors/{error}.html", error=True)
except: abort(400)