From cc8f30eba2485d15522b6f9971182ed41185e160 Mon Sep 17 00:00:00 2001 From: Jesus E Date: Mon, 19 Jun 2023 21:23:25 -0400 Subject: Relax error and send error_code to template --- youtube/__init__.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) (limited to 'youtube/__init__.py') diff --git a/youtube/__init__.py b/youtube/__init__.py index 3604b21..b0e8761 100644 --- a/youtube/__init__.py +++ b/youtube/__init__.py @@ -115,7 +115,18 @@ def error_page(e): error_message=exc_info()[1].error_message, slim=slim ), 502) - return flask.render_template('error.html', traceback=traceback.format_exc(), slim=slim), 500 + elif (exc_info()[0] == util.FetchError + and exc_info()[1].code == '404' + ): + error_message = ('Error: The page you are looking for isn\'t here. ¯\_(ツ)_/¯') + return flask.render_template('error.html', + error_code=exc_info()[1].code, + error_message=error_message, + slim=slim), 404 + return flask.render_template('error.html', traceback=traceback.format_exc(), + error_code=exc_info()[1].code, + slim=slim), 500 + # return flask.render_template('error.html', traceback=traceback.format_exc(), slim=slim), 500 font_choices = { -- cgit v1.2.3