aiteamutils 0.2.37__tar.gz → 0.2.38__tar.gz

{aiteamutils-0.2.37 → aiteamutils-0.2.38}/PKG-INFO

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

{aiteamutils-0.2.37 → aiteamutils-0.2.38}/aiteamutils/database.py

@@ -9,6 +9,7 @@ from sqlalchemy import or_
 from fastapi import Request, Depends
 from ulid import ULID
 from sqlalchemy.sql import Select
+import logging
 
 from .exceptions import ErrorCode, CustomException
 from .base_model import Base, BaseColumn
@@ -31,18 +32,33 @@ def get_database_service() -> 'DatabaseService':
     if _database_service is None:
         raise CustomException(
             ErrorCode.DB_CONNECTION_ERROR,
-            detail="database_service",
+            detail="Database service is not initialized. Call init_database_service() first.",
             source_function="get_database_service"
         )
     return _database_service
 
 async def get_db() -> AsyncGenerator[AsyncSession, None]:
-    """데이터베이스 세션을 생성하고 반환하는 비동기 제너레이터."""
+    """데이터베이스 세션을 생성하고 반환하는 비동기 제너레이터.
+    
+    Yields:
+        AsyncSession: 데이터베이스 세션
+        
+    Raises:
+        CustomException: 세션 생성 실패 시
+    """
     db_service = get_database_service()
-    async with db_service.get_session() as session:
-        try:
+    try:
+        async with db_service.get_session() as session:
             yield session
-        finally:
+    except Exception as e:
+        raise CustomException(
+            ErrorCode.DB_CONNECTION_ERROR,
+            detail=f"Failed to get database session: {str(e)}",
+            source_function="get_db",
+            original_error=e
+        )
+    finally:
+        if 'session' in locals():
             await session.close()
 
 def get_database_session(db: AsyncSession = Depends(get_db)) -> 'DatabaseService':
@@ -921,14 +937,40 @@ def init_database_service(
         
     Returns:
         DatabaseService: 초기화된 데이터베이스 서비스 인스턴스
+        
+    Raises:
+        CustomException: 데이터베이스 초기화 실패 시
     """
-    global _database_service
-    _database_service = DatabaseService(
-        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
-    )
-    return _database_service
+    try:
+        global _database_service
+        if _database_service is not None:
+            logging.info("Database service already initialized")
+            return _database_service
+            
+        logging.info(f"Initializing database service with URL: {db_url}")
+        _database_service = DatabaseService(
+            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 not _database_service.engine:
+            raise CustomException(
+                ErrorCode.DB_CONNECTION_ERROR,
+                detail="Database engine initialization failed",
+                source_function="init_database_service"
+            )
+            
+        logging.info("Database service initialized successfully")
+        return _database_service
+    except Exception as e:
+        logging.error(f"Failed to initialize database service: {str(e)}")
+        raise CustomException(
+            ErrorCode.DB_CONNECTION_ERROR,
+            detail=f"Failed to initialize database service: {str(e)}",
+            source_function="init_database_service",
+            original_error=e
+        )

{aiteamutils-0.2.37 → aiteamutils-0.2.38}/aiteamutils/dependencies.py

@@ -67,11 +67,22 @@ def get_service(name: str):
 
     Returns:
         Callable: 서비스 인스턴스를 반환하는 의존성 함수
+        
+    Raises:
+        CustomException: 서비스 생성 실패 시
     """
     def _get_service(db_service: DatabaseService = Depends(get_database_service)):
-        repository_class, service_class = service_registry.get(name)
-        repository = repository_class(db_service)
-        return service_class(repository, db_service)
+        try:
+            repository_class, service_class = service_registry.get(name)
+            repository = repository_class(db_service)
+            return service_class(repository)
+        except Exception as e:
+            raise CustomException(
+                ErrorCode.SERVICE_NOT_REGISTERED,
+                detail=f"Failed to create service {name}: {str(e)}",
+                source_function="dependencies.get_service",
+                original_error=e
+            )
     return _get_service
 
 oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/users/token")

aiteamutils-0.2.38/aiteamutils/version.py

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

aiteamutils-0.2.37/aiteamutils/version.py

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