style(database): Refactor repository name

database/__init__.py

@@ -1,3 +0,0 @@
-from .connection import get_connection, get_db_config
-
-__all__ = ['get_connection', 'get_db_config']

db/connection.py

@@ -1,4 +1,4 @@
-# database/connection.py
+# db/connection.py
 import psycopg2
 import os
 from contextlib import contextmanager
@@ -13,7 +13,7 @@ def get_db_config() -> dict:
     return {
         'host': os.getenv('DB_HOST', 'localhost'),
         'port': int(os.getenv('DB_PORT', 5432)),
-        'database': os.getenv('DB_NAME', 'warehouse_db'),
+        'db': os.getenv('DB_NAME', 'warehouse_db'),
         'user': os.getenv('DB_USER', 'postgres'),
         'password': os.getenv('DB_PASSWORD', '')
     }

db/repositories/ai_prediction_repository.py

@@ -1,6 +1,6 @@
 from typing import List, Optional
 from datetime import datetime, date
-from database.connection import get_connection
+from db.connection import get_connection
 from model.ai_prediction import AIPrediction
 
 

db/repositories/inventory_repository.py

@@ -1,7 +1,7 @@
-# database/repositories/inventory_repository.py
+# db/repositories/inventory_repository.py
 from typing import List, Optional, Tuple
 from datetime import datetime
-from database.connection import get_connection
+from db.connection import get_connection
 from model.inventory import InventoryRecord
 
 

db/repositories/product_repository.py

@@ -1,6 +1,6 @@
-# database/repositories/product_repository.py
+# db/repositories/product_repository.py
 from typing import List, Optional
-from database.connection import get_connection
+from db.connection import get_connection
 from model.product import Product
 
 

db/repositories/robot_repository.py

@@ -1,6 +1,6 @@
-# database/repositories/robot_repository.py
+# db/repositories/robot_repository.py
 from typing import List, Optional
-from database.connection import get_connection
+from db.connection import get_connection
 from model.robot import Robot
 
 class RobotRepository:

db/repositories/user_repository.py

@@ -1,6 +1,6 @@
 from typing import List, Optional
 from model.user import User
-from database.connection import get_connection
+from db.connection import get_connection
 
 class UserRepository:
     def get_all(self) -> List[User]: