aiteamutils 0.2.55__tar.gz → 0.2.57__tar.gz

{aiteamutils-0.2.55 → aiteamutils-0.2.57}/PKG-INFO

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

{aiteamutils-0.2.55 → aiteamutils-0.2.57}/aiteamutils/config.py

@@ -1,6 +1,5 @@
 """설정 모듈."""
 from typing import Union
-from .database import DatabaseServiceManager
 from .exceptions import CustomException, ErrorCode
 
 class Settings:
@@ -11,13 +10,28 @@ class Settings:
         jwt_algorithm: str = "HS256",
         access_token_expire_minutes: int = 30,
         token_issuer: str = "ai-team",
-        token_audience: str = "ai-team"
+        token_audience: str = "ai-team",
+        db_url: str = None,
+        db_echo: bool = False,
+        db_pool_size: int = 5,
+        db_max_overflow: int = 10,
+        db_pool_timeout: int = 30,
+        db_pool_recycle: int = 1800
     ):
+        # JWT 설정
         self.JWT_SECRET = jwt_secret
         self.JWT_ALGORITHM = jwt_algorithm
         self.ACCESS_TOKEN_EXPIRE_MINUTES = access_token_expire_minutes
         self.TOKEN_ISSUER = token_issuer
         self.TOKEN_AUDIENCE = token_audience
+        
+        # 데이터베이스 설정
+        self.DB_URL = db_url
+        self.DB_ECHO = db_echo
+        self.DB_POOL_SIZE = db_pool_size
+        self.DB_MAX_OVERFLOW = db_max_overflow
+        self.DB_POOL_TIMEOUT = db_pool_timeout
+        self.DB_POOL_RECYCLE = db_pool_recycle
 
 _settings: Union[Settings, None] = None
 
@@ -55,18 +69,14 @@ async def init_settings(
         jwt_algorithm=jwt_algorithm,
         access_token_expire_minutes=access_token_expire_minutes,
         token_issuer=token_issuer,
-        token_audience=token_audience
+        token_audience=token_audience,
+        db_url=db_url,
+        db_echo=db_echo,
+        db_pool_size=db_pool_size,
+        db_max_overflow=db_max_overflow,
+        db_pool_timeout=db_pool_timeout,
+        db_pool_recycle=db_pool_recycle
     )
-    
-    if db_url:
-        await DatabaseServiceManager.get_instance(
-            db_url=db_url,
-            db_echo=db_echo,
-            db_pool_size=db_pool_size,
-            db_max_overflow=db_max_overflow,
-            db_pool_timeout=db_pool_timeout,
-            db_pool_recycle=db_pool_recycle
-        )
 
 def get_settings() -> Settings:
     """현재 설정을 반환하는 함수

{aiteamutils-0.2.55 → aiteamutils-0.2.57}/aiteamutils/dependencies.py

@@ -15,6 +15,16 @@ T = TypeVar("T", bound=BaseService)
 R = TypeVar("R", bound=BaseRepository)
 
 _service_registry: Dict[str, Dict[str, Any]] = {}
+_session_provider = None
+
+def setup_dependencies(session_provider: Callable[[], AsyncGenerator[AsyncSession, None]]) -> None:
+    """의존성 설정을 초기화합니다.
+    
+    Args:
+        session_provider: 데이터베이스 세션을 제공하는 함수
+    """
+    global _session_provider
+    _session_provider = session_provider
 
 def register_service(
     service_class: Type[T],
@@ -28,6 +38,13 @@ def register_service(
         repository_class: 저장소 클래스 (선택)
         **kwargs: 추가 의존성
     """
+    if _session_provider is None:
+        raise CustomException(
+            ErrorCode.INTERNAL_ERROR,
+            detail="Dependencies not initialized",
+            source_function="dependencies.register_service"
+        )
+        
     service_name = service_class.__name__
     _service_registry[service_name] = {
         "service_class": service_class,
@@ -103,9 +120,16 @@ def get_service(service_name: str) -> Callable:
     Returns:
         Callable: 서비스 의존성 함수
     """
+    if _session_provider is None:
+        raise CustomException(
+            ErrorCode.INTERNAL_ERROR,
+            detail="Dependencies not initialized",
+            source_function="dependencies.get_service"
+        )
+    
     async def _get_service_dependency(
         request: Request,
-        session: AsyncSession
+        session: AsyncSession = Depends(_session_provider)
     ) -> BaseService:
         return await _get_service(service_name, session, request)
     return _get_service_dependency

aiteamutils-0.2.57/aiteamutils/version.py

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

aiteamutils-0.2.55/aiteamutils/version.py

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