From 401205aea2274ba9a522e7e41910fef135ab7fcb 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 |  7 +++++--
 db/repositories/inventory_repository.py     | 11 +++++++++--
 db/repositories/robot_repository.py         | 15 +++++++++++----
 db/repositories/user_repository.py          |  6 +++++-
 4 files changed, 30 insertions(+), 9 deletions(-)

diff --git a/db/repositories/ai_prediction_repository.py b/db/repositories/ai_prediction_repository.py
index afa25b0..66a60ec 100644
--- a/db/repositories/ai_prediction_repository.py
+++ b/db/repositories/ai_prediction_repository.py
@@ -77,8 +77,11 @@ class AIPredictionsRepository:
             logger.error(f"Ошибка получения последних прогнозов по товарам: {e}")
             return []
 
-    def create_prediction(self, product_id: str, prediction_date: date,
-                          days_until_stockout: int, recommended_order: 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:
diff --git a/db/repositories/inventory_repository.py b/db/repositories/inventory_repository.py
index 365f48f..a3c3536 100644
--- a/db/repositories/inventory_repository.py
+++ b/db/repositories/inventory_repository.py
@@ -48,8 +48,15 @@ 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 c460562..4279355 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,13 @@ 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 +131,10 @@ 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 31f527e..17ac782 100644
--- a/db/repositories/user_repository.py
+++ b/db/repositories/user_repository.py
@@ -57,7 +57,11 @@ 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: