feat(api/ws.py): WebSoket

Added Libs for work with webSockets

.env.tmp

@@ -0,0 +1,2 @@
+KEY=  # Key for JWT token
+POSTGRES_URL=postgresql://

api/auth.py

@@ -0,0 +1,28 @@
+from flask import Blueprint, request, jsonify
+from model.user import User
+
+auth = Blueprint("auth", __name__)
+
+
+@auth.route('/login', methods = ['POST'])
+def login():
+    if request.is_json:
+        req = request.json
+
+        email = req.get('email')
+        password = req.get('password')
+
+        if not email or not password:
+            return "Request must have email and password", 400
+
+        if len(email.strip()) < 4 or '@' not in email or '.' not in email:
+            return "Email is incorrect", 400
+
+        if len(password.strip()) < 8:
+            return "Password is too short", 400
+
+        user = User(email, password)
+        return jsonify(user.toJson())
+
+    else:
+        return "Request is not a json", 400

api/robots.py

@@ -0,0 +1,24 @@
+from flask import Blueprint, request, jsonify
+from model.user import User
+#from module.db.repositories.robot_reposytory import update_robot get_by_id
+from loguru import logger as log
+robots = Blueprint("robots", __name__)
+
+robots.route('/data', methods = ['POST'])
+def data():
+    if request.headers.get("Authorization"):
+        if request.is_json:
+            req = request.json
+
+            id = req.get('robot_id')
+            tms = req.get('timestamp')
+            loc = req.get('location')
+            scanRes = req.get('scan_results')
+            battery = req.get('battery_level')
+
+            if update_robot(id, "received", battery, loc["zone"], loc["row"], loc["self"]):
+                res = {"status": "received", "message_id": "123"} #for that moment i don`t now what is message
+                return jsonify(res)
+            else:
+                log.error('failed to update robot data')
+                return("Server error")

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

app.py

@@ -1,4 +1,15 @@
+from sys import exit
 from flask import Flask
+from api.auth import auth
+from utils.loadDotEnv import initializeENV
+from utils.PostgressConnect import PSQLConnect, PSQLCursor
+
+if not initializeENV():
+    exit(-1)
+
+#conn = PSQLConnect()
+#cur = PSQLCursor(conn)
 
 app = Flask(__name__)
 
+app.register_blueprint(auth, url_prefix='/api/auth')

dependencies.txt

@@ -1,3 +1,6 @@
 flask==3.1.2
+flask-socketio
 python-dotenv
-psycopg-binary==3.2.10
+psycopg2-binary
+pyjwt
+loguru

model/user.py

@@ -0,0 +1,20 @@
+from dataclasses import dataclass
+import json
+from utils.token import generateKey
+
+@dataclass
+class User:
+    id: int
+    name: str
+    role: str
+    token: str
+
+    def __init__(self, email: str, passwd: str):
+        #us = getUsModel() #возвращает словарь
+        self.id = 1#us['id']
+        self.name = 'Bob'#us['name']
+        self.role = 'Backend'#us['role']
+        self.token = generateKey(email, passwd)
+
+    def toJson(self):
+        return {"user": {"id": self.id, "name": self.name, "role": self.role}, "token": self.token}

utils/InitalizeDB.py

@@ -1,79 +0,0 @@
-import psycopg
-from PostgressConnect import PSQLConnect, PSQLCursor
-from loadDotEnv import initializeENV
-
-state = initializeENV()
-conn = PSQLConnect()
-cur = PSQLCursor(conn)
-
-cur.execute('''
-            -- Пользователи системы
-CREATE TABLE users (
-    id SERIAL PRIMARY KEY,
-    email VARCHAR(255) UNIQUE NOT NULL,
-    password_hash VARCHAR(255) NOT NULL,
-    name VARCHAR(255) NOT NULL,
-    role VARCHAR(50) NOT NULL, -- 'operator', 'admin', 'viewer'
-    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
-);
-
--- Роботы
-CREATE TABLE robots (
-    id VARCHAR(50) PRIMARY KEY, -- 'RB-001'
-    status VARCHAR(50) DEFAULT 'active',
-    battery_level INTEGER,
-    last_update TIMESTAMP,
-    current_zone VARCHAR(10),
-    current_row INTEGER,
-    current_shelf INTEGER
-);
-
-
--- Товары
-
-CREATE TABLE products (
-    id VARCHAR(50) PRIMARY KEY, -- 'TEL-4567'
-    name VARCHAR(255) NOT NULL,
-    category VARCHAR(100),
-    min_stock INTEGER DEFAULT 10,
-    optimal_stock INTEGER DEFAULT 100
-);
-
-
--- История инвентаризации
-CREATE TABLE inventory_history (
-    id SERIAL PRIMARY KEY,
-    robot_id VARCHAR(50) REFERENCES robots(id),
-    product_id VARCHAR(50) REFERENCES productsuid),
-    quantity INTEGER NOT NULL,
-    zone VARCHAR(10) NOT NULL,
-    row_number INTEGER,
-    shelf_number INTEGER,
-    status VARCHAR(50), -- 'OK', 'LOW_STOCK', 'CRITICAL'
-    scanned_at TIMESTAMP NOT NULL,
-    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
-);
-
-
--- Прогнозы ИИ
-CREATE TABLE ai_predictions (
-    id SERIAL PRIMARY KEY,
-    product_id VARCHAR(50) REFERENCES products(id),
-    prediction_date DATE NOT NULL,
-    days_until_stockout INTEGER,
-    recommended_order INTEGER,
-    confidence_score DECIMAL(3,2),
-    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
-);
-
--- Индексы для оптимизации
-REATE INDEX idx_inventory_scanned ON inventory_history(scanned_at DESC);
-CREATE INDEX idx_inventory_product ON inventory_history(product_id);
-CREATE INDEX idx_inventory_zone ON inventory_history(zone);
-            ''')
-conn.commit()
-
-cur.close()
-conn.close()
-
-print("DB is Initialized")

utils/PostgressConnect.py

@@ -1,10 +1,10 @@
-import psycopg
+import psycopg2
 import os
 
 def PSQLConnect():
-    conn = psycopg.connect(os.getenv('POSTDRESS_CONNECTION'))
+    conn = psycopg2.connect(os.getenv('POSTGRES_URL'))
     return conn
 
 def PSQLCursor(conn):
-    cur =  conn.cursor()
+    cur = conn.cursor()
     return cur

utils/loadDotEnv.py

@@ -1,12 +1,14 @@
 import os
 from dotenv import load_dotenv
+from loguru import logger as log
 
-def initializeENV():
+DOTENV_PATH = '.env'
-    dotenv_path = '../.env'
+
-    if os.path.exists(dotenv_path):
+def initializeENV() -> bool:
-        load_dotenv(dotenv_path)
+    if os.path.exists(DOTENV_PATH):
-        print('.env is loaded')
+        load_dotenv(DOTENV_PATH)
-        return 1
+        log.info('.env is loaded')
+        return True
     else:
-        print('.env isn`t loaded')
+        log.error('.env isn`t loaded')
-        return 0
+        return False

utils/token.py

@@ -0,0 +1,8 @@
+import jwt
+import os
+from time import time
+
+def generateKey(email, passwd):
+    key = os.getenv('KEY')
+    encoded = jwt.encode({email: passwd, 'iat': time()}, key, algorithm="HS256")
+    return encoded