Manually close transaction in flask

Author: antonpirker

sentry_sdk/integrations/flask.py

@@ -36,6 +36,7 @@
         before_render_template,
         got_request_exception,
         request_started,
+        request_finished,
     )
     from markupsafe import Markup
 except ImportError:
@@ -82,6 +83,7 @@ def setup_once():
 
         before_render_template.connect(_add_sentry_trace)
         request_started.connect(_request_started)
+        request_finished.connect(_request_finished)
         got_request_exception.connect(_capture_exception)
 
         old_app = Flask.__call__
@@ -152,6 +154,11 @@ def _request_started(app, **kwargs):
     scope.add_event_processor(evt_processor)
 
 
+def _request_finished(sender, response, **kwargs):
+    # Manually close the transaction because Bottle does not call `close()` on the WSGI response
+    sentry_sdk.get_current_scope().transaction.__exit__(None, None, None)
+
+
 class FlaskRequestExtractor(RequestExtractor):
     def env(self):
         # type: () -> Dict[str, str]