Fix issue with flask url handler returning an iterator.

If the iterator does some actual job then yield's part of response, sentry sdk would not catch anything after the url handler itself returned, and will miss everything being done in the iterator. This fixes the issue.

Signed-off-by: Wei Mingzhi <[email protected]>

Author: weimzh

sentry_sdk/integrations/wsgi.py

@@ -134,12 +134,17 @@ def __call__(self, environ, start_response):
                                     _sentry_start_response, start_response, transaction
                                 ),
                             )
+                            if hasattr(response, "__iter__"):
+                                scoped_response = _ScopedResponse(scope, response)
+                                for it in scoped_response:
+                                    yield it
                         except BaseException:
                             reraise(*_capture_exception())
         finally:
             _wsgi_middleware_applied.set(False)
 
-        return _ScopedResponse(scope, response)
+        if not hasattr(response, "__iter__"):
+            return _ScopedResponse(scope, response)
 
 
 def _sentry_start_response(  # type: ignore