aiteamutils 0.2.45__tar.gz → 0.2.47__tar.gz

{aiteamutils-0.2.45 → aiteamutils-0.2.47}/PKG-INFO

@@ -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 → aiteamutils-0.2.47}/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-0.2.45 → aiteamutils-0.2.47}/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-0.2.45 → aiteamutils-0.2.47}/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-0.2.47/aiteamutils/version.py

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

aiteamutils-0.2.45/aiteamutils/version.py

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