1 files changed, 11 insertions, 3 deletions
diff --git a/pysite/views/ws/echo.py b/pysite/views/ws/echo.py
index 06dffdc4..a10f9b57 100644
--- a/pysite/views/ws/echo.py
+++ b/pysite/views/ws/echo.py
@@ -1,4 +1,8 @@
 # coding=utf-8
+import logging
+
+from geventwebsocket.websocket import WebSocket
+
 from pysite.websockets import WS
 
 
@@ -6,13 +10,17 @@ class EchoWebsocket(WS):
     path = "/echo"
     name = "ws_echo"
 
+    def __init__(self, socket: WebSocket):
+        super().__init__(socket)
+        self.log = logging.getLogger()
+
     def on_open(self):
-        print("Echo | WS opened.")
+        self.log.debug("Echo | WS opened.")
         self.send("Hey, welcome!")
 
     def on_message(self, message):
-        print(f"Echo | Message: {message}")
+        self.log.debug(f"Echo | Message: {message}")
         self.send(message)
 
     def on_close(self):
-        print("Echo | WS closed.")
+        self.log.debug("Echo | WS closed.")