aiteamutils 0.2.36__py3-none-any.whl → 0.2.37__py3-none-any.whl

aiteamutils/security.py

@@ -15,6 +15,9 @@ from .config import get_settings
 
 pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
 
+# 전역 rate limit 상태 저장
+_rate_limits: Dict[str, Dict[str, Any]] = {}
+
 class RateLimitExceeded(CustomException):
     """Rate limit 초과 예외."""
     
@@ -200,14 +203,10 @@ def rate_limit(
     key_func: Optional[Callable] = None
 ):
     """Rate limiting 데코레이터."""
-    rate_limits: Dict[str, Dict[str, Any]] = {}
-    
     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
@@ -240,16 +239,16 @@ def rate_limit(
             now = datetime.now(UTC)
             
             # 현재 rate limit 정보 가져오기
-            rate_info = rate_limits.get(rate_limit_key)
+            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 설정
-                rate_limits[rate_limit_key] = {
+                _rate_limits[rate_limit_key] = {
                     "count": 1,
                     "start_time": now
                 }
-                logging.info(f"[rate_limit] Created new rate limit: {rate_limits[rate_limit_key]}")
+                logging.info(f"[rate_limit] Created new rate limit: {_rate_limits[rate_limit_key]}")
             else:
                 # 기존 rate limit 업데이트
                 if rate_info["count"] >= max_requests:
@@ -268,14 +267,6 @@ def rate_limit(
             
             try:
                 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
@@ -290,7 +281,7 @@ def rate_limit(
                     source_function=func.__name__,
                     original_error=e
                 )
-                
+        
         return wrapper
     return decorator
 

aiteamutils/version.py

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

{aiteamutils-0.2.36.dist-info → aiteamutils-0.2.37.dist-info}/METADATA

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

{aiteamutils-0.2.36.dist-info → aiteamutils-0.2.37.dist-info}/RECORD

@@ -8,9 +8,9 @@ aiteamutils/database.py,sha256=CaH73g8PPNcmLCz4Xr0DBtNSrLcpRlyt0F1zO5Tigfo,33811
 aiteamutils/dependencies.py,sha256=GgN8sFM7qGNmOfpLaFrhuHOgPlBVXWrEgxv29jxjEgI,4226
 aiteamutils/enums.py,sha256=ipZi6k_QD5-3QV7Yzv7bnL0MjDz-vqfO9I5L77biMKs,632
 aiteamutils/exceptions.py,sha256=YV-ISya4wQlHk4twvGo16I5r8h22-tXpn9wa-b3WwDM,15231
-aiteamutils/security.py,sha256=U6EjpruMK6PB_gfUsX1kbrnv5pdkrg-vbJ266HDmelU,16563
+aiteamutils/security.py,sha256=9gvEqDtE3RJaoCWqELPCjkg-IsSqZVrpMP6XPZaodWU,16024
 aiteamutils/validators.py,sha256=3N245cZFjgwtW_KzjESkizx5BBUDaJLbbxfNO4WOFZ0,7764
-aiteamutils/version.py,sha256=xIPwqXH-2Fd8zt8Gh1JwJx_ISWM3WKZzxCLLIcTS6JU,42
-aiteamutils-0.2.36.dist-info/METADATA,sha256=0StxHP94IIzkjMUHtwHz2vJ-n9lFKhhMh3aSax2LZ3g,1718
-aiteamutils-0.2.36.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-aiteamutils-0.2.36.dist-info/RECORD,,
+aiteamutils/version.py,sha256=CmXYar-ylEXzX7J29Pftix8-ss-hnVGdG7rr9Oikfq0,42
+aiteamutils-0.2.37.dist-info/METADATA,sha256=TBQe6PWAOtODst1zDCR506fxkYQv-V4nWEo4yhAon08,1718
+aiteamutils-0.2.37.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+aiteamutils-0.2.37.dist-info/RECORD,,