a a*jB"@sddlmZddlZddlZddlmZmZddlmZddl Z ddl m Z m Z mZmZddl mZmZmZGdddejZGd d d ee e eZdS) ) annotationsN)Counter defaultdict)floor)MovingWindowSupportSlidingWindowCounterSupportStorageTimestampedSlidingWindow)OptionalTypeUnioncs$eZdZdddfdd ZZS) LockableEntryfloatNone)expiryreturncs$t|_|j||_tdSN)timeatimersuper__init__)selfr __class__Q/var/www/html/assistant/venv/lib/python3.9/site-packages/limits/storage/memory.pyrs  zLockableEntry.__init__)__name__ __module__ __qualname__r __classcell__rrrrr sr csVeZdZdZdgZd;ddddfd d Zed d d dZdd ddZdd ddZ ddddddd"d#d$Zdddd%d&Zdddd'd(d)Zdddd*d+d,d-Zd?dddddd"d.d/Zddddd0d1d2d3Zddd0d'd4d5Zdd d6d7Zd8d d9d:ZZS)@ MemoryStoragez rate limit storage using :class:`collections.Counter` as an in memory storage for fixed and elastic window strategies, and a simple list to implement moving window strategy. ZmemoryNFz Optional[str]boolstr)uriwrap_exceptions_c sVt|_ttj|_i|_i|_td|j |_ |j t j |fd|i|dS)N{Gz?r$)rstorager threadingRLocklocks expirationseventsTimer_MemoryStorage__expire_eventstimerstartrr)rr#r$r%rrrr%s  zMemoryStorage.__init__z3Union[Type[Exception], tuple[Type[Exception], ...]])rcCstSr) ValueErrorrrrrbase_exceptions0szMemoryStorage.base_exceptionsrc Cst|jD]h}t|j|D]T}|<|jtkrV||j|vrV|j||Wdq 1sj0Yq qt|jD]@}|j|tkr|j|d|j|d|j |dqdSr) listr,keysrrremover+r'popr*)rkeyeventrrrZ__expire_events6s2zMemoryStorage.__expire_eventscCs(|js$td|j|_|jdS)Nr&)r/is_aliver(r-r.r0r2rrrZ__schedule_expiryCs zMemoryStorage.__schedule_expiryrint)r8relastic_expiryamountrcCs||||j|F|j||7<|sB|j||krTt||j|<Wdn1sh0Y|j|dS)aD increments the counter for a given rate limit key :param key: the key to increment :param expiry: amount in seconds for the key to expire in :param elastic_expiry: whether to keep extending the rate limit window every hit. :param amount: the number to increment by Nr)get_MemoryStorage__schedule_expiryr*r'rr+)rr8rr=r>rrrincrHs  0zMemoryStorage.incr)r8r>rcCsd||||j|*t|j||d|j|<Wdn1sL0Y|j|dS)z decrements the counter for a given rate limit key :param key: the key to decrement :param amount: the number to decrement by rN)r?r@r*maxr')rr8r>rrrdecr^s   8zMemoryStorage.decr)r8rcCsN|j|dtkr@|j|d|j|d|j|d|j|dS)zB :param key: the key to get the counter value for rN)r+r?rr'r7r*rr8rrrr?ls zMemoryStorage.getcCs<|j|d|j|d|j|d|j|ddS)z> :param key: the key to clear rate limits for N)r'r7r+r,r*rDrrrclearxszMemoryStorage.clear)r8limitrr>rcs||kr dS|j|g|t}z|j|||}WntyVd}Yn0|rn|j|krndSfddt|D|j|dd<dSdS)z :param key: rate limit key to acquire an entry in :param limit: amount of entries allowed :param expiry: expiry of the entry :param amount: the number of entries to acquire FNcsg|] }tqSr)r ).0r%rrr z/MemoryStorage.acquire_entry..rT)r, setdefaultr@r IndexErrorrrange)rr8rFrr> timestampentryrrHr acquire_entrys  $zMemoryStorage.acquire_entrycCs|j|tS)z; :param key: the key to get the expiry for )r+r?rrDrrr get_expiryszMemoryStorage.get_expiry)r8rrcs6t|j|r2tfdd|j|DSdS)z returns the number of entries already acquired :param key: rate limit key to acquire an entry in :param expiry: expiry of the entry csg|]}|jkr|qSr)r)rGkrrNrrrIrJz2MemoryStorage.get_num_acquired..r)rr,r?len)rr8rrrSrget_num_acquireds  zMemoryStorage.get_num_acquiredztuple[float, int])r8rFrrcCsVt}|||}|j|gdddD] }|j||kr,|j|fSq,||fS)z returns the starting point and the number of entries in the moving window :param key: rate limit key :param expiry: expiry of entry :return: (start of window, number of acquired entries) N)rrUr,r?r)rr8rFrrNZacquireditemrrrget_moving_windows  zMemoryStorage.get_moving_windowc Cs||kr dSt}||||\}}|||||\}} } } || || } t| ||krbdS|j|d||d} || || } t| |kr|||dSdSdS)NF)r>T)rsliding_window_keys_get_sliding_window_inforrArC) rr8rFrr>now previous_key current_keyprevious_count previous_ttl current_countr%Zweighted_countrrracquire_sliding_window_entrys&  z*MemoryStorage.acquire_sliding_window_entryztuple[int, float, int, float])r]r^rr\rc Csb||}||}|dkr&td}nd|||d|}d||d||}||||fS)Nrr;)r?r) rr]r^rr\r_rar`Z current_ttlrrrr[s   z&MemoryStorage._get_sliding_window_infocCs*t}||||\}}|||||Sr)rrZr[)rr8rr\r]r^rrrget_sliding_windowsz MemoryStorage.get_sliding_windowcCsdS)z- check if storage is healthy Trr2rrrcheckszMemoryStorage.checkz Optional[int]cCsBtt|jt|j}|j|j|j|j|Sr)rBrTr'r,rEr+r*)r num_itemsrrrresets     zMemoryStorage.reset)NF)Fr;)r;)r;)r;)rrr__doc__ZSTORAGE_SCHEMErpropertyr3r.r@rArCr?rErPrQrUrXrbr[rcrdrfrrrrrr s.    "r ) __future__rr(r collectionsrrmathrZ limits.typingZlimitsZlimits.storage.baserrrr r r r _RLockr r rrrrs