Merge remote-tracking branch 'origin/feat/db-init' into feat/db-init

# Conflicts:
#	api/loginapi.py
#	db/connection.py
#	db/repositories/ai_prediction_repository.py
#	db/repositories/inventory_repository.py
#	db/repositories/product_repository.py
#	db/repositories/robot_repository.py
#	db/repositories/user_repository.py
This commit is contained in:
Kita Trofimov
2025-10-27 23:04:10 +03:00
12 changed files with 118 additions and 90 deletions
+9 -9
View File
@@ -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,
product_id: str,
prediction_date: date,
days_until_stockout: int,
recommended_order: int,
confidence_score: float) -> Optional[int]:
def create_prediction(
self,
product_id: str,
prediction_date: date,
days_until_stockout: int,
recommended_order: int,
confidence_score: float
) -> Optional[int]:
try:
with get_connection() as conn:
with conn.cursor() as cur:
+12 -12
View File
@@ -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,15 +46,17 @@ class InventoryRepository:
logger.error(f"Ошибка получения записи инвентаризации {record_id}: {e}")
return None
def create_record(self,
robot_id: str,
product_id: str,
quantity: int,
zone: str,
row_number: int,
shelf_number: int,
status: str,
scanned_at: datetime) -> Optional[int]:
def create_record(
self,
robot_id: str,
product_id: str,
quantity: int,
zone: str,
row_number: int,
shelf_number: int,
status: str,
scanned_at: datetime
) -> Optional[int]:
try:
with get_connection() as conn:
with conn.cursor() as cur:
+1 -3
View File
@@ -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:
+16 -14
View File
@@ -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,
robot_id: str,
status: str = None,
battery_level: int = None,
current_zone: str = None,
current_row: int = None,
current_shelf: int = None) -> bool:
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:
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,
robot_id: str,
status: str = 'active',
battery_level: int = 100) -> bool:
def create_robot(
self,
robot_id: str,
status: str = 'active',
battery_level: int = 100
) -> bool:
try:
with get_connection() as conn:
with conn.cursor() as cur:
+11 -26
View File
@@ -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,
email: str,
password_hash: str,
name: str,
role: str) -> Optional[User]:
def create_user(
self,
email: str,
password_hash: str,
name: str,
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: