fix: avoid missing websocket attribute if werkzeug < v1 (#407)

Signed-off-by: Keming <kemingy94@gmail.com>
0b01001001 · Mar 4, 2025 · f4b8306 · f4b8306
1 parent 68585d2
commit f4b8306
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/spectree/plugins/flask_plugin.py b/spectree/plugins/flask_plugin.py
@@ -28,7 +28,7 @@ def find_routes(self):
                 continue
             if rule.endpoint.startswith("openapi"):
                 continue
-            if rule.websocket:
+            if getattr(rule, "websocket", False):
                 continue
             if (
                 self.blueprint_state

diff --git a/spectree/plugins/quart_plugin.py b/spectree/plugins/quart_plugin.py
@@ -31,7 +31,7 @@ def find_routes(self):
                 continue
             if rule.endpoint.startswith("openapi"):
                 continue
-            if rule.websocket:
+            if getattr(rule, "websocket", False):
                 continue
             if (
                 self.blueprint_state