From 33c3b81f6ea7c82c3441b03a164abb8ef5aa9308 Mon Sep 17 00:00:00 2001 From: Kita Trofimov <144846094+NiTro005@users.noreply.github.com> Date: Sun, 26 Oct 2025 18:52:54 +0300 Subject: [PATCH] style(database): Refactor repository name --- database/__init__.py | 3 --- {database => db}/connection.py | 4 ++-- {database => db}/repositories/__init__.py | 0 {database => db}/repositories/ai_prediction_repository.py | 2 +- {database => db}/repositories/inventory_repository.py | 4 ++-- {database => db}/repositories/product_repository.py | 4 ++-- {database => db}/repositories/robot_repository.py | 4 ++-- {database => db}/repositories/user_repository.py | 2 +- 8 files changed, 10 insertions(+), 13 deletions(-) delete mode 100644 database/__init__.py rename {database => db}/connection.py (94%) rename {database => db}/repositories/__init__.py (100%) rename {database => db}/repositories/ai_prediction_repository.py (98%) rename {database => db}/repositories/inventory_repository.py (98%) rename {database => db}/repositories/product_repository.py (98%) rename {database => db}/repositories/robot_repository.py (98%) rename {database => db}/repositories/user_repository.py (99%) diff --git a/database/__init__.py b/database/__init__.py deleted file mode 100644 index c25aeba..0000000 --- a/database/__init__.py +++ /dev/null @@ -1,3 +0,0 @@ -from .connection import get_connection, get_db_config - -__all__ = ['get_connection', 'get_db_config'] \ No newline at end of file diff --git a/database/connection.py b/db/connection.py similarity index 94% rename from database/connection.py rename to db/connection.py index cf575d4..d1391fe 100644 --- a/database/connection.py +++ b/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', '') } diff --git a/database/repositories/__init__.py b/db/repositories/__init__.py similarity index 100% rename from database/repositories/__init__.py rename to db/repositories/__init__.py diff --git a/database/repositories/ai_prediction_repository.py b/db/repositories/ai_prediction_repository.py similarity index 98% rename from database/repositories/ai_prediction_repository.py rename to db/repositories/ai_prediction_repository.py index 31788c6..3d3b430 100644 --- a/database/repositories/ai_prediction_repository.py +++ b/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 diff --git a/database/repositories/inventory_repository.py b/db/repositories/inventory_repository.py similarity index 98% rename from database/repositories/inventory_repository.py rename to db/repositories/inventory_repository.py index bc101b5..4ba33fd 100644 --- a/database/repositories/inventory_repository.py +++ b/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 diff --git a/database/repositories/product_repository.py b/db/repositories/product_repository.py similarity index 98% rename from database/repositories/product_repository.py rename to db/repositories/product_repository.py index 966a4e4..8d55ce2 100644 --- a/database/repositories/product_repository.py +++ b/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 diff --git a/database/repositories/robot_repository.py b/db/repositories/robot_repository.py similarity index 98% rename from database/repositories/robot_repository.py rename to db/repositories/robot_repository.py index d57e3f8..a259e34 100644 --- a/database/repositories/robot_repository.py +++ b/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: diff --git a/database/repositories/user_repository.py b/db/repositories/user_repository.py similarity index 99% rename from database/repositories/user_repository.py rename to db/repositories/user_repository.py index 22bb523..7b1d979 100644 --- a/database/repositories/user_repository.py +++ b/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]: