Do not show simple-websocket install prompt if it is already installed

miguelgrinberg · May 22, 2021 · 03a52c8 · 03a52c8
1 parent 22a7ea4
commit 03a52c8
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/flask_socketio/__init__.py b/flask_socketio/__init__.py
@@ -582,9 +582,12 @@ def run(self, app, host=None, port=None, **kwargs):
                 self.sockio_mw.wsgi_app, evalex=True)
 
         if self.server.eio.async_mode == 'threading':
-            from werkzeug._internal import _log
-            _log('warning', 'WebSocket transport not available. Install '
-                            'simple-websocket for improved performance.')
+            try:
+                import simple_websocket
+            except ImportError:
+                from werkzeug._internal import _log
+                _log('warning', 'WebSocket transport not available. Install '
+                                'simple-websocket for improved performance.')
             app.run(host=host, port=port, threaded=True,
                     use_reloader=use_reloader, **kwargs)
         elif self.server.eio.async_mode == 'eventlet':