aiteamutils 0.2.50__py3-none-any.whl → 0.2.51__py3-none-any.whl

aiteamutils/database.py

@@ -997,62 +997,30 @@ class DatabaseService:
             )
 
 async def get_db() -> AsyncGenerator[AsyncSession, None]:
-    """데이터베이스 세션을 생성하고 반환하는 비동기 제너레이터.
-    
+    """데이터베이스 세션 의존성
+
     Yields:
         AsyncSession: 데이터베이스 세션
-        
+
     Raises:
-        CustomException: 세션 생성 실패 시
+        CustomException: 데이터베이스 연결 오류
     """
-    if not await DatabaseServiceManager.is_initialized():
-        raise CustomException(
-            ErrorCode.DB_CONNECTION_ERROR,
-            detail="Database service is not initialized",
-            source_function="get_db"
-        )
-    
-    db_service = await DatabaseServiceManager.get_instance()
-    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"
-        )
+    db_service = await get_database_service()
     
-    try:
-        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"
-            )
-        
-        # 세션이 유효한지 확인
+    async with db_service.get_session() as session:
         try:
+            # 세션이 유효한지 확인
             await session.execute(select(1))
+            yield session
         except Exception as e:
-            await session.close()
+            if isinstance(e, CustomException):
+                raise e
             raise CustomException(
                 ErrorCode.DB_CONNECTION_ERROR,
-                detail="Database session is not valid",
+                detail=f"Failed to get database session: {str(e)}",
                 source_function="get_db",
                 original_error=e
             )
-            
-        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)}",
-            source_function="get_db",
-            original_error=e
-        )
-    finally:
-        await session.close()
 
 async def get_database_service() -> DatabaseService:
     """DatabaseService 의존성

aiteamutils/version.py

@@ -1,2 +1,2 @@
 """버전 정보"""
-__version__ = "0.2.50"
+__version__ = "0.2.51"

{aiteamutils-0.2.50.dist-info → aiteamutils-0.2.51.dist-info}/METADATA

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

{aiteamutils-0.2.50.dist-info → aiteamutils-0.2.51.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=rEKuZ8_3eBjP1KpTNVgWrVLAtKX_IkZJzul2pa8bDlg,40075
+aiteamutils/database.py,sha256=Be0hr833X3zMz2gaJYugvE9E7o05uBmXKPhZiGFJQTM,38984
 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=lxXpdmBOudu-YVW902wEuXxWiy_elYifbZv5G4sOhdo,42
-aiteamutils-0.2.50.dist-info/METADATA,sha256=WjFxSSNzciuhgenfILvBNXtMKSY-HIAd6m88tZXrEiU,1718
-aiteamutils-0.2.50.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-aiteamutils-0.2.50.dist-info/RECORD,,
+aiteamutils/version.py,sha256=8MKWHJHM8zYlzctvZN4uCH_rdHayPQd-Sxyv9JJ2554,42
+aiteamutils-0.2.51.dist-info/METADATA,sha256=u7kI-u4xQZcrWK20EtNRzzzJNJvZ22GRD3eI8I7-u-M,1718
+aiteamutils-0.2.51.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+aiteamutils-0.2.51.dist-info/RECORD,,