aiteamutils 0.2.31__py3-none-any.whl → 0.2.33__py3-none-any.whl

aiteamutils/config.py

@@ -1,6 +1,7 @@
 """설정 모듈."""
 from typing import Union
 from .database import init_database_service
+from .exceptions import CustomException, ErrorCode
 
 class Settings:
     """기본 설정 클래스"""
@@ -74,8 +75,12 @@ def get_settings() -> Settings:
         Settings: 설정 객체
         
     Raises:
-        RuntimeError: 설정이 초기화되지 않은 경우
+        CustomException: 설정이 초기화되지 않은 경우
     """
     if _settings is None:
-        raise RuntimeError("Settings not initialized. Call init_settings first.")
+        raise CustomException(
+            ErrorCode.INTERNAL_ERROR,
+            detail="settings",
+            source_function="get_settings"
+        )
     return _settings 

aiteamutils/database.py

@@ -26,10 +26,14 @@ def get_database_service() -> 'DatabaseService':
         DatabaseService: DatabaseService 인스턴스
         
     Raises:
-        RuntimeError: DatabaseService가 초기화되지 않은 경우
+        CustomException: DatabaseService가 초기화되지 않은 경우
     """
     if _database_service is None:
-        raise RuntimeError("DatabaseService not initialized. Call init_settings with db_url first.")
+        raise CustomException(
+            ErrorCode.DB_CONNECTION_ERROR,
+            detail="database_service",
+            source_function="get_database_service"
+        )
     return _database_service
 
 async def get_db() -> AsyncGenerator[AsyncSession, None]:
@@ -100,13 +104,21 @@ class DatabaseService:
             )
             self.db = session
         else:
-            raise ValueError("Either db_url or session must be provided")
+            raise CustomException(
+                ErrorCode.DB_CONNECTION_ERROR,
+                detail="db_url|session",
+                source_function="DatabaseService.__init__"
+            )
 
     @asynccontextmanager
     async def get_session(self) -> AsyncGenerator[AsyncSession, None]:
         """데이터베이스 세션을 생성하고 반환하는 비동기 컨텍스트 매니저."""
         if self.session_factory is None:
-            raise RuntimeError("Session factory not initialized")
+            raise CustomException(
+                ErrorCode.DB_CONNECTION_ERROR,
+                detail="session_factory",
+                source_function="DatabaseService.get_session"
+            )
         
         async with self.session_factory() as session:
             try:

aiteamutils/dependencies.py

@@ -26,10 +26,14 @@ class ServiceRegistry:
             service_class (Type): Service 클래스
 
         Raises:
-            ValueError: 이미 등록된 서비스인 경우
+            CustomException: 이미 등록된 서비스인 경우
         """
         if name in self._services:
-            raise ValueError(f"Service '{name}' is already registered.")
+            raise CustomException(
+                ErrorCode.INTERNAL_ERROR,
+                detail=f"service|{name}",
+                source_function="ServiceRegistry.register"
+            )
         self._services[name] = (repository_class, service_class)
 
     def get(self, name: str) -> Tuple[Type, Type]:
@@ -42,10 +46,14 @@ class ServiceRegistry:
             Tuple[Type, Type]: (Repository 클래스, Service 클래스) 튜플
 
         Raises:
-            ValueError: 등록되지 않은 서비스인 경우
+            CustomException: 등록되지 않은 서비스인 경우
         """
         if name not in self._services:
-            raise ValueError(f"Service '{name}' is not registered.")
+            raise CustomException(
+                ErrorCode.NOT_FOUND,
+                detail=f"service|{name}",
+                source_function="ServiceRegistry.get"
+            )
         return self._services[name]
 
 # ServiceRegistry 초기화
@@ -60,10 +68,10 @@ def get_service(name: str):
     Returns:
         Callable: 서비스 인스턴스를 반환하는 의존성 함수
     """
-    def _get_service(db: AsyncSession = Depends(get_db)):
+    def _get_service(db_service: DatabaseService = Depends(get_database_service)):
         repository_class, service_class = service_registry.get(name)
-        repository = repository_class(db)
-        return service_class(repository, db)
+        repository = repository_class(db_service)
+        return service_class(repository, db_service)
     return _get_service
 
 oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/users/token")

aiteamutils/security.py

@@ -157,8 +157,19 @@ def verify_password(plain_password: str, hashed_password: str) -> bool:
         
     Returns:
         bool: 비밀번호 일치 여부
+        
+    Raises:
+        CustomException: 비밀번호 검증 실패 시
     """
-    return pwd_context.verify(plain_password, hashed_password)
+    try:
+        return pwd_context.verify(plain_password, hashed_password)
+    except Exception as e:
+        raise CustomException(
+            ErrorCode.INVALID_PASSWORD,
+            detail=plain_password,
+            source_function="security.verify_password",
+            original_error=e
+        )
 
 def hash_password(password: str) -> str:
     """비밀번호를 해시화합니다.
@@ -168,8 +179,19 @@ def hash_password(password: str) -> str:
         
     Returns:
         str: 해시된 비밀번호
+        
+    Raises:
+        CustomException: 비밀번호 해시화 실패 시
     """
-    return pwd_context.hash(password)
+    try:
+        return pwd_context.hash(password)
+    except Exception as e:
+        raise CustomException(
+            ErrorCode.INTERNAL_ERROR,
+            detail=password,
+            source_function="security.hash_password",
+            original_error=e
+        )
 
 def rate_limit(
     max_requests: int,
@@ -288,10 +310,10 @@ async def create_jwt_token(
         required_fields = {"username", "ulid"}
         missing_fields = required_fields - set(user_data.keys())
         if missing_fields:
-            raise TokenCreationError(
-                detail=f"Missing required fields: {', '.join(missing_fields)}",
-                source_function="create_jwt_token",
-                token_type=token_type
+            raise CustomException(
+                ErrorCode.REQUIRED_FIELD_MISSING,
+                detail="|".join(missing_fields),
+                source_function="security.create_jwt_token"
             )
         
         if token_type == "access":
@@ -337,10 +359,10 @@ async def create_jwt_token(
                 algorithm=settings.JWT_ALGORITHM
             )
         except Exception as e:
-            raise TokenCreationError(
-                detail="Failed to encode JWT token",
-                source_function="create_jwt_token",
-                token_type=token_type,
+            raise CustomException(
+                ErrorCode.INTERNAL_ERROR,
+                detail=f"token|{token_type}",
+                source_function="security.create_jwt_token",
                 original_error=e
             )
         
@@ -362,13 +384,13 @@ async def create_jwt_token(
         
         return token
         
-    except (TokenCreationError, SecurityError) as e:
+    except CustomException as e:
         raise e
     except Exception as e:
-        raise SecurityError(
+        raise CustomException(
             ErrorCode.INTERNAL_ERROR,
             detail=str(e),
-            source_function="create_jwt_token",
+            source_function="security.create_jwt_token",
             original_error=e
         )
 
@@ -376,7 +398,18 @@ async def verify_jwt_token(
     token: str,
     expected_type: Optional[Literal["access", "refresh"]] = None
 ) -> Dict[str, Any]:
-    """JWT 토큰을 검증합니다."""
+    """JWT 토큰을 검증합니다.
+    
+    Args:
+        token: 검증할 JWT 토큰
+        expected_type: 예상되는 토큰 타입
+        
+    Returns:
+        Dict[str, Any]: 토큰 페이로드
+        
+    Raises:
+        CustomException: 토큰 검증 실패 시
+    """
     try:
         settings = get_settings()
         # 토큰 디코딩
@@ -390,25 +423,27 @@ async def verify_jwt_token(
         
         # 토큰 타입 검증 (expected_type이 주어진 경우에만)
         if expected_type and payload.get("token_type") != expected_type:
-            raise TokenError(
-                detail=f"Expected {expected_type} token but got {payload.get('token_type')}",
-                source_function="verify_jwt_token"
+            raise CustomException(
+                ErrorCode.INVALID_TOKEN,
+                detail=f"token|{expected_type}|{payload.get('token_type')}",
+                source_function="security.verify_jwt_token"
             )
         
         return payload
         
     except JWTError as e:
-        raise TokenError(
-            detail=str(e),
-            source_function="verify_jwt_token",
+        raise CustomException(
+            ErrorCode.INVALID_TOKEN,
+            detail=token[:10] + "...",
+            source_function="security.verify_jwt_token",
             original_error=e
         )
-    except (TokenError, SecurityError) as e:
+    except CustomException as e:
         raise e
     except Exception as e:
-        raise SecurityError(
+        raise CustomException(
             ErrorCode.INTERNAL_ERROR,
             detail=str(e),
-            source_function="verify_jwt_token",
+            source_function="security.verify_jwt_token",
             original_error=e
         ) 

aiteamutils/version.py

@@ -1,2 +1,2 @@
 """버전 정보"""
-__version__ = "0.2.31"
+__version__ = "0.2.33"

{aiteamutils-0.2.31.dist-info → aiteamutils-0.2.33.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiteamutils
-Version: 0.2.31
+Version: 0.2.33
 Summary: AI Team Utilities
 Project-URL: Homepage, https://github.com/yourusername/aiteamutils
 Project-URL: Issues, https://github.com/yourusername/aiteamutils/issues

{aiteamutils-0.2.31.dist-info → aiteamutils-0.2.33.dist-info}/RECORD

@@ -3,14 +3,14 @@ aiteamutils/base_model.py,sha256=ODEnjvUVoxQ1RPCfq8-uZTfTADIA4c7Z3E6G4EVsSX0,270
 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=7pJHBml8RRXC9FpoRBx0emofFkvtxsyk9_Rcur2F9hI,2724
-aiteamutils/database.py,sha256=raYKRZjEfefM3SnoPYozPu7OUGm62XWOUqZzN4_cayw,33479
-aiteamutils/dependencies.py,sha256=16Q_76gu9Vff-1XzYar_RREGEegIbUyZJzBbYf7HiR0,3943
+aiteamutils/config.py,sha256=kFKMeIx1KcuEwwx4VjZdCgoTOHCkG3ySYVJ0G6cvMoA,2849
+aiteamutils/database.py,sha256=CaH73g8PPNcmLCz4Xr0DBtNSrLcpRlyt0F1zO5Tigfo,33811
+aiteamutils/dependencies.py,sha256=GgN8sFM7qGNmOfpLaFrhuHOgPlBVXWrEgxv29jxjEgI,4226
 aiteamutils/enums.py,sha256=ipZi6k_QD5-3QV7Yzv7bnL0MjDz-vqfO9I5L77biMKs,632
 aiteamutils/exceptions.py,sha256=YV-ISya4wQlHk4twvGo16I5r8h22-tXpn9wa-b3WwDM,15231
-aiteamutils/security.py,sha256=jxJsWSjJOMVlNaBFxroIXrAgseJYKl_vQ_kfmHblPIA,13498
+aiteamutils/security.py,sha256=nqAyypyiZ26s-eKv7c6YPDTK6HcqimFnoCr12Wtg1DY,14449
 aiteamutils/validators.py,sha256=3N245cZFjgwtW_KzjESkizx5BBUDaJLbbxfNO4WOFZ0,7764
-aiteamutils/version.py,sha256=YK0jUv01ukicmNcU34hKfOfAtqr0QfPU2FTKHjSy49g,42
-aiteamutils-0.2.31.dist-info/METADATA,sha256=0j6hq7VMfiaw5f0W9IBD67LK2BSBaXRuRI4cP5_6Xw0,1718
-aiteamutils-0.2.31.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-aiteamutils-0.2.31.dist-info/RECORD,,
+aiteamutils/version.py,sha256=0u62c8uMhBIqtJ4INrTiDbrTgFVL2CBQ_nfjWnhrldU,42
+aiteamutils-0.2.33.dist-info/METADATA,sha256=DUbbkj6hQ_0C7ODwi8AuQJ1W1BYAMF8_45epV7wl27I,1718
+aiteamutils-0.2.33.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+aiteamutils-0.2.33.dist-info/RECORD,,