o ?7i2 @sdZddlmZmZddlmZddlmZddlm Z ddl m Z ddl Z e eZde d efd d Zd efd d ZeZddZdS)z- Rate limiting middleware for API endpoints. )Limiter_rate_limit_exceeded_handler)get_remote_address)RateLimitExceeded)Request)settingsNrequestreturncCs|jdkrdSt|S)z Custom key function that skips rate limiting for OPTIONS requests. Args: request: FastAPI request object Returns: str: Client identifier for rate limiting OPTIONSskip)methodr)rr A/lsinfo/ai/hellotax_ai/base_platform/app/middleware/rate_limit.pycustom_key_funcs rc Cs^d}ttdr5tjr5ztj}tdtjWnty4}ztd|d}WYd}~njd}~wwttdrtjrz;ttdd}ttdd }|r]d |d tjd tj d |}n dtjd tj d |}tdtjd tj Wn!ty}ztd|d}WYd}~n d}~wwtdt t tj dtj dgtj|d}|S)zl Create and configure rate limiter instance. Returns: Limiter: Configured rate limiter z memory:// REDIS_URLzUsing Redis for rate limiting: zEFailed to configure Redis for rate limiting, falling back to memory: N REDIS_HOSTREDIS_PASSWORDREDIS_DBrz redis://:@:/zredis://zQRedis not configured, using in-memory rate limiting (not suitable for production)z/minutez/hour)key_funcdefault_limitsenabled storage_uri)hasattrrrloggerinfo Exceptionwarningrgetattr REDIS_PORTrrRATE_LIMIT_PER_MINUTERATE_LIMIT_PER_HOURRATE_LIMIT_ENABLED)reredis_passwordredis_dblimiterr r r get_limitersB  "    r)cCs"tjrt|j_|ttdSdS)zk Setup rate limiting for FastAPI application. Args: app: FastAPI application instance N)rr$r(stateadd_exception_handlerrr)appr r rsetup_rate_limitingPsr-)__doc__slowapirr slowapi.utilrslowapi.errorsrfastapir app.configrlogging getLogger__name__rstrrr)r(r-r r r rs     /