aiteamutils 0.2.45__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

@@ -60,7 +60,19 @@ def get_service(name: str):
         try:
             repository_class, service_class = service_registry.get(name)
             repository = repository_class(db_service)
-            return service_class(repository, 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
+                
         except CustomException as e:
             raise e
         except Exception as e:
@@ -95,7 +107,7 @@ async def get_current_user(
             token,
             settings.JWT_SECRET,
             algorithms=[settings.JWT_ALGORITHM],
-            audience="ai-team"
+            audience=settings.TOKEN_AUDIENCE
         )
         user_ulid = payload.get("sub")
         if not user_ulid:

aiteamutils/security.py

@@ -351,7 +351,7 @@ async def create_jwt_token(
             token_data = {
                 # 등록 클레임
                 "iss": settings.TOKEN_ISSUER,
-                "sub": user_data["username"],
+                "sub": user_data["ulid"],
                 "aud": settings.TOKEN_AUDIENCE,
                 "exp": expires_at,
                 
@@ -376,7 +376,7 @@ async def create_jwt_token(
             expires_at = datetime.now(UTC) + timedelta(days=14)
             token_data = {
                 "iss": settings.TOKEN_ISSUER,
-                "sub": user_data["username"],
+                "sub": user_data["ulid"],
                 "exp": expires_at,
                 "token_type": token_type,
                 "user_ulid": user_data["ulid"]

aiteamutils/version.py

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiteamutils
-Version: 0.2.45
+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.45.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=cNV1ym6PIUQqwsQVtpgw92MKWefB3Bz781QSuC3XswQ,4317
+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=9gvEqDtE3RJaoCWqELPCjkg-IsSqZVrpMP6XPZaodWU,16024
+aiteamutils/security.py,sha256=KSZf3WJAlW0UXVM1GEHwuGy2x27UHwtMr8aq-GuIXZk,16016
 aiteamutils/validators.py,sha256=3N245cZFjgwtW_KzjESkizx5BBUDaJLbbxfNO4WOFZ0,7764
-aiteamutils/version.py,sha256=3bhfB0cuuuM8bgbYGCGGzo-Hv2L0C7qDN7ifS_5ye8w,42
-aiteamutils-0.2.45.dist-info/METADATA,sha256=8QyZGVPAX3i8lglFHb8fi3xSH7qRM9ZXG7tzhpy3_5E,1718
-aiteamutils-0.2.45.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-aiteamutils-0.2.45.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,,