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

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

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

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

@@ -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.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=PZOggFB8KfkoCxzHhP5GGeBjee-FM6Hr_no8c7uhAMU,40129
-aiteamutils/dependencies.py,sha256=5thwIWSEMTEQVTchh0GKb4R9zp_2l5ndp1UcIeHNRJ0,4426
+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=XFXbPLB5U50FoKWMU59F15d-62BCu1k0wvKYNUZHDFk,42
-aiteamutils-0.2.46.dist-info/METADATA,sha256=sfCXXCOOOHxx5hfRgLROi9ePXoOVZIhhsWQ22w1gZuY,1718
-aiteamutils-0.2.46.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-aiteamutils-0.2.46.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,,