o ú•;hý"ã@sˆddlmZddlZddlZddlZddlmZmZddlm Z ddl Z ddl m Z mZmZmZGdd„dƒZGdd „d ee eeƒZdS) é)Ú annotationsN)ÚCounterÚ defaultdict)Úfloor)ÚMovingWindowSupportÚSlidingWindowCounterSupportÚStorageÚTimestampedSlidingWindowc@seZdZddd„ZdS) ÚEntryÚexpiryÚfloatÚreturnÚNonecCst ¡|_|j||_dS©N)ÚtimeÚatimer )Úselfr ©rúH/var/www/html/venv/lib/python3.10/site-packages/limits/storage/memory.pyÚ__init__s zEntry.__init__N)r r r r)Ú__name__Ú __module__Ú __qualname__rrrrrr sr csìeZdZdZdgZdCdD‡fd d „ ZdEdd„ZdFdd„ZdGdd„ZdGdd„Z e dHdd„ƒZ dIdJd"d#„Z dIdKd$d%„Z dLd&d'„ZdMd(d)„ZdIdNd+d,„ZdOd-d.„ZdPd0d1„Z dIdNd2d3„ZdQd8d9„ZdRd:d;„ZdSdd?„ZdUdAdB„Z‡ZS)VÚ MemoryStoragezÀ rate limit storage using :class:`collections.Counter` as an in memory storage for fixed and sliding window strategies, and a simple list to implement moving window strategy. ZmemoryNFÚuriú str | NoneÚwrap_exceptionsÚboolÚ_Ústrc sVtƒ|_ttjƒ|_i|_i|_t d|j ¡|_ |j   ¡t ƒj |fd|i|¤ŽdS)Nç{®Gáz„?r)rÚstoragerÚ threadingÚRLockÚlocksÚ expirationsÚeventsÚTimerÚ_MemoryStorage__expire_eventsÚtimerÚstartÚsuperr)rrrr©Ú __class__rrr$s  zMemoryStorage.__init__r údict[str, limits.typing.Any]cCs|j ¡}|d=|d=|S)Nr)r$)Ú__dict__Úcopy©rÚstaterrrÚ __getstate__-s zMemoryStorage.__getstate__r2rcCs6|j |¡ttjƒ|_t d|j¡|_|j  ¡dS©Nr ) r/Úupdaterr"r#r$r'r(r)r*r1rrrÚ __setstate__3s  zMemoryStorage.__setstate__c Csòt|j ¡ƒD]G}|j|8|j |g¡}r1tj|t ¡ dd„d}|j|d|…|j|<|j |d¡s?|j |d¡Wdƒn1sIwYqt|j  ¡ƒD] }|j |t ¡krv|j  |d¡|j  |d¡|j |d¡qVdS)NcSó|j Sr)r )ÚeventrrrÚ>óz/MemoryStorage.__expire_events..©Úkey) Úlistr&Úkeysr$ÚgetÚbisectÚ bisect_leftrÚpopr%r!)rr<r&ÚoldestrrrZ__expire_events9s& ÿ€ù€€üzMemoryStorage.__expire_eventscCs,|j ¡st d|j¡|_|j ¡dSdSr4)r)Úis_aliver"r'r(r*©rrrrZ__schedule_expiryIs þzMemoryStorage.__schedule_expiryú-type[Exception] | tuple[type[Exception], ...]cCstSr)Ú ValueErrorrErrrÚbase_exceptionsNszMemoryStorage.base_exceptionsér<r r ÚamountÚintcCs|| |¡| ¡|j|!|j||7<|j||kr(t ¡||j|<Wdƒn1s2wY|j |d¡S)zá 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 amount: the number to increment by Nr)r?Ú_MemoryStorage__schedule_expiryr$r!rr%)rr<r rJrrrÚincrTs  €ýzMemoryStorage.incrcCsd| |¡| ¡|j|t|j||dƒ|j|<Wdƒn1s&wY|j |d¡S)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?rLr$Úmaxr!)rr<rJrrrÚdecrds  ÿzMemoryStorage.decrcCsN|j |d¡t ¡kr |j |d¡|j |d¡|j |d¡|j |d¡S)zB :param key: the key to get the counter value for rN)r%r?rr!rBr$©rr<rrrr?rs zMemoryStorage.getcCs<|j |d¡|j |d¡|j |d¡|j |d¡dS)z> :param key: the key to clear rate limits for N)r!rBr%r&r$rPrrrÚclear~szMemoryStorage.clearÚlimitc CsÊ||krdS| ¡|j|L|j |g¡t ¡}z |j|||}Wn ty1d}Ynw|rD|j||krD WdƒdSt|ƒg||j|dd…< WdƒdS1s^wYdS)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 FNrT)rLr$r&Ú setdefaultrÚ IndexErrorrr )rr<rRr rJÚ timestampÚentryrrrÚ acquire_entry‡s"  ÿ÷ $ôzMemoryStorage.acquire_entrycCs|j |t ¡¡S)z; :param key: the key to get the expiry for )r%r?rrPrrrÚ get_expiry szMemoryStorage.get_expiryútuple[float, int]cCsNt ¡}|j |g¡}r#tj||| dd„d}||dj|fS|dfS)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) cSr7r)r)rVrrrr9³r:z1MemoryStorage.get_moving_window..r;rIr)rr&r?r@rAr)rr<rRr rUr&rCrrrÚget_moving_window§s ÿzMemoryStorage.get_moving_windowc Cs¦||krdSt ¡}| |||¡\}}| ||||¡\}} } } || || } t| ƒ||kr1dS|j|d||d} || || } t| ƒ|krQ| ||¡dSdS)NFé)rJT)rÚsliding_window_keysÚ_get_sliding_window_inforrMrO) rr<rRr rJÚnowÚ previous_keyÚ current_keyÚprevious_countÚ previous_ttlÚ current_countrZweighted_countrrrÚacquire_sliding_window_entry¸s&û  z*MemoryStorage.acquire_sliding_window_entryr_r`r^útuple[int, float, int, float]c Csb| |¡}| |¡}|dkrtdƒ}n d|||d|}d||d||}||||fS)NrrI)r?r ) rr_r`r r^rarcrbZ current_ttlrrrr]Ùs    z&MemoryStorage._get_sliding_window_infocCs*t ¡}| |||¡\}}| ||||¡Sr)rr\r]©rr<r r^r_r`rrrÚget_sliding_windowész MemoryStorage.get_sliding_windowcCs2t ¡}| |||¡\}}| |¡| |¡dSr)rr\rQrfrrrÚclear_sliding_windowðs z"MemoryStorage.clear_sliding_windowcCsdS)z- check if storage is healthy TrrErrrÚchecköszMemoryStorage.checkú int | NonecCsBtt|jƒt|jƒƒ}|j ¡|j ¡|j ¡|j ¡|Sr)rNÚlenr!r&rQr%r$)rÚ num_itemsrrrÚresetýs     zMemoryStorage.reset)NF)rrrrrr)r r.)r2r.r r)r r)r rF)rI)r<rr r rJrKr rK)r<rrJrKr rK)r<rr rK)r<rr r) r<rrRrKr rKrJrKr r)r<rr r )r<rrRrKr rKr rY) r_rr`rr rKr^r r re)r<rr rKr re)r<rr rKr r)r r)r rj)rrrÚ__doc__ZSTORAGE_SCHEMErr3r6r(rLÚpropertyrHrMrOr?rQrWrXrZrdr]rgrhrirmÚ __classcell__rrr,rrs0          û !   r)Ú __future__rr@r"rÚ collectionsrrÚmathrZ limits.typingZlimitsZlimits.storage.baserrrr r rrrrrÚs   ÿ