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:
@@ -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:
|
||||
|
||||
@@ -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,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:
|
||||
|
||||
@@ -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:
|
||||
|
||||
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user