aiteamutils 0.2.2__tar.gz → 0.2.4__tar.gz

{aiteamutils-0.2.2 → aiteamutils-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiteamutils
-Version: 0.2.2
+Version: 0.2.4
 Summary: AI Team Utilities
 Project-URL: Homepage, https://github.com/yourusername/aiteamutils
 Project-URL: Issues, https://github.com/yourusername/aiteamutils/issues
@@ -27,17 +27,21 @@ pip install aiteamutils
 ## 사용 예시
 
 ```python
-from aiteamutils.database import DatabaseManager
-from aiteamutils.base_model import Base
+from aiteamutils.database import DatabaseService
 
-# DB 매니저 초기화
-db = DatabaseManager("postgresql+asyncpg://user:pass@localhost/db")
+# DB 서비스 초기화
+db_service = DatabaseService("postgresql+asyncpg://user:pass@localhost/db")
 
 # DB 세션 사용
-async with db.get_session() as session:
+async with db_service.get_db() as session:
     # DB 작업 수행
     pass
 
+# 트랜잭션 사용
+async with db_service.transaction():
+    # 트랜잭션 내 작업 수행
+    result = await db_service.create_entity(UserModel, {"name": "test"})
+
 # 예외 처리
 from aiteamutils.exceptions import CustomException, ErrorCode
 
@@ -55,6 +59,8 @@ except CustomException as e:
   - 세션 관리
   - 트랜잭션 관리
   - 기본 CRUD 작업
+  - 외래키 검증
+  - 유니크 필드 검증
 
 - 인증/인가 유틸리티
   - JWT 토큰 관리

{aiteamutils-0.2.2 → aiteamutils-0.2.4}/README.md

@@ -11,17 +11,21 @@ pip install aiteamutils
 ## 사용 예시
 
 ```python
-from aiteamutils.database import DatabaseManager
-from aiteamutils.base_model import Base
+from aiteamutils.database import DatabaseService
 
-# DB 매니저 초기화
-db = DatabaseManager("postgresql+asyncpg://user:pass@localhost/db")
+# DB 서비스 초기화
+db_service = DatabaseService("postgresql+asyncpg://user:pass@localhost/db")
 
 # DB 세션 사용
-async with db.get_session() as session:
+async with db_service.get_db() as session:
     # DB 작업 수행
     pass
 
+# 트랜잭션 사용
+async with db_service.transaction():
+    # 트랜잭션 내 작업 수행
+    result = await db_service.create_entity(UserModel, {"name": "test"})
+
 # 예외 처리
 from aiteamutils.exceptions import CustomException, ErrorCode
 
@@ -39,6 +43,8 @@ except CustomException as e:
   - 세션 관리
   - 트랜잭션 관리
   - 기본 CRUD 작업
+  - 외래키 검증
+  - 유니크 필드 검증
 
 - 인증/인가 유틸리티
   - JWT 토큰 관리

{aiteamutils-0.2.2 → aiteamutils-0.2.4}/aiteamutils/__init__.py

@@ -1,5 +1,5 @@
 from .base_model import Base
-from .database import DatabaseManager
+from .database import DatabaseService
 from .exceptions import (
     CustomException,
     ErrorCode,
@@ -21,8 +21,7 @@ from .base_repository import BaseRepository
 from .validators import validate_with
 from .enums import ActivityType
 from .cache import CacheManager
-
-__version__ = "0.1.0"
+from .version import __version__
 
 __all__ = [
     # Base Models
@@ -31,7 +30,7 @@ __all__ = [
     "BaseRepository",
     
     # Database
-    "DatabaseManager",
+    "DatabaseService",
     
     # Exceptions
     "CustomException",

{aiteamutils-0.2.2 → aiteamutils-0.2.4}/aiteamutils/base_repository.py

@@ -1,12 +1,13 @@
 """기본 레포지토리 모듈."""
-from typing import TypeVar, Generic, Dict, Any, List, Optional, Type
-from sqlalchemy.orm import DeclarativeBase
+from typing import TypeVar, Generic, Dict, Any, List, Optional, Type, Union
+from sqlalchemy.orm import DeclarativeBase, Load
 from sqlalchemy.exc import IntegrityError, SQLAlchemyError
 from sqlalchemy import select, or_, and_
-from .database import DatabaseManager
+from .database import DatabaseService
 from .exceptions import CustomException, ErrorCode
 from sqlalchemy.orm import joinedload
 from sqlalchemy.sql import Select
+from fastapi import Request
 
 ModelType = TypeVar("ModelType", bound=DeclarativeBase)
 
@@ -14,10 +15,10 @@ class BaseRepository(Generic[ModelType]):
    ##################
     # 1. 초기화 영역 #
     ##################
-    def __init__(self, db_service: DatabaseManager, model: Type[ModelType]):
+    def __init__(self, db_service: DatabaseService, model: Type[ModelType]):
         """
         Args:
-            db_service (DatabaseManager): 데이터베이스 서비스 인스턴스
+            db_service (DatabaseService): 데이터베이스 서비스 인스턴스
             model (Type[ModelType]): 모델 클래스
         """
         self.db_service = db_service

{aiteamutils-0.2.2 → aiteamutils-0.2.4}/aiteamutils/base_service.py

@@ -1,9 +1,9 @@
 """기본 서비스 모듈."""
 from datetime import datetime
 from typing import TypeVar, Generic, Dict, Any, List, Optional, Type, Union
-from sqlalchemy.orm import DeclarativeBase
+from sqlalchemy.orm import DeclarativeBase, Load
 from sqlalchemy.exc import IntegrityError, SQLAlchemyError
-from .database import DatabaseManager
+from .database import DatabaseService
 from .exceptions import CustomException, ErrorCode
 from .base_repository import BaseRepository
 from .security import hash_password

{aiteamutils-0.2.2 → aiteamutils-0.2.4}/aiteamutils/database.py

@@ -17,9 +17,9 @@ from .config import settings
 
 T = TypeVar("T", bound=BaseColumn)
 
-class DatabaseManager:
+class DatabaseService:
     def __init__(self, db_url: str):
-        """DatabaseManager 초기화.
+        """DatabaseService 초기화.
         
         Args:
             db_url: 데이터베이스 URL
@@ -173,7 +173,7 @@ class DatabaseManager:
                 raise CustomException(
                     ErrorCode.DUPLICATE_ERROR,
                     detail=f"{model.__tablename__}|{field}|{value}",
-                    source_function="DatabaseManager.create_entity",
+                    source_function="DatabaseService.create_entity",
                     original_error=e
                 )
             elif "violates foreign key constraint" in error_str.lower():
@@ -181,14 +181,14 @@ class DatabaseManager:
                 raise CustomException(
                     ErrorCode.FOREIGN_KEY_VIOLATION,
                     detail=error_str,
-                    source_function="DatabaseManager.create_entity",
+                    source_function="DatabaseService.create_entity",
                     original_error=e
                 )
             else:
                 raise CustomException(
                     ErrorCode.DB_CREATE_ERROR,
                     detail=f"Failed to create {model.__name__}: {str(e)}",
-                    source_function="DatabaseManager.create_entity",
+                    source_function="DatabaseService.create_entity",
                     original_error=e
                 )
         except CustomException as e:
@@ -199,7 +199,7 @@ class DatabaseManager:
             raise CustomException(
                 ErrorCode.UNEXPECTED_ERROR,
                 detail=f"Unexpected error while creating {model.__name__}: {str(e)}",
-                source_function="DatabaseManager.create_entity",
+                source_function="DatabaseService.create_entity",
                 original_error=e
             )
 
@@ -243,7 +243,7 @@ class DatabaseManager:
             raise CustomException(
                 ErrorCode.DB_QUERY_ERROR,
                 detail=str(e),
-                source_function="DatabaseManager.retrieve_entity",
+                source_function="DatabaseService.retrieve_entity",
                 original_error=e
             )
 
@@ -307,7 +307,7 @@ class DatabaseManager:
             raise CustomException(
                 ErrorCode.DB_UPDATE_ERROR,
                 detail=f"Failed to update {model.__name__}: {str(e)}",
-                source_function="DatabaseManager.update_entity",
+                source_function="DatabaseService.update_entity",
                 original_error=e
             )
         except Exception as e:
@@ -315,7 +315,7 @@ class DatabaseManager:
             raise CustomException(
                 ErrorCode.UNEXPECTED_ERROR,
                 detail=f"Unexpected error while updating {model.__name__}: {str(e)}",
-                source_function="DatabaseManager.update_entity",
+                source_function="DatabaseService.update_entity",
                 original_error=e
             )
 
@@ -337,7 +337,7 @@ class DatabaseManager:
             raise CustomException(
                 ErrorCode.DB_DELETE_ERROR,
                 detail=f"Failed to delete entity: {str(e)}",
-                source_function="DatabaseManager.delete_entity",
+                source_function="DatabaseService.delete_entity",
                 original_error=e
             )
         except Exception as e:
@@ -345,7 +345,7 @@ class DatabaseManager:
             raise CustomException(
                 ErrorCode.UNEXPECTED_ERROR,
                 detail=f"Unexpected error while deleting entity: {str(e)}",
-                source_function="DatabaseManager.delete_entity",
+                source_function="DatabaseService.delete_entity",
                 original_error=e
             )
 
@@ -392,7 +392,7 @@ class DatabaseManager:
             raise CustomException(
                 ErrorCode.DB_DELETE_ERROR,
                 detail=f"Failed to soft delete {model.__name__}: {str(e)}",
-                source_function="DatabaseManager.soft_delete_entity",
+                source_function="DatabaseService.soft_delete_entity",
                 original_error=e
             )
         except Exception as e:
@@ -400,7 +400,7 @@ class DatabaseManager:
             raise CustomException(
                 ErrorCode.UNEXPECTED_ERROR,
                 detail=f"Unexpected error while soft deleting {model.__name__}: {str(e)}",
-                source_function="DatabaseManager.soft_delete_entity",
+                source_function="DatabaseService.soft_delete_entity",
                 original_error=e
             )
 
@@ -468,14 +468,14 @@ class DatabaseManager:
             raise CustomException(
                 ErrorCode.DB_READ_ERROR,
                 detail=f"Failed to list {model.__name__}: {str(e)}",
-                source_function="DatabaseManager.list_entities",
+                source_function="DatabaseService.list_entities",
                 original_error=e
             )
         except Exception as e:
             raise CustomException(
                 ErrorCode.UNEXPECTED_ERROR,
                 detail=f"Unexpected error while listing {model.__name__}: {str(e)}",
-                source_function="DatabaseManager.list_entities",
+                source_function="DatabaseService.list_entities",
                 original_error=e
             )
 
@@ -524,14 +524,14 @@ class DatabaseManager:
                 raise CustomException(
                     error_code,
                     detail=detail,
-                    source_function="DatabaseManager.validate_unique_fields"
+                    source_function="DatabaseService.validate_unique_fields"
                 )
 
         except CustomException as e:
             raise CustomException(
                 e.error_code,
                 detail=e.detail,
-                source_function="DatabaseManager.validate_unique_fields",
+                source_function="DatabaseService.validate_unique_fields",
                 original_error=e.original_error,
                 parent_source_function=e.source_function
             )
@@ -539,7 +539,7 @@ class DatabaseManager:
             raise CustomException(
                 ErrorCode.DB_QUERY_ERROR,
                 detail=str(e),
-                source_function="DatabaseManager.validate_unique_fields",
+                source_function="DatabaseService.validate_unique_fields",
                 original_error=e
             )
 
@@ -577,7 +577,7 @@ class DatabaseManager:
                     raise CustomException(
                         ErrorCode.FOREIGN_KEY_VIOLATION,
                         detail=f"{referenced_table.name}|{field}|{value}",
-                        source_function="DatabaseManager.validate_foreign_key_fields"
+                        source_function="DatabaseService.validate_foreign_key_fields"
                     )
 
     #######################
@@ -637,7 +637,7 @@ class DatabaseManager:
                 raise CustomException(
                     ErrorCode.NOT_FOUND,
                     detail=f"{model.__name__}|{entity_id}",
-                    source_function=source_function or "DatabaseManager.soft_delete"
+                    source_function=source_function or "DatabaseService.soft_delete"
                 )
             
             # 2. 소프트 삭제 처리
@@ -665,7 +665,7 @@ class DatabaseManager:
             raise CustomException(
                 e.error_code,
                 detail=e.detail,
-                source_function=source_function or "DatabaseManager.soft_delete",
+                source_function=source_function or "DatabaseService.soft_delete",
                 original_error=e.original_error,
                 parent_source_function=e.source_function
             )
@@ -674,7 +674,7 @@ class DatabaseManager:
             raise CustomException(
                 ErrorCode.DB_DELETE_ERROR,
                 detail=str(e),
-                source_function=source_function or "DatabaseManager.soft_delete",
+                source_function=source_function or "DatabaseService.soft_delete",
                 original_error=e
             )
 
@@ -704,14 +704,14 @@ class DatabaseManager:
             raise CustomException(
                 ErrorCode.DB_READ_ERROR,
                 detail=f"Failed to get {model.__name__}: {str(e)}",
-                source_function="DatabaseManager.get_entity",
+                source_function="DatabaseService.get_entity",
                 original_error=e
             )
         except Exception as e:
             raise CustomException(
                 ErrorCode.UNEXPECTED_ERROR,
                 detail=f"Unexpected error while getting {model.__name__}: {str(e)}",
-                source_function="DatabaseManager.get_entity",
+                source_function="DatabaseService.get_entity",
                 original_error=e
             )
 
@@ -757,14 +757,14 @@ class DatabaseManager:
             raise CustomException(
                 ErrorCode.DB_QUERY_ERROR,
                 detail=str(e),
-                source_function="DatabaseManager.execute",
+                source_function="DatabaseService.execute",
                 original_error=e
             )
         except Exception as e:
             raise CustomException(
                 ErrorCode.UNEXPECTED_ERROR,
                 detail=str(e),
-                source_function="DatabaseManager.execute",
+                source_function="DatabaseService.execute",
                 original_error=e
             )
 
@@ -777,7 +777,7 @@ class DatabaseManager:
             raise CustomException(
                 ErrorCode.DB_QUERY_ERROR,
                 detail=str(e),
-                source_function="DatabaseManager.commit",
+                source_function="DatabaseService.commit",
                 original_error=e
             )
 
@@ -789,7 +789,7 @@ class DatabaseManager:
             raise CustomException(
                 ErrorCode.DB_QUERY_ERROR,
                 detail=str(e),
-                source_function="DatabaseManager.rollback",
+                source_function="DatabaseService.rollback",
                 original_error=e
             )
 
@@ -802,7 +802,7 @@ class DatabaseManager:
             raise CustomException(
                 ErrorCode.DB_QUERY_ERROR,
                 detail=str(e),
-                source_function="DatabaseManager.flush",
+                source_function="DatabaseService.flush",
                 original_error=e
             )
 
@@ -818,6 +818,6 @@ class DatabaseManager:
             raise CustomException(
                 ErrorCode.DB_QUERY_ERROR,
                 detail=str(e),
-                source_function="DatabaseManager.refresh",
+                source_function="DatabaseService.refresh",
                 original_error=e
             )

{aiteamutils-0.2.2 → aiteamutils-0.2.4}/aiteamutils/validators.py

@@ -1,13 +1,16 @@
 """유효성 검사 관련 유틸리티 함수들을 모아둔 모듈입니다."""
 
-from typing import Any, Optional, Dict
+from typing import Type, Dict, Any, Callable, TypeVar, Optional, List, Union
+from functools import wraps
+from sqlalchemy import Table
 from sqlalchemy.ext.asyncio import AsyncSession
-from sqlalchemy import select
-from pydantic import field_validator
+from fastapi import Request
+from inspect import signature
+from pydantic import BaseModel
 import re
 
 from .exceptions import ErrorCode, CustomException
-from .database import DatabaseManager
+from .database import DatabaseService
 from .base_model import Base
 
 def validate_with(validator_func, unique_check=None, skip_if_none=False):

aiteamutils-0.2.4/aiteamutils/version.py

@@ -0,0 +1,2 @@
+"""버전 정보"""
+__version__ = "0.2.4" 

{aiteamutils-0.2.2 → aiteamutils-0.2.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "aiteamutils"
-version = "0.2.2"
+dynamic = ["version"]
 authors = [
   { name="AI Team" },
 ]
@@ -27,5 +27,8 @@ dependencies = [
 Homepage = "https://github.com/yourusername/aiteamutils"
 Issues = "https://github.com/yourusername/aiteamutils/issues"
 
+[tool.hatch.version]
+path = "aiteamutils/version.py"
+
 [tool.hatch.build.targets.wheel]
 packages = ["aiteamutils"]