From 6b558086c1532e0b4499c30eaf7de7f7a95518c1 Mon Sep 17 00:00:00 2001
From: Edward Betts <edward@4angle.com>
Date: Sat, 4 May 2024 11:44:44 +0300
Subject: [PATCH] Fix error handling

---
 web_view.py | 46 ++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 44 insertions(+), 2 deletions(-)

diff --git a/web_view.py b/web_view.py
index 0b25724..03ab0f3 100755
--- a/web_view.py
+++ b/web_view.py
@@ -2,8 +2,11 @@
 
 """Views for the web app."""
 
+import inspect
 import json
 import re
+import sys
+import traceback
 import typing
 from time import sleep, time
 
@@ -34,8 +37,6 @@ from matcher import (
 )
 from matcher.data import property_map
 
-# from werkzeug.debug.tbtools import get_current_traceback
-
 StrDict = dict[str, typing.Any]
 
 srid = 4326
@@ -75,6 +76,47 @@ def dict_repr_values(d):
     return {key: repr(value) for key, value in d.items()}
 
 
+@app.errorhandler(werkzeug.exceptions.InternalServerError)
+def exception_handler(
+    e: werkzeug.exceptions.InternalServerError,
+) -> tuple[str | werkzeug.wrappers.Response, int]:
+    """Handle exception."""
+    exec_type, exc_value, current_traceback = sys.exc_info()
+    assert exc_value
+    tb = werkzeug.debug.tbtools.DebugTraceback(exc_value)
+
+    # summary = tb.render_traceback_html(include_title=False)
+
+    last_frame = list(traceback.walk_tb(current_traceback))[-1][0]
+    last_frame_args = inspect.getargs(last_frame.f_code)
+    if flask.request.path.startswith("/api/"):
+        return (
+            cors_jsonify(
+                {
+                    "success": False,
+                    "error": tb._te.exc_type.__name__,
+                    "traceback": tb.render_traceback_text(),
+                    "locals": dict_repr_values(last_frame.f_locals),
+                    "last_function": {
+                        "name": last_frame.f_code.co_name,
+                        "args": repr(last_frame_args),
+                    },
+                }
+            ),
+            500,
+        )
+
+    return (
+        flask.render_template(
+            "show_error.html",
+            tb=tb,
+            last_frame=last_frame,
+            last_frame_args=last_frame_args,
+        ),
+        500,
+    )
+
+
 def cors_jsonify(*args, **kwargs) -> flask.Response:
     """Add CORS header to JSON."""
     response = flask.jsonify(*args, **kwargs)