Automatic exception handling for flask apps

Author: bhogayatakb

middleware/__init__.py

@@ -58,3 +58,28 @@ def _uninstrument(self, **kwargs):
 # Load the instrumentor
 ExceptionInstrumentor().instrument()
 
+# Automatic exception handling for flask
+from flask import Flask, request
+from flask.signals import got_request_exception, appcontext_pushed
+import traceback
+
+def _capture_exception(sender, exception, **extra):
+    exc_type, exc_value, exc_traceback = sys.exc_info()  # Get exception details
+    if exc_type and exc_value and exc_traceback:
+        record_exception(exc_type, exc_value, exc_traceback)
+    else:
+        print("Unable to capture exception details.")
+
+def try_register_flask_handler(app: Flask):
+    """Registers the exception handler using Flask signals."""
+    got_request_exception.connect(_capture_exception, app)
+    print("✅ Flask error handler registered via signal.")
+
+def _auto_register(sender, **extra):
+    """Automatically registers the handler when a Flask app context is pushed."""
+    try_register_flask_handler(sender)
+
+# Connect to Flask's appcontext_pushed to register automatically
+appcontext_pushed.connect(_auto_register)
+
+

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "middleware-io"
-version = "2.1.2rc4"
+version = "2.1.2rc5"
 requires-python = ">=3.8"
 description = "Middleware's APM tool enables Python developers to effortlessly monitor their applications, gathering distributed tracing, metrics, logs, and profiling data for valuable insights and performance optimization."
 authors = [{ name = "middleware-dev" }]
@@ -21,6 +21,7 @@ keywords = [
 ]
 dependencies =[
     "distro==1.9.0",
+    "flask==2.3.2",
     "opentelemetry-api==1.27.0",
     "opentelemetry-exporter-otlp-proto-grpc==1.27.0",
     "opentelemetry-instrumentation==0.48b0",