From 12be29340ba87ef07dc94de5b52355767308549d Mon Sep 17 00:00:00 2001
From: Kirill <kremizov3@gmail.com>
Date: Fri, 31 Oct 2025 13:47:55 +0300
Subject: [PATCH] feat(api/ws.py): WebSoket

---
 api/ws.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 api/ws.py

diff --git a/api/ws.py b/api/ws.py
new file mode 100644
index 0000000..a6c87a3
--- /dev/null
+++ b/api/ws.py
@@ -0,0 +1,26 @@
+from app import app
+from flask_socketio import SocketIO, emit
+from loguru import logger as log
+#from module.db.repositories.robot_reposytory import update_robot get_by_id get_all
+#from module.db.repositories.inventory_repository import get_all as getRecords
+ws = socketIO(app)
+
+@ws.on('connect')
+def is_connect():
+    log.info('client is connected')
+
+@ws.on('disconnect')
+def is_disconnect():
+    log.info('client is disconnected')
+
+@ws.on('robot_update')
+def robotUpdate():
+    robots = get_all()
+    emit('response', jsonify({"type": "robot_update", "data": {robots.toJSON()}})
+
+@ws.on('inventory_alert')
+def robotUpdate():
+    records = getRecords()
+    emit('response', jsonify({"type": "inventory_alert", "data": {records.toJSON()}})
+    
+