aiteamutils 0.2.38__tar.gz → 0.2.39__tar.gz

{aiteamutils-0.2.38 → aiteamutils-0.2.39}/PKG-INFO

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

{aiteamutils-0.2.38 → aiteamutils-0.2.39}/aiteamutils/dependencies.py

@@ -3,6 +3,7 @@ from fastapi import Depends, status
 from fastapi.security import OAuth2PasswordBearer
 from jose import JWTError, jwt
 from sqlalchemy.ext.asyncio import AsyncSession
+import logging
 
 from .database import DatabaseService, get_database_service, get_db
 from .exceptions import CustomException, ErrorCode
@@ -12,10 +13,12 @@ class ServiceRegistry:
     """서비스 레지스트리를 관리하는 클래스"""
     def __init__(self):
         self._services: Dict[str, Tuple[Type, Type]] = {}
+        self._initialized = False
 
     def clear(self):
         """등록된 모든 서비스를 초기화합니다."""
         self._services.clear()
+        self._initialized = False
 
     def register(self, name: str, repository_class: Type, service_class: Type):
         """서비스를 레지스트리에 등록
@@ -28,13 +31,28 @@ class ServiceRegistry:
         Raises:
             CustomException: 이미 등록된 서비스인 경우
         """
-        if name in self._services:
+        try:
+            if name in self._services:
+                logging.warning(f"Service '{name}' is already registered. Skipping...")
+                return
+                
+            if not repository_class or not service_class:
+                raise CustomException(
+                    ErrorCode.INTERNAL_ERROR,
+                    detail=f"Invalid service classes for '{name}'",
+                    source_function="ServiceRegistry.register"
+                )
+                
+            self._services[name] = (repository_class, service_class)
+            logging.info(f"Service '{name}' registered successfully")
+            
+        except Exception as e:
             raise CustomException(
                 ErrorCode.INTERNAL_ERROR,
-                detail=f"service|{name}",
-                source_function="ServiceRegistry.register"
+                detail=f"Failed to register service '{name}': {str(e)}",
+                source_function="ServiceRegistry.register",
+                original_error=e
             )
-        self._services[name] = (repository_class, service_class)
 
     def get(self, name: str) -> Tuple[Type, Type]:
         """등록된 서비스를 조회
@@ -50,11 +68,19 @@ class ServiceRegistry:
         """
         if name not in self._services:
             raise CustomException(
-                ErrorCode.NOT_FOUND,
-                detail=f"service|{name}",
+                ErrorCode.SERVICE_NOT_REGISTERED,
+                detail=f"Service '{name}' is not registered",
                 source_function="ServiceRegistry.get"
             )
         return self._services[name]
+        
+    def is_initialized(self) -> bool:
+        """서비스 레지스트리 초기화 여부를 반환합니다."""
+        return self._initialized
+        
+    def set_initialized(self):
+        """서비스 레지스트리를 초기화 상태로 설정합니다."""
+        self._initialized = True
 
 # ServiceRegistry 초기화
 service_registry = ServiceRegistry()
@@ -73,13 +99,35 @@ def get_service(name: str):
     """
     def _get_service(db_service: DatabaseService = Depends(get_database_service)):
         try:
-            repository_class, service_class = service_registry.get(name)
-            repository = repository_class(db_service)
-            return service_class(repository)
+            # 서비스 레지스트리에서 클래스 조회
+            try:
+                repository_class, service_class = service_registry.get(name)
+            except CustomException as e:
+                raise CustomException(
+                    ErrorCode.SERVICE_NOT_REGISTERED,
+                    detail=f"Service '{name}' is not registered",
+                    source_function="dependencies.get_service",
+                    original_error=e
+                )
+            
+            # 서비스 인스턴스 생성
+            try:
+                repository = repository_class(db_service)
+                return service_class(repository)
+            except Exception as e:
+                raise CustomException(
+                    ErrorCode.INTERNAL_ERROR,
+                    detail=f"Failed to create service instance for '{name}': {str(e)}",
+                    source_function="dependencies.get_service",
+                    original_error=e
+                )
+                
+        except CustomException as e:
+            raise e
         except Exception as e:
             raise CustomException(
-                ErrorCode.SERVICE_NOT_REGISTERED,
-                detail=f"Failed to create service {name}: {str(e)}",
+                ErrorCode.INTERNAL_ERROR,
+                detail=f"Unexpected error while creating service '{name}': {str(e)}",
                 source_function="dependencies.get_service",
                 original_error=e
             )

{aiteamutils-0.2.38 → aiteamutils-0.2.39}/aiteamutils/exceptions.py

@@ -1,6 +1,6 @@
 """예외 처리 모듈."""
 import logging
-from enum import Enum
+from enum import Enum, IntEnum
 from typing import Dict, Any, Optional, Tuple, List
 from fastapi import Request
 from fastapi.responses import JSONResponse
@@ -69,6 +69,7 @@ class ErrorCode(Enum):
     NOT_FOUND = ErrorResponse(5001, "GENERAL_NOT_FOUND", 404, "리소스를 찾을 수 없습니다")
     INTERNAL_ERROR = ErrorResponse(5002, "GENERAL_INTERNAL_ERROR", 500, "내부 서버 오류")
     SERVICE_UNAVAILABLE = ErrorResponse(5003, "GENERAL_SERVICE_UNAVAILABLE", 503, "서비스를 사용할 수 없습니다")
+    SERVICE_NOT_REGISTERED = ErrorResponse(5003, "GENERAL_SERVICE_UNAVAILABLE", 503, "서비스를 사용할 수 없습니다")
 
 class CustomException(Exception):
     """사용자 정의 예외 클래스"""

aiteamutils-0.2.39/aiteamutils/version.py

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

aiteamutils-0.2.39/startup.py

@@ -0,0 +1,54 @@
+async def startup_event() -> None:
+    """Application startup event handler."""
+    try:
+        # 1. 설정 및 DB 초기화
+        init_settings(
+            jwt_secret=settings.JWT_SECRET,
+            jwt_algorithm=settings.JWT_ALGORITHM,
+            db_url=settings.CORE_DB_URL,
+            db_echo=settings.DB_ECHO,
+            db_pool_size=settings.DB_POOL_SIZE,
+            db_max_overflow=settings.DB_MAX_OVERFLOW,
+            db_pool_timeout=settings.DB_POOL_TIMEOUT,
+            db_pool_recycle=settings.DB_POOL_RECYCLE
+        )
+        db_service = get_database_service()
+
+        if not db_service or not db_service.engine:
+            raise RuntimeError("데이터베이스 서비스 초기화 실패")
+            
+        # 2. DB 연결 테스트
+        async with db_service.engine.begin() as conn:
+            await conn.run_sync(lambda _: None)
+            logging.info("데이터베이스 연결 성공")
+            
+        # 3. 테이블 생성
+        async with db_service.engine.begin() as conn:
+            await conn.run_sync(Base.metadata.create_all)
+            logging.info("데이터베이스 테이블 생성 완료")
+        
+        # 4. 서비스 등록
+        register_services()
+        service_registry.set_initialized()  # 서비스 등록 완료 표시
+
+        logging.info("서비스 등록 완료")
+        
+    except Exception as e:
+        logging.error(f"애플리케이션 시작 실패: {str(e)}")
+        raise RuntimeError(f"애플리케이션 시작 실패: {str(e)}")
+
+async def shutdown_event() -> None:
+    """Application shutdown event handler."""
+    try:
+        # 1. 서비스 레지스트리 초기화
+        service_registry.clear()
+        logging.info("서비스 레지스트리 초기화 완료")
+        
+        # 2. 데이터베이스 연결 종료
+        db_service = get_database_service()
+        if db_service and db_service.engine:
+            await db_service.engine.dispose()
+            logging.info("데이터베이스 연결 종료")
+    except Exception as e:
+        logging.error(f"애플리케이션 종료 중 오류 발생: {str(e)}")
+        raise 

aiteamutils-0.2.38/aiteamutils/version.py

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