aiteamutils 0.2.47__py3-none-any.whl → 0.2.48__py3-none-any.whl

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/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/version.py

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

{aiteamutils-0.2.47.dist-info → aiteamutils-0.2.48.dist-info}/METADATA

@@ -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.dist-info → aiteamutils-0.2.48.dist-info}/RECORD

@@ -4,13 +4,13 @@ aiteamutils/base_repository.py,sha256=qdwQ7Sj2fUqxpDg6cWM48n_QbwPK_VUlG9zTSem8iC
 aiteamutils/base_service.py,sha256=E4dHGE0DvhmRyFplh46SwKJOSF_nUL7OAsCkf_ZJF_8,24733
 aiteamutils/cache.py,sha256=07xBGlgAwOTAdY5mnMOQJ5EBxVwe8glVD7DkGEkxCtw,1373
 aiteamutils/config.py,sha256=OM_b7g8sqZ3zY_DSF9ry-zn5wn4dlXdx5OhjfTGr0TE,2876
-aiteamutils/database.py,sha256=xUvxSRG49AY5DIWkucKMsaMMtfg5GumgJRwwwiUXm5w,40173
-aiteamutils/dependencies.py,sha256=UC19hLf5LHM4fdqVQHo7fHxytD47lkXuiSQvU2Vd4s0,4827
+aiteamutils/database.py,sha256=rEKuZ8_3eBjP1KpTNVgWrVLAtKX_IkZJzul2pa8bDlg,40075
+aiteamutils/dependencies.py,sha256=0mNmZiPlA8wTj4cgmjGXs6y9kkIQ5RtLZHdEeHxyblo,4490
 aiteamutils/enums.py,sha256=ipZi6k_QD5-3QV7Yzv7bnL0MjDz-vqfO9I5L77biMKs,632
 aiteamutils/exceptions.py,sha256=_lKWXq_ujNj41xN6LDE149PwsecAP7lgYWbOBbLOntg,15368
 aiteamutils/security.py,sha256=KSZf3WJAlW0UXVM1GEHwuGy2x27UHwtMr8aq-GuIXZk,16016
 aiteamutils/validators.py,sha256=3N245cZFjgwtW_KzjESkizx5BBUDaJLbbxfNO4WOFZ0,7764
-aiteamutils/version.py,sha256=o2IXukuF7KexqFBwlqOtUGuZABgd7WMZRzbAfk0_sdU,42
-aiteamutils-0.2.47.dist-info/METADATA,sha256=BqLPd9L224PbbrWL3Z4zcblCVV3oRcarMpS3o2Iipb8,1718
-aiteamutils-0.2.47.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-aiteamutils-0.2.47.dist-info/RECORD,,
+aiteamutils/version.py,sha256=QOuBXnrQHPfVpSFpLi9n2XtFYH6CXKSg3p-oiLspDBg,42
+aiteamutils-0.2.48.dist-info/METADATA,sha256=9gDfB-4023TWWGai8JWqujsazybfHZPDqnzrc9equJw,1718
+aiteamutils-0.2.48.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+aiteamutils-0.2.48.dist-info/RECORD,,