From 98d11b62a660dc89a61d092b9287d9b749590d65 Mon Sep 17 00:00:00 2001
From: Edward Betts <edward@4angle.com>
Date: Sun, 25 Feb 2024 21:42:56 +0000
Subject: [PATCH] Record errors in the database

Closes: #12
---
 geocode/errors_to_db.py | 45 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 geocode/errors_to_db.py

diff --git a/geocode/errors_to_db.py b/geocode/errors_to_db.py
new file mode 100644
index 0000000..96069ed
--- /dev/null
+++ b/geocode/errors_to_db.py
@@ -0,0 +1,45 @@
+"""Save errors to the database."""
+
+import logging
+import traceback
+
+import flask
+
+from .database import session
+from .model import ErrorLog
+
+
+class DatabaseLoggingHandler(logging.Handler):
+    """A logging handler that logs errors to a database."""
+
+    def emit(self, record: logging.LogRecord) -> None:
+        """Save log reord to database."""
+        trace = None
+        if record.exc_info:
+            trace = traceback.format_exception(*record.exc_info)
+        # request: flask.Request = record.request  # type: ignore
+        request = flask.request
+        error_log = ErrorLog(
+            error_type=record.levelname,
+            error_message=record.getMessage(),
+            traceback="".join(trace) if trace else "No traceback available",
+            context_info={
+                "url": request.url,
+                "args": dict(request.args),
+                "endpoint": request.endpoint,
+                "pathname": record.pathname,
+                "lineno": record.lineno,
+                "module": record.module,
+                "funcName": record.funcName,
+                "remote_addr": request.remote_addr,
+            },
+        )
+        session.add(error_log)
+        session.commit()
+
+
+def setup_error_recorder(app: flask.Flask) -> None:
+    """Save errors to the database."""
+    db_handler = DatabaseLoggingHandler()
+    db_handler.setLevel(logging.ERROR)
+    app.logger.addHandler(db_handler)