From 9ecb6a83ba26b2c17cb51c32c114bc2f8edfb5df Mon Sep 17 00:00:00 2001
From: Kita Trofimov <144846094+NiTro005@users.noreply.github.com>
Date: Sun, 26 Oct 2025 19:41:11 +0300
Subject: [PATCH] style(database): Fixed the issue with long function
 declarations

---
 db/repositories/ai_prediction_repository.py | 11 ++++++++---
 db/repositories/inventory_repository.py     | 13 +++++++++++--
 db/repositories/robot_repository.py         | 19 +++++++++++++++----
 db/repositories/user_repository.py          |  8 +++++++-
 4 files changed, 41 insertions(+), 10 deletions(-)

diff --git a/db/repositories/ai_prediction_repository.py b/db/repositories/ai_prediction_repository.py
index 8c6251d..acc3e7f 100644
--- a/db/repositories/ai_prediction_repository.py
+++ b/db/repositories/ai_prediction_repository.py
@@ -77,9 +77,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:
diff --git a/db/repositories/inventory_repository.py b/db/repositories/inventory_repository.py
index 5bcbc31..0d5611d 100644
--- a/db/repositories/inventory_repository.py
+++ b/db/repositories/inventory_repository.py
@@ -48,8 +48,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:
diff --git a/db/repositories/robot_repository.py b/db/repositories/robot_repository.py
index 74879b9..7ebd872 100644
--- a/db/repositories/robot_repository.py
+++ b/db/repositories/robot_repository.py
@@ -1,4 +1,3 @@
-# db/repositories/robot_repository.py
 from typing import List, Optional
 import logging
 from db.connection import get_connection
@@ -44,8 +43,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:
@@ -127,7 +133,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:
diff --git a/db/repositories/user_repository.py b/db/repositories/user_repository.py
index d3a2a11..4c7413b 100644
--- a/db/repositories/user_repository.py
+++ b/db/repositories/user_repository.py
@@ -57,7 +57,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: