aiteamutils 0.2.40__py3-none-any.whl → 0.2.42__py3-none-any.whl

aiteamutils/database.py

@@ -47,8 +47,21 @@ async def get_db() -> AsyncGenerator[AsyncSession, None]:
         CustomException: 세션 생성 실패 시
     """
     db_service = get_database_service()
+    if not db_service or not db_service.engine:
+        raise CustomException(
+            ErrorCode.DB_CONNECTION_ERROR,
+            detail="Database service or engine is not properly initialized",
+            source_function="get_db"
+        )
+    
     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"
+                )
             yield session
     except Exception as e:
         raise CustomException(
@@ -58,7 +71,7 @@ async def get_db() -> AsyncGenerator[AsyncSession, None]:
             original_error=e
         )
     finally:
-        if 'session' in locals():
+        if 'session' in locals() and session is not None:
             await session.close()
 
 def get_database_session(db: AsyncSession = Depends(get_db)) -> 'DatabaseService':
@@ -69,8 +82,26 @@ def get_database_session(db: AsyncSession = Depends(get_db)) -> 'DatabaseService
 
     Returns:
         DatabaseService: DatabaseService 인스턴스
+        
+    Raises:
+        CustomException: 데이터베이스 세션이 유효하지 않은 경우
     """
-    return DatabaseService(session=db)
+    if db is None:
+        raise CustomException(
+            ErrorCode.DB_CONNECTION_ERROR,
+            detail="Database session is not initialized",
+            source_function="get_database_session"
+        )
+    
+    db_service = DatabaseService(session=db)
+    if db_service.db is None:
+        raise CustomException(
+            ErrorCode.DB_CONNECTION_ERROR,
+            detail="Failed to initialize database session",
+            source_function="get_database_session"
+        )
+    
+    return db_service
 
 class DatabaseService:
     def __init__(
@@ -685,16 +716,33 @@ class DatabaseService:
         
         Returns:
             생성된 로그 엔티티
-        """
-        # 공통 필드 추가 (ULID를 문자열로 변환)
-        log_data["ulid"] = str(ULID())
         
-        # request가 있는 경우 user-agent와 ip 정보 추가
-        if request:
-            log_data["user_agent"] = request.headers.get("user-agent")
-            log_data["ip_address"] = request.headers.get("x-forwarded-for")
-        
-        return await self.create_entity(model, log_data)
+        Raises:
+            CustomException: 로그 생성 실패 시
+        """
+        try:
+            # 공통 필드 추가 (ULID를 문자열로 변환)
+            log_data["ulid"] = str(ULID())
+            
+            # request가 있는 경우 user-agent와 ip 정보 추가
+            if request:
+                log_data["user_agent"] = request.headers.get("user-agent")
+                log_data["ip_address"] = request.headers.get("x-forwarded-for")
+            
+            # 데이터 전처리
+            processed_data = self.preprocess_data(model, log_data)
+            entity = model(**processed_data)
+            
+            # 로그 엔티티 저장
+            self.db.add(entity)
+            await self.db.flush()
+            
+            return entity
+            
+        except Exception as e:
+            logging.error(f"Failed to create log: {str(e)}")
+            # 로그 생성 실패는 원래 작업에 영향을 주지 않도록 함
+            return None
 
     async def soft_delete(
         self,

aiteamutils/version.py

@@ -1,2 +1,2 @@
 """버전 정보"""
-__version__ = "0.2.40"
+__version__ = "0.2.42"

{aiteamutils-0.2.40.dist-info → aiteamutils-0.2.42.dist-info}/METADATA

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

{aiteamutils-0.2.40.dist-info → aiteamutils-0.2.42.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=tr0Yn8VPYA9QHiKCUzciVlQ2J1RAwNo2K9lGMH4rY3s,1334
 aiteamutils/config.py,sha256=kFKMeIx1KcuEwwx4VjZdCgoTOHCkG3ySYVJ0G6cvMoA,2849
-aiteamutils/database.py,sha256=ATFiwG_o6o1eeUa0VVQ_2E_ExC2Yg-Ui_eA62SL6vQk,35344
+aiteamutils/database.py,sha256=oI36mW94Hnn5laS3YcsAyuofFsQzbj3TXT-HRb3SYk8,37029
 aiteamutils/dependencies.py,sha256=hsJ-kc8ic4U6vKFtUIWjhBE1_Bm-sya5UqSb2zMH5oM,5731
 aiteamutils/enums.py,sha256=ipZi6k_QD5-3QV7Yzv7bnL0MjDz-vqfO9I5L77biMKs,632
 aiteamutils/exceptions.py,sha256=_lKWXq_ujNj41xN6LDE149PwsecAP7lgYWbOBbLOntg,15368
 aiteamutils/security.py,sha256=9gvEqDtE3RJaoCWqELPCjkg-IsSqZVrpMP6XPZaodWU,16024
 aiteamutils/validators.py,sha256=3N245cZFjgwtW_KzjESkizx5BBUDaJLbbxfNO4WOFZ0,7764
-aiteamutils/version.py,sha256=r4koaMfodnLfd4IMWK3LkRqImEOtPrTfnvj9iHsKNak,42
-aiteamutils-0.2.40.dist-info/METADATA,sha256=6EErFhk04VSuc_BO7JcenZSYnZYnp-gAJ82eVqFLiFg,1718
-aiteamutils-0.2.40.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-aiteamutils-0.2.40.dist-info/RECORD,,
+aiteamutils/version.py,sha256=smfyzeeMHy2PIy96IdATpp51v_Y3aEPYazLKs09uNSM,42
+aiteamutils-0.2.42.dist-info/METADATA,sha256=Bn-kxta6CUfeOYPzE03yrlm-4k5p5NqKOzln7YB9E7U,1718
+aiteamutils-0.2.42.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+aiteamutils-0.2.42.dist-info/RECORD,,