Integrate websockets into the Flask webapp

2 files changed, 18 insertions, 0 deletions

diff --git a/pysite/views/ws/__init__.py b/pysite/views/ws/__init__.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/pysite/views/ws/__init__.py
diff --git a/pysite/views/ws/echo.py b/pysite/views/ws/echo.py
new file mode 100644
index 00000000..135adfcf
--- /dev/null
+++ b/pysite/views/ws/echo.py
@@ -0,0 +1,18 @@
+# coding=utf-8
+from pysite.websockets import Websocket
+
+
+class EchoWebsocket(Websocket):
+    path = "/echo"
+    name = "ws_echo"
+
+    def on_open(self):
+        print("Echo | WS opened.")
+        self.send("Hey, welcome!")
+
+    def on_message(self, message):
+        print(f"Echo | Message: {message}")
+        self.send(message)
+
+    def on_close(self):
+        print("Echo | WS closed.")