aiteamutils 0.2.41__tar.gz → 0.2.42__tar.gz

{aiteamutils-0.2.41 → aiteamutils-0.2.42}/PKG-INFO

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

{aiteamutils-0.2.41 → aiteamutils-0.2.42}/aiteamutils/database.py

@@ -47,8 +47,21 @@ async def get_db() -> AsyncGenerator[AsyncSession, None]:
         CustomException: 세션 생성 실패 시
     """
     db_service = get_database_service()
+    if not db_service or not db_service.engine:
+        raise CustomException(
+            ErrorCode.DB_CONNECTION_ERROR,
+            detail="Database service or engine is not properly initialized",
+            source_function="get_db"
+        )
+    
     try:
         async with db_service.get_session() as session:
+            if session is None:
+                raise CustomException(
+                    ErrorCode.DB_CONNECTION_ERROR,
+                    detail="Failed to create database session",
+                    source_function="get_db"
+                )
             yield session
     except Exception as e:
         raise CustomException(
@@ -58,7 +71,7 @@ async def get_db() -> AsyncGenerator[AsyncSession, None]:
             original_error=e
         )
     finally:
-        if 'session' in locals():
+        if 'session' in locals() and session is not None:
             await session.close()
 
 def get_database_session(db: AsyncSession = Depends(get_db)) -> 'DatabaseService':
@@ -69,8 +82,26 @@ def get_database_session(db: AsyncSession = Depends(get_db)) -> 'DatabaseService
 
     Returns:
         DatabaseService: DatabaseService 인스턴스
+        
+    Raises:
+        CustomException: 데이터베이스 세션이 유효하지 않은 경우
     """
-    return DatabaseService(session=db)
+    if db is None:
+        raise CustomException(
+            ErrorCode.DB_CONNECTION_ERROR,
+            detail="Database session is not initialized",
+            source_function="get_database_session"
+        )
+    
+    db_service = DatabaseService(session=db)
+    if db_service.db is None:
+        raise CustomException(
+            ErrorCode.DB_CONNECTION_ERROR,
+            detail="Failed to initialize database session",
+            source_function="get_database_session"
+        )
+    
+    return db_service
 
 class DatabaseService:
     def __init__(

aiteamutils-0.2.42/aiteamutils/version.py

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

aiteamutils-0.2.41/aiteamutils/version.py

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