From 3c3f65f1c26dbed7ac3d55799884769adfb728aa Mon Sep 17 00:00:00 2001
From: Kovid Goyal <kovid@kovidgoyal.net>
Date: Wed, 28 Oct 2015 14:48:32 +0530
Subject: [PATCH] ...

---
 src/calibre/srv/web_socket.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/calibre/srv/web_socket.py b/src/calibre/srv/web_socket.py
index 8bf10a38b0..fcea75c533 100644
--- a/src/calibre/srv/web_socket.py
+++ b/src/calibre/srv/web_socket.py
@@ -550,8 +550,13 @@ class EchoHandler(object):
     def handle_websocket_close(self, connection_id):
         self.ws_connections.pop(connection_id, None)
 
-if __name__ == '__main__':
+def run_echo_server():
     s = ServerLoop(create_http_handler(websocket_handler=EchoHandler()))
     with HandleInterrupt(s.wakeup):
         s.serve_forever()
+
+if __name__ == '__main__':
+    # import cProfile
+    # cProfile.runctx('r()', {'r':run_echo_server}, {}, filename='stats.profile')
+    run_echo_server()
 # }}}