aiteamutils 0.2.46__tar.gz → 0.2.48__tar.gz

{aiteamutils-0.2.46 → aiteamutils-0.2.48}/PKG-INFO

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

{aiteamutils-0.2.46 → aiteamutils-0.2.48}/aiteamutils/database.py

@@ -1021,27 +1021,30 @@ async def get_db() -> AsyncGenerator[AsyncSession, None]:
         )
     
     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"
-                )
+        session = db_service.get_session()
+        if session is None:
+            raise CustomException(
+                ErrorCode.DB_CONNECTION_ERROR,
+                detail="Failed to create database session",
+                source_function="get_db"
+            )
+        
+        # 세션이 유효한지 확인
+        try:
+            await session.execute(select(1))
+        except Exception as e:
+            await session.close()
+            raise CustomException(
+                ErrorCode.DB_CONNECTION_ERROR,
+                detail="Database session is not valid",
+                source_function="get_db",
+                original_error=e
+            )
             
-            # 세션이 유효한지 확인
-            try:
-                await session.execute(select(1))
-            except Exception as e:
-                raise CustomException(
-                    ErrorCode.DB_CONNECTION_ERROR,
-                    detail="Database session is not valid",
-                    source_function="get_db",
-                    original_error=e
-                )
-                
-            yield session
+        yield session
     except Exception as e:
+        if isinstance(e, CustomException):
+            raise e
         raise CustomException(
             ErrorCode.DB_CONNECTION_ERROR,
             detail=f"Failed to get database session: {str(e)}",
@@ -1049,8 +1052,7 @@ async def get_db() -> AsyncGenerator[AsyncSession, None]:
             original_error=e
         )
     finally:
-        if 'session' in locals() and session is not None:
-            await session.close()
+        await session.close()
 
 async def get_database_service() -> DatabaseService:
     """DatabaseService 의존성

{aiteamutils-0.2.46 → aiteamutils-0.2.48}/aiteamutils/dependencies.py

@@ -5,7 +5,7 @@ from jose import JWTError, jwt
 from sqlalchemy.ext.asyncio import AsyncSession
 import logging
 
-from .database import DatabaseServiceManager, get_db, get_database_service
+from .database import DatabaseServiceManager, DatabaseService, get_db, get_database_service
 from .exceptions import CustomException, ErrorCode
 from .config import get_settings
 
@@ -56,7 +56,10 @@ def get_service(name: str):
     Returns:
         Callable: 서비스 인스턴스를 반환하는 의존성 함수
     """
-    async def _get_service(db_service = Depends(get_database_service), session = Depends(get_db)):
+    def _get_service(
+        db_service: DatabaseService = Depends(get_database_service),
+        session: AsyncSession = Depends(get_db)
+    ):
         try:
             repository_class, service_class = service_registry.get(name)
             repository = repository_class(db_service)

aiteamutils-0.2.48/aiteamutils/version.py

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

aiteamutils-0.2.46/aiteamutils/version.py

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