fix(wsgi): Use hub.trace

untitaker · untitaker · commit de516eef7060 · 2019-06-11T18:52:31.000+02:00
diff --git a/sentry_sdk/integrations/wsgi.py b/sentry_sdk/integrations/wsgi.py
@@ -78,22 +78,17 @@ def __call__(self, environ, start_response):
         hub = Hub(Hub.current)
 
         with hub:
-            span = None
             with capture_internal_exceptions():
                 with hub.configure_scope() as scope:
                     scope.clear_breadcrumbs()
                     scope._name = "wsgi"
                     scope.add_event_processor(_make_wsgi_event_processor(environ))
-                    scope.span = span = Span.continue_from_environ(environ)
 
-            try:
-                rv = self.app(environ, start_response)
-            except BaseException:
-                reraise(*_capture_exception(hub))
-            finally:
-                if span is not None:
-                    span.finish()
-                    hub.finish_trace(span)
+            with hub.trace(Span.continue_from_environ(environ)):
+                try:
+                    rv = self.app(environ, start_response)
+                except BaseException:
+                    reraise(*_capture_exception(hub))
 
         return _ScopedResponse(hub, rv)
 
