From c26b9acc8d51bb5efe4e50114a6c723552e27a5f Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Tue, 28 Nov 2023 16:51:31 +0300
Subject: Handle generic MySQL errors at the top-level.

---
 qc_app/errors.py | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'qc_app/errors.py')

diff --git a/qc_app/errors.py b/qc_app/errors.py
index 0da6b3c..902e7e0 100644
--- a/qc_app/errors.py
+++ b/qc_app/errors.py
@@ -2,6 +2,7 @@
 import logging
 import traceback
 
+import MySQLdb as mdb
 from flask import Flask, render_template
 
 def handle_general_exception(exc: Exception):
@@ -9,6 +10,12 @@ def handle_general_exception(exc: Exception):
     logging.error("Error: Generic unhandled exception!!\n%s", trace)
     return render_template("unhandled_exception.html", trace=trace)
 
+def handle_general_MySQLdb_errors(exc: mdb.MySQLError):
+    trace = traceback.format_exc()
+    logging.error("MySQLError: Generic MySQL Error!!\n%s", trace)
+    return render_template("unhandled_exception.html", trace=trace)
+
 def register_error_handlers(app: Flask):
     """Register top-level error/exception handlers."""
     app.register_error_handler(Exception, handle_general_exception)
+    app.register_error_handler(mdb.MySQLError, handle_general_exception)
-- 
cgit v1.2.3