.env.tmp

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

api/auth.py

@@ -0,0 +1,35 @@
+from flask import Blueprint, request, jsonify
+from model.user import User
+from db.repositories.user_repository import UserRepository  # FIXME: authenticate_user as get_user
+from utils.token import generateKey as getToken
+
+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 = UserRepository().authenticate_user(email, password)
+        if not user:
+            return "Wrong credentials", 400
+
+        token = getToken(user)
+
+        return jsonify({'token': token, 'user': {'id': user.id, 'name': user.name, 'role': user.role}})
+
+    else:
+        return "Request is not a json", 400

api/loginapi.py

@@ -1,13 +0,0 @@
-from flask import Blueprint, request
-
-from model import user
-
-loginBP =  Blueprint("loginapi", __name__)
-
-@loginBP.route('/api/login', methods = ['POST'])
-def login():
-    email = request.form['email']
-    password = request.form['password']
-    #if(isvalid(email, password)):
-    us = user.initialize(email, password)
-    return us.toJSON()

app.py

@@ -1,6 +1,11 @@
+from sys import exit
 from flask import Flask
-from api.loginapi import loginBP
+from api.auth import auth
+from utils.loadDotEnv import initializeENV
+
+if not initializeENV():
+    exit(-1)
 
 app = Flask(__name__)
 
-app.register_blueprint(loginBP)
+app.register_blueprint(auth, url_prefix='/api/auth')

db/repositories/ai_prediction_repository.py

@@ -1,11 +1,9 @@
 from typing import List, Optional
 from datetime import datetime, date
-import logging
+from loguru import logger
 from db.connection import get_connection
 from model.ai_prediction import AIPrediction
 
-logger = logging.getLogger(__name__)
-
 class AIPredictionsRepository:
     def get_all(self) -> List[AIPrediction]:
         try:
@@ -77,12 +75,14 @@ class AIPredictionsRepository:
             logger.error(f"Ошибка получения последних прогнозов по товарам: {e}")
             return []
 
-    def create_prediction(self,
+    def create_prediction(
+        self,
         product_id: str,
         prediction_date: date,
         days_until_stockout: int,
         recommended_order: int,
-                          confidence_score: float) -> Optional[int]:
+        confidence_score: float
+    ) -> Optional[int]:
         try:
             with get_connection() as conn:
                 with conn.cursor() as cur:

db/repositories/inventory_repository.py

@@ -1,12 +1,10 @@
 # db/repositories/inventory_repository.py
 from typing import List, Optional, Tuple
 from datetime import datetime
-import logging
+from loguru import logger
 from db.connection import get_connection
 from model.inventory import InventoryRecord
 
-logger = logging.getLogger(__name__)
-
 class InventoryRepository:
     def get_all(self) -> List[InventoryRecord]:
         try:
@@ -48,7 +46,8 @@ class InventoryRepository:
             logger.error(f"Ошибка получения записи инвентаризации {record_id}: {e}")
             return None
 
-    def create_record(self,
+    def create_record(
+        self,
         robot_id: str,
         product_id: str,
         quantity: int,
@@ -56,7 +55,8 @@ class InventoryRepository:
         row_number: int,
         shelf_number: int,
         status: str,
-                      scanned_at: datetime) -> Optional[int]:
+        scanned_at: datetime
+    ) -> Optional[int]:
         try:
             with get_connection() as conn:
                 with conn.cursor() as cur:

db/repositories/product_repository.py

@@ -1,10 +1,8 @@
 from typing import List, Optional
-import logging
+from loguru import logger
 from db.connection import get_connection
 from model.product import Product
 
-logger = logging.getLogger(__name__)
-
 class ProductRepository:
     def get_all(self) -> List[Product]:
         try:

db/repositories/robot_repository.py

@@ -1,10 +1,8 @@
 from typing import List, Optional
-import logging
+from loguru import logger
 from db.connection import get_connection
 from model.robot import Robot
 
-logger = logging.getLogger(__name__)
-
 class RobotRepository:
     def get_all(self) -> List[Robot]:
         try:
@@ -43,13 +41,15 @@ class RobotRepository:
             logger.error(f"Ошибка получения робота {robot_id}: {e}")
             return None
 
-    def update_robot(self,
+    def update_robot(
+        self,
         robot_id: str,
         status: str = None,
         battery_level: int = None,
         current_zone: str = None,
         current_row: int = None,
-                     current_shelf: int = None) -> bool:
+        current_shelf: int = None
+    ) -> bool:
         try:
             with get_connection() as conn:
                 with conn.cursor() as cur:
@@ -131,10 +131,12 @@ class RobotRepository:
             logger.error(f"Ошибка получения роботов в зоне {zone}: {e}")
             return []
 
-    def create_robot(self,
+    def create_robot(
+        self,
         robot_id: str,
         status: str = 'active',
-                     battery_level: int = 100) -> bool:
+        battery_level: int = 100
+    ) -> bool:
         try:
             with get_connection() as conn:
                 with conn.cursor() as cur:

db/repositories/user_repository.py

@@ -1,10 +1,8 @@
 from typing import List, Optional
-import logging
+from loguru import logger
 from model.user import User
 from db.connection import get_connection
 
-logger = logging.getLogger(__name__)
-
 class UserRepository:
     def get_all(self) -> List[User]:
         try:
@@ -57,11 +55,13 @@ class UserRepository:
             logger.error(f"Ошибка получения пользователя по email {email}: {e}")
             return None
 
-    def create_user(self,
+    def create_user(
+        self,
         email: str,
         password_hash: str,
         name: str,
-                    role: str) -> Optional[User]:
+        role: str
+    ) -> Optional[User]:
         try:
             with get_connection() as conn:
                 with conn.cursor() as cur:
@@ -166,6 +166,9 @@ class UserRepository:
             return False
 
     def authenticate_user(self, email: str, password_hash: str) -> Optional[User]:
+        if not self.user_exists(email):
+            return
+
         try:
             with get_connection() as conn:
                 with conn.cursor() as cur:
@@ -183,24 +186,6 @@ class UserRepository:
             logger.error(f"Ошибка аутентификации пользователя {email}: {e}")
             return None
 
-    def is_valid_authenticate(self, email: str, password_hash: str) -> bool:
-        try:
-            with get_connection() as conn:
-                with conn.cursor() as cur:
-                    cur.execute("""
-                        SELECT 1 FROM users 
-                        WHERE email = %s AND password_hash = %s
-                    """, (email, password_hash))
-                    is_valid = cur.fetchone() is not None
-                    if is_valid:
-                        logger.debug(f"Валидные учетные данные для пользователя {email}")
-                    else:
-                        logger.warning(f"Невалидные учетные данные для пользователя {email}")
-                    return is_valid
-        except Exception as e:
-            logger.error(f"Ошибка проверки учетных данных пользователя {email}: {e}")
-            return False
-
     def user_exists(self, email: str) -> bool:
         try:
             with get_connection() as conn:

dependencies.txt

@@ -1,4 +1,5 @@
 flask==3.1.2
 python-dotenv
-psycopg-binary
+psycopg2-binary
 pyjwt
+loguru

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

@@ -1,7 +1,18 @@
 import jwt
 import os
+from time import time
+from model.user import User
 
-def generateKey(email, passwd):
+def generateKey(user: User) -> dict:
     key = os.getenv('KEY')
-    encoded = jwt.encode({f"{email}": f"{passwd}"}, key, algorithm="HS256")
+    encoded = jwt.encode(
+        {
+            'id': user.id,
+            'name': user.name,
+            'role': user.role,
+            'iat': time()
+        },
+        key,
+        algorithm="HS256"
+    )
     return encoded