Fix issue with flask url handler returning an iterator.

If the iterator acts like doing some actual job then yield part of response, sentry sdk would not catch anything after the url handler itself ended.  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