aiteamutils 0.2.56__tar.gz → 0.2.58__tar.gz

{aiteamutils-0.2.56 → aiteamutils-0.2.58}/PKG-INFO

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

{aiteamutils-0.2.56 → aiteamutils-0.2.58}/aiteamutils/__init__.py

@@ -21,6 +21,7 @@ from .base_repository import BaseRepository
 from .validators import validate_with
 from .enums import ActivityType
 from .version import __version__
+from .dependencies import setup_dependencies, register_service
 
 __all__ = [
     # Base Models
@@ -31,6 +32,10 @@ __all__ = [
     # Database
     "DatabaseService",
     
+    # Dependencies
+    "setup_dependencies",
+    "register_service",
+    
     # Exceptions
     "CustomException",
     "ErrorCode",

{aiteamutils-0.2.56 → aiteamutils-0.2.58}/aiteamutils/dependencies.py

@@ -15,6 +15,23 @@ T = TypeVar("T", bound=BaseService)
 R = TypeVar("R", bound=BaseRepository)
 
 _service_registry: Dict[str, Dict[str, Any]] = {}
+_session_provider = None
+
+__all__ = [
+    "setup_dependencies",
+    "register_service",
+    "get_service",
+    "get_current_user"
+]
+
+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 +45,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 +127,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.58/aiteamutils/version.py

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

aiteamutils-0.2.56/aiteamutils/version.py

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