forked from Github-Mirrors/canaille
309511e91c
This allows blueprint to handle 404 errors. This is not supported directly in Flask. https://flask.palletsprojects.com/en/stable/errorhandling/#handling However, the blueprint cannot handle 404 routing errors because the 404 occurs at the routing level before the blueprint can be determined.
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
import os
|
|
|
|
import flask
|
|
|
|
try:
|
|
import flask_themer
|
|
except ImportError:
|
|
flask_themer = None
|
|
|
|
|
|
if flask_themer:
|
|
render_template = flask_themer.render_template
|
|
|
|
def setup_themer(app):
|
|
theme_config = app.config["CANAILLE"]["THEME"]
|
|
additional_themes_dir = (
|
|
os.path.abspath(os.path.dirname(theme_config))
|
|
if theme_config and os.path.exists(theme_config)
|
|
else None
|
|
)
|
|
themer = flask_themer.Themer(
|
|
app,
|
|
loaders=[flask_themer.FileSystemThemeLoader(additional_themes_dir)]
|
|
if additional_themes_dir
|
|
else None,
|
|
)
|
|
|
|
@themer.current_theme_loader
|
|
def get_current_theme():
|
|
# if config['THEME'] may be a theme name or a path
|
|
return app.config["CANAILLE"]["THEME"].split("/")[-1]
|
|
|
|
@app.errorhandler(400)
|
|
def bad_request(error):
|
|
return render_template("error.html", description=error, error_code=400), 400
|
|
|
|
@app.errorhandler(403)
|
|
def unauthorized(error):
|
|
return render_template("error.html", description=error, error_code=403), 403
|
|
|
|
@app.errorhandler(404)
|
|
def page_not_found(error):
|
|
from canaille.app.flask import redirect_to_bp_handlers
|
|
|
|
return redirect_to_bp_handlers(app, error) or render_template(
|
|
"error.html", description=error, error_code=404
|
|
), 404
|
|
|
|
@app.errorhandler(500)
|
|
def server_error(error): # pragma: no cover
|
|
return render_template("error.html", description=error, error_code=500), 500
|
|
|
|
else: # pragma: no cover
|
|
render_template = flask.render_template
|
|
|
|
def setup_themer(app):
|
|
@app.errorhandler(404)
|
|
def page_not_found(error):
|
|
from canaille.app.flask import redirect_to_bp_handlers
|
|
|
|
if not redirect_to_bp_handlers(app, error):
|
|
raise error
|