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

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

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

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

@@ -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.dist-info → aiteamutils-0.2.47.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=xUvxSRG49AY5DIWkucKMsaMMtfg5GumgJRwwwiUXm5w,40173
+aiteamutils/dependencies.py,sha256=UC19hLf5LHM4fdqVQHo7fHxytD47lkXuiSQvU2Vd4s0,4827
 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=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,,