a a*j@sXddlZddlmZmZmZmZmZddlmZm Z Gddde Z Gddde Z dS)N)CallableIteratorListOptionalUnion) RateLimitItem parse_manyc @seZdZdZeedefeeeedeffe ee eeeeedeffeede fee ede ffe dd ddZ e e ddd Ze edd d ZdS) Limitz@ simple wrapper to encapsulate limits and their context .N) limitkey_funcscope per_methodmethods error_message exempt_whencostoverride_defaultsreturnc Cs:||_||_||_||_||_||_||_||_| |_dSN) r r _Limit__scoper rrrrr) selfr r r r rrrrrrL/var/www/html/assistant/venv/lib/python3.9/site-packages/slowapi/wrappers.py__init__ s zLimit.__init__rcCs|jdur|SdS)zg Check if the limit is exempt. Return True to exempt the route from the limit. NF)rrrrr is_exempt"szLimit.is_exemptcCs.|jdurdSt|jr$|tjS|jSdS)N)rcallablerequestZendpointrrrrr *s  z Limit.scope)__name__ __module__ __qualname____doc__rrstrrrboolrintrpropertyrr rrrrr s    r c @seZdZdZeeedeffedefeeeedeffeee eeeeedeffeedefee ede ffed ddZ e e dddZd d Zd S) LimitGroupzb represents a group of related limits either from a string or a callable that returns one .) limit_provider key_functionr r rrrrrc CsR||_||_||_||_|r*dd|Dp,||_||_||_||_| |_d|_ dS)NcSsg|] }|qSr)lower).0mrrr Mz'LimitGroup.__init__..) _LimitGroup__limit_provider_LimitGroup__scoper*r rrrrrr) rr)r*r r rrrrrrrrr=s zLimitGroup.__init__rc cst|jrzdt|jjvrpdt|jjvsJJd|jjd|jdur\t d|||j}q|}n|j}t |}|D].}t ||j|j |j |j|j|j|j|j VqdS)NkeyrzLimit provider function z needs a `request` argumentz`request` object can't be None)rr0inspect signature parameterskeysr*r r Exceptionrr r1r rrrrr)rZ limit_rawZ limit_itemsr rrr__iter__Ts0   zLimitGroup.__iter__cCs ||_|Sr)r)rrrrr with_requestoszLimitGroup.with_requestN)r r!r"r#rr$rrr%rr&rrr r8r9rrrrr(8s   r() r3typingrrrrrZlimitsrrobjectr r(rrrrs1