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()}}) + +