Prevent AssertionError with Werkzeug when connection ends

miguelgrinberg · Mar 11, 2022 · f6142a6 · f6142a6
1 parent 0d9d0b5
commit f6142a6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/flask_sock/__init__.py b/src/flask_sock/__init__.py
@@ -78,6 +78,8 @@ def __call__(self, *args, **kwargs):
                             return ALREADY_HANDLED
                         elif ws.mode == 'gunicorn':
                             raise StopIteration()
+                        elif ws.mode == 'werkzeug':
+                            raise ConnectionError()
                         else:
                             return []