aiteamutils 0.2.34__tar.gz → 0.2.36__tar.gz

{aiteamutils-0.2.34 → aiteamutils-0.2.36}/PKG-INFO

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

{aiteamutils-0.2.34 → aiteamutils-0.2.36}/aiteamutils/security.py

@@ -6,6 +6,7 @@ from fastapi import Request, HTTPException, status
 from functools import wraps
 from jose import jwt, JWTError
 from passlib.context import CryptContext
+import logging
 
 from .exceptions import CustomException, ErrorCode
 from .database import DatabaseService
@@ -204,6 +205,10 @@ def rate_limit(
     def decorator(func: Callable) -> Callable:
         @wraps(func)
         async def wrapper(*args, **kwargs):
+            logging.info(f"[rate_limit] Starting rate limit check for {func.__name__}")
+            logging.info(f"[rate_limit] Args: {args}")
+            logging.info(f"[rate_limit] Kwargs: {kwargs}")
+            
             # Request 객체 찾기
             request = None
             for arg in args:
@@ -216,6 +221,7 @@ def rate_limit(
                         request = arg
                         break
             if not request:
+                logging.error("[rate_limit] Request object not found in args or kwargs")
                 raise CustomException(
                     ErrorCode.INTERNAL_ERROR,
                     detail="Request object not found",
@@ -229,10 +235,13 @@ def rate_limit(
                 client_ip = request.client.host
                 rate_limit_key = f"rate_limit:{client_ip}:{func.__name__}"
             
+            logging.info(f"[rate_limit] Rate limit key: {rate_limit_key}")
+            
             now = datetime.now(UTC)
             
             # 현재 rate limit 정보 가져오기
             rate_info = rate_limits.get(rate_limit_key)
+            logging.info(f"[rate_limit] Current rate info: {rate_info}")
             
             if rate_info is None or (now - rate_info["start_time"]).total_seconds() >= window_seconds:
                 # 새로운 rate limit 설정
@@ -240,11 +249,13 @@ def rate_limit(
                     "count": 1,
                     "start_time": now
                 }
+                logging.info(f"[rate_limit] Created new rate limit: {rate_limits[rate_limit_key]}")
             else:
                 # 기존 rate limit 업데이트
                 if rate_info["count"] >= max_requests:
                     # rate limit 초과
                     remaining_seconds = window_seconds - (now - rate_info["start_time"]).total_seconds()
+                    logging.warning(f"[rate_limit] Rate limit exceeded. Remaining seconds: {remaining_seconds}")
                     raise RateLimitExceeded(
                         detail=rate_limit_key,
                         source_function=func.__name__,
@@ -253,15 +264,26 @@ def rate_limit(
                         window_seconds=window_seconds
                     )
                 rate_info["count"] += 1
+                logging.info(f"[rate_limit] Updated rate info: {rate_info}")
             
             try:
-                # 원래 함수 실행
-                return await func(*args, **kwargs)
+                logging.info(f"[rate_limit] Executing original function: {func.__name__}")
+                # db_service가 있는지 확인
+                for arg in args:
+                    if hasattr(arg, 'db_service'):
+                        logging.info(f"[rate_limit] Found db_service in args: {arg.db_service}")
+                for value in kwargs.values():
+                    if hasattr(value, 'db_service'):
+                        logging.info(f"[rate_limit] Found db_service in kwargs: {value.db_service}")
+                
+                result = await func(*args, **kwargs)
+                logging.info("[rate_limit] Function executed successfully")
+                return result
             except CustomException as e:
-                # CustomException은 그대로 전파
+                logging.error(f"[rate_limit] CustomException occurred: {str(e)}")
                 raise e
             except Exception as e:
-                # 다른 예외는 INTERNAL_ERROR로 래핑
+                logging.error(f"[rate_limit] Unexpected error occurred: {str(e)}")
                 raise CustomException(
                     ErrorCode.INTERNAL_ERROR,
                     detail=str(e),

aiteamutils-0.2.36/aiteamutils/version.py

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

aiteamutils-0.2.34/aiteamutils/version.py

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