from files.helpers.wrappers import * from flask import * from urllib.parse import quote, urlencode import time from files.__main__ import app, limiter @app.errorhandler(400) def error_400(e): if request.headers.get("Authorization") or request.headers.get("xhr"): return {"error": "400 Bad Request"}, 400 else: return render_template('errors/400.html', err=True), 400 @app.errorhandler(401) def error_401(e): if request.headers.get("Authorization") or request.headers.get("xhr"): return {"error": "401 Not Authorized"}, 401 else: path = request.path qs = urlencode(dict(request.values)) argval = quote(f"{path}?{qs}", safe='') return redirect(f"/login?redirect={argval}") @app.errorhandler(403) def error_403(e): description = e.description if description == "You don't have the permission to access the requested resource. It is either read-protected or not readable by the server.": description = '' if request.headers.get("Authorization") or request.headers.get("xhr"): if not description: description = "403 Forbidden" return {"error": description}, 403 else: if not description: description = "YOU AREN'T WELCOME HERE GO AWAY" return render_template('errors/403.html', description=description, err=True), 403 @app.errorhandler(404) def error_404(e): if request.headers.get("Authorization") or request.headers.get("xhr"): return {"error": "404 Not Found"}, 404 else: return render_template('errors/404.html', err=True), 404 @app.errorhandler(405) def error_405(e): if request.headers.get("Authorization") or request.headers.get("xhr"): return {"error": "405 Method Not Allowed"}, 405 else: return render_template('errors/405.html', err=True), 405 @app.errorhandler(413) def error_413(e): return {"error": "Max file size is 8 MB (16 MB for paypigs)"}, 413 @app.errorhandler(429) def error_429(e): if request.headers.get("Authorization") or request.headers.get("xhr"): return {"error": "429 Too Many Requests"}, 429 else: return render_template('errors/429.html', err=True), 429 @app.errorhandler(500) def error_500(e): g.db.rollback() if request.headers.get("Authorization") or request.headers.get("xhr"): return {"error": "500 Internal Server Error"}, 500 else: return render_template('errors/500.html', err=True), 500 @app.post("/allow_nsfw") def allow_nsfw(): session["over_18"] = int(time.time()) + 3600 redir = request.values.get("redir") if redir: if redir.startswith(f'{SITE_FULL}/'): return redirect(redir) if redir.startswith('/'): return redirect(f'{SITE_FULL}{redir}') return redirect('/')