# 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

@@ -0,0 +1,3 @@
+from .connection import get_connection, get_db_config
+

@@ -0,0 +9,4 @@
+initializeENV()
+
+
+def get_db_config() -> dict:

@@ -0,0 +25,4 @@
+    try:
+        config = get_db_config()
+        conn = psycopg2.connect(**config)
+        print("Подключение к БД установлено")

@@ -0,0 +28,4 @@
+        print("Подключение к БД установлено")
+        yield conn
+    except psycopg2.OperationalError as e:
+        print(f"Ошибка подключения к БД: {e}")

@@ -0,0 +31,4 @@
+        print(f"Ошибка подключения к БД: {e}")
+        raise
+    except Exception as e:
+        print(f"Неожиданная ошибка: {e}")

@@ -0,0 +38,4 @@
+    finally:
+        if conn:
+            conn.close()
+            print("БД закрыта")

@@ -0,0 +47,4 @@
+            with conn.cursor() as cur:
+                cur.execute("SELECT version();")
+                version = cur.fetchone()
+                print(f" Версия PostgreSQL: {version[0]}")

@@ -0,0 +50,4 @@
+                print(f" Версия PostgreSQL: {version[0]}")
+                return True
+    except Exception as e:
+        print(f"Тест подключения бд провален: {e}")

@@ -0,0 +22,4 @@
+                        ) for row in cur.fetchall()
+                    ]
+        except Exception as e:
+            print(f"Ошибка получения прогнозов: {e}")

@@ -0,0 +33,4 @@
+                    row = cur.fetchone()
+                    return AIPrediction(*row) if row else None
+        except Exception as e:
+            print(f"Ошибка получения прогноза {prediction_id}: {e}")

@@ -0,0 +48,4 @@
+                    """, (product_id, limit))
+                    return [AIPrediction(*row) for row in cur.fetchall()]
+        except Exception as e:
+            print(f"Ошибка получения прогноза по товару {product_id}: {e}")

@@ -0,0 +62,4 @@
+                    """)
+                    return [AIPrediction(*row) for row in cur.fetchall()]
+        except Exception as e:
+            print(f"Ошибка получения последних прогнозов по товарам: {e}")

@@ -0,0 +82,4 @@
+                    conn.commit()
+                    return prediction_id
+        except Exception as e:
+            print(f"Ошибка создания прогноза: {e}")

@@ -0,0 +98,4 @@
+                    conn.commit()
+                    return deleted_count
+        except Exception as e:
+            print(f"Ошибка удаления старых прогнозов: {e}")

@@ -0,0 +27,4 @@
+                        ) for row in cur.fetchall()
+                    ]
+        except Exception as e:
+            print(f"Ошибка получения истории инвентаризации: {e}")

@@ -0,0 +38,4 @@
+                    row = cur.fetchone()
+                    return InventoryRecord(*row) if row else None
+        except Exception as e:
+            print(f"Ошибка получения записи инвентаризации {record_id}: {e}")

@@ -0,0 +57,4 @@
+                    conn.commit()
+                    return record_id
+        except Exception as e:
+            print(f"Ошибка создания записи инвентаризации: {e}")

@@ -0,0 +71,4 @@
+                    """)
+                    return [InventoryRecord(*row) for row in cur.fetchall()]
+        except Exception as e:
+            print(f"Ошибка получения последней записи инвентаризации по каждому товару: {e}")

@@ -20,3 +15,2 @@
+        return self.role == 'admin'
 
-    def toJSON(self):
-        return json.dumps({"token": f'{self.token}', "user": {"id": self.id, "role": f'{self.role}', "name": f'{self.name}'}})

@@ -20,3 +16,2 @@
 
-    def toJSON(self):
-        return json.dumps({"token": f'{self.token}', "user": {"id": self.id, "role": f'{self.role}', "name": f'{self.name}'}})
+    def is_operator(self) -> bool:

@@ -23,0 +17,4 @@
+    def is_operator(self) -> bool:
+        return self.role == 'operator'
+
+    def is_viewer(self) -> bool:

@@ -0,0 +34,4 @@
+        print(f"Неожиданная ошибка: {e}")
+        if conn:
+            conn.rollback()
+        raise

@@ -0,0 +9,4 @@
+        try:
+            with get_connection() as conn:
+                with conn.cursor() as cur:
+                    cur.execute("SELECT * FROM ai_predictions ORDER BY prediction_date DESC, product_id")

@@ -0,0 +42,4 @@
+            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]:

@@ -0,0 +35,4 @@
+            print(f"Ошибка получения роботов {robot_id}: {e}")
+            return None
+
+    def update_robot(self, robot_id: str, status: str = None, battery_level: int = None,

@@ -0,0 +14,4 @@
+                        password_hash=row[2],
+                        name=row[3],
+                        role=row[4],
+                        created_at=row[5]

@@ -0,0 +38,4 @@
+def test_connection() -> bool:
+    try:
+        with get_connection() as conn:
+            cur = PSQLCursor(conn)