aiteamutils 0.2.55__py3-none-any.whl → 0.2.57__py3-none-any.whl

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

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

{aiteamutils-0.2.55.dist-info → aiteamutils-0.2.57.dist-info}/METADATA

@@ -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.dist-info → aiteamutils-0.2.57.dist-info}/RECORD

@@ -3,14 +3,14 @@ aiteamutils/base_model.py,sha256=ODEnjvUVoxQ1RPCfq8-uZTfTADIA4c7Z3E6G4EVsSX0,270
 aiteamutils/base_repository.py,sha256=vqsundoN0h7FVvgqTBEnnJNMcFpvMK0s_nxBWdIYg-U,7846
 aiteamutils/base_service.py,sha256=s2AcA-6_ogOQKgt2xf_3AG2s6tqBceU4nJoXO1II7S8,24588
 aiteamutils/cache.py,sha256=07xBGlgAwOTAdY5mnMOQJ5EBxVwe8glVD7DkGEkxCtw,1373
-aiteamutils/config.py,sha256=OM_b7g8sqZ3zY_DSF9ry-zn5wn4dlXdx5OhjfTGr0TE,2876
+aiteamutils/config.py,sha256=YdalpJb70-txhGJAS4aaKglEZAFVWgfzw5BXSWpkUz4,3232
 aiteamutils/database.py,sha256=x0x5gnSyGfwo_klL9O65RnGOQID6c9tH2miwFveVyoE,6326
-aiteamutils/dependencies.py,sha256=UrPjMfUlmG1uGGeWI0H7fINfxChL-LTfD-odFUA5xqQ,5965
+aiteamutils/dependencies.py,sha256=vkSEiXYsgeTEh8mEVtI_sMBjCmcUhxLyqFflAR5MpPU,6792
 aiteamutils/enums.py,sha256=ipZi6k_QD5-3QV7Yzv7bnL0MjDz-vqfO9I5L77biMKs,632
 aiteamutils/exceptions.py,sha256=_lKWXq_ujNj41xN6LDE149PwsecAP7lgYWbOBbLOntg,15368
 aiteamutils/security.py,sha256=xFVrjttxwXB1TTjqgRQQgQJQohQBT28vuW8FVLjvi-M,10103
 aiteamutils/validators.py,sha256=3N245cZFjgwtW_KzjESkizx5BBUDaJLbbxfNO4WOFZ0,7764
-aiteamutils/version.py,sha256=q4mITsP_7Bmz6MAA1ME4HFQVgT80ojkzg8h-5uUxde4,42
-aiteamutils-0.2.55.dist-info/METADATA,sha256=UrrWJHMtqDp8HsWEk3hfzMkKcfm11VbgJv21yorOR1s,1718
-aiteamutils-0.2.55.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-aiteamutils-0.2.55.dist-info/RECORD,,
+aiteamutils/version.py,sha256=iEd2OwHHOAuMG_ICyOZTObkqFKxGQjbDcLvqr9IZKXk,42
+aiteamutils-0.2.57.dist-info/METADATA,sha256=y8dfe__6SqRo2LLuki2isEhePbvNg_sGNJiIzMm3U2s,1718
+aiteamutils-0.2.57.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+aiteamutils-0.2.57.dist-info/RECORD,,