aiteamutils 0.2.47__tar.gz → 0.2.48__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiteamutils
-Version: 0.2.47
+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.47 → aiteamutils-0.2.48}/aiteamutils/database.py

@@ -1020,28 +1020,28 @@ async def get_db() -> AsyncGenerator[AsyncSession, None]:
             source_function="get_db"
         )
     
-    session = 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
@@ -1052,8 +1052,7 @@ async def get_db() -> AsyncGenerator[AsyncSession, None]:
             original_error=e
         )
     finally:
-        if session:
-            await session.close()
+        await session.close()
 
 async def get_database_service() -> DatabaseService:
     """DatabaseService 의존성

{aiteamutils-0.2.47 → 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,23 +56,16 @@ def get_service(name: str):
     Returns:
         Callable: 서비스 인스턴스를 반환하는 의존성 함수
     """
-    async def _get_service(db_service = Depends(get_database_service)):
+    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)
             service = service_class(repository, db_service)
-            
-            # 세션 생성 및 설정
-            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="dependencies.get_service"
-                    )
-                service.session = session
-                return service
-                
+            service.session = session
+            return service
         except CustomException as e:
             raise e
         except Exception as e:

aiteamutils-0.2.48/aiteamutils/version.py

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

aiteamutils-0.2.47/aiteamutils/version.py

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