smileos

src/errors.py

@@ -1,6 +1,10 @@
-from flask import Blueprint, render_template
+from flask import Blueprint, render_template, request
 from werkzeug.exceptions import HTTPException
-from os import getenv as env
+
+try:
+    from src.name import get_name
+except ImportError:
+    from name import get_name
 
 
 bp = Blueprint("errors", __name__)
@@ -9,19 +13,19 @@ bp = Blueprint("errors", __name__)
 @bp.route('/500')
 @bp.app_errorhandler(500)
 def internal_server_error(error:HTTPException=None):
-    return render_template('errors/500.html', error=error), 500
+    return render_template('errors/500.html', error=error, name=get_name(request)), 500
 
 
 @bp.route('/404')
 @bp.app_errorhandler(404)
 def not_found(error:HTTPException=None):
-    return render_template('errors/404.html', error=error), 404
+    return render_template('errors/404.html', error=error, name=get_name(request)), 404
 
 
 @bp.route('/400')
 @bp.app_errorhandler(400)
 def bad_request(error:HTTPException=None):
-    return render_template('errors/400.html', error=error), 400
+    return render_template('errors/400.html', error=error, name=get_name(request)), 400
 
 
 @bp.route('/idk')
@@ -35,5 +39,6 @@ def idk(error:HTTPException=None):
         'errors/error.html',
         code = error.code,
         description = error.description,
-        name = error.name
+        err_name = error.name,
+        name=get_name(request)
     ), error.code