aiteamutils 0.2.2__py3-none-any.whl → 0.2.4__py3-none-any.whl

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/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/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/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/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/version.py

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

{aiteamutils-0.2.2.dist-info → aiteamutils-0.2.4.dist-info}/METADATA

@@ -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.4.dist-info/RECORD

@@ -0,0 +1,16 @@
+aiteamutils/__init__.py,sha256=-pgR1RkjBjcIgfkKSeEy65G6jsDStFM4DdiQU8Fwcmg,1239
+aiteamutils/base_model.py,sha256=ODEnjvUVoxQ1RPCfq8-uZTfTADIA4c7Z3E6G4EVsSX0,2708
+aiteamutils/base_repository.py,sha256=qdwQ7Sj2fUqxpDg6cWM48n_QbwPK_VUlG9zTSem8iCk,18968
+aiteamutils/base_service.py,sha256=E4dHGE0DvhmRyFplh46SwKJOSF_nUL7OAsCkf_ZJF_8,24733
+aiteamutils/cache.py,sha256=tr0Yn8VPYA9QHiKCUzciVlQ2J1RAwNo2K9lGMH4rY3s,1334
+aiteamutils/config.py,sha256=vC6k6E2-Y4mD0E0kw6WVgSatCl9K_BtTwrVFhLrhCzs,665
+aiteamutils/database.py,sha256=U71cexPsSmMTKgp098I574PupqnuPltujI4QKHBG2Cc,29952
+aiteamutils/dependencies.py,sha256=EJeVtq_lACuoheVhkX23N9xiak9bGD-t3-2JtlgBki0,4850
+aiteamutils/enums.py,sha256=ipZi6k_QD5-3QV7Yzv7bnL0MjDz-vqfO9I5L77biMKs,632
+aiteamutils/exceptions.py,sha256=YV-ISya4wQlHk4twvGo16I5r8h22-tXpn9wa-b3WwDM,15231
+aiteamutils/security.py,sha256=AZszaTxVEGi1jU1sX3QXHGgshp1lVvd0xXvZejXvs_w,12643
+aiteamutils/validators.py,sha256=9BHm9Zy9paHLArWXbXYjXBQYjOs4XEvM7kAaGEFvz7Q,7747
+aiteamutils/version.py,sha256=qj505TIaEPXVKdETV9MIkFSGabI0IySNMFg-Ms5MO48,42
+aiteamutils-0.2.4.dist-info/METADATA,sha256=nCAT9SdxKDTxRGaO9nxKZs5bhCKzG8C09uo0-ahDgT0,1717
+aiteamutils-0.2.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+aiteamutils-0.2.4.dist-info/RECORD,,

aiteamutils-0.2.2.dist-info/RECORD

@@ -1,15 +0,0 @@
-aiteamutils/__init__.py,sha256=zmfBXBwNWdbJKCt1rmHk_czHJLmVF-oqRuqq8tf0t0U,1229
-aiteamutils/base_model.py,sha256=ODEnjvUVoxQ1RPCfq8-uZTfTADIA4c7Z3E6G4EVsSX0,2708
-aiteamutils/base_repository.py,sha256=TQAXajQaaXIAJAFr9n5R2ikcu1TSq-E-_AjXgsND5c4,18927
-aiteamutils/base_service.py,sha256=cqPP74ya6r8twJyU3Bc6ARxKiRL8fQIIZZnhDHxBQXE,24727
-aiteamutils/cache.py,sha256=tr0Yn8VPYA9QHiKCUzciVlQ2J1RAwNo2K9lGMH4rY3s,1334
-aiteamutils/config.py,sha256=vC6k6E2-Y4mD0E0kw6WVgSatCl9K_BtTwrVFhLrhCzs,665
-aiteamutils/database.py,sha256=NwLgDIGSaWTORR5eU3W2rYSUlKPKJCWqF2f5yzLyf7A,29952
-aiteamutils/dependencies.py,sha256=EJeVtq_lACuoheVhkX23N9xiak9bGD-t3-2JtlgBki0,4850
-aiteamutils/enums.py,sha256=ipZi6k_QD5-3QV7Yzv7bnL0MjDz-vqfO9I5L77biMKs,632
-aiteamutils/exceptions.py,sha256=YV-ISya4wQlHk4twvGo16I5r8h22-tXpn9wa-b3WwDM,15231
-aiteamutils/security.py,sha256=AZszaTxVEGi1jU1sX3QXHGgshp1lVvd0xXvZejXvs_w,12643
-aiteamutils/validators.py,sha256=GA2idwDtbQynle7tya10cAnJ6DvLIFsJ746We0V-30k,7630
-aiteamutils-0.2.2.dist-info/METADATA,sha256=kOqyfN5HkeRdM6L7jEttopFCIbbFxQLqKdxWQhtLLT0,1527
-aiteamutils-0.2.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-aiteamutils-0.2.2.dist-info/RECORD,,