aiteamutils 0.2.46__tar.gz → 0.2.47__tar.gz

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

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

@@ -1020,6 +1020,7 @@ 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:
@@ -1042,6 +1043,8 @@ async def get_db() -> AsyncGenerator[AsyncSession, None]:
                 
             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,7 +1052,7 @@ async def get_db() -> AsyncGenerator[AsyncSession, None]:
             original_error=e
         )
     finally:
-        if 'session' in locals() and session is not None:
+        if session:
             await session.close()
 
 async def get_database_service() -> DatabaseService:

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

@@ -56,13 +56,23 @@ def get_service(name: str):
     Returns:
         Callable: 서비스 인스턴스를 반환하는 의존성 함수
     """
-    async def _get_service(db_service = Depends(get_database_service), session = Depends(get_db)):
+    async def _get_service(db_service = Depends(get_database_service)):
         try:
             repository_class, service_class = service_registry.get(name)
             repository = repository_class(db_service)
             service = service_class(repository, db_service)
-            service.session = session
-            return 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
+                
         except CustomException as e:
             raise e
         except Exception as e:

aiteamutils-0.2.47/aiteamutils/version.py

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

aiteamutils-0.2.46/aiteamutils/version.py

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