o rÃiý"ã@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úT/var/www/html/pca-backend/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. ÚmemoryNFÚ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__r3rcCs6|j |¡ttjƒ|_t d|j¡|_|j  ¡dS©Nr!) r0Úupdaterr#r$r%r(r)r*r+r2rrrÚ __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'ÚoldestrrrÚ__expire_events9s& ÿ€ù€€üzMemoryStorage.__expire_eventscCs,|j ¡st d|j¡|_|j ¡dSdSr5)r*Úis_aliver#r(r)r+©rrrrÚ__schedule_expiryIs þzMemoryStorage.__schedule_expiryú-type[Exception] | tuple[type[Exception], ...]cCstSr)Ú ValueErrorrGrrrÚ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 rMrrrÚ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@rOr%Úmaxr")rr=rMrrrÚ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"rCr%©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"rCr&r'r%rSrrrÚ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)rOr%r'Ú setdefaultrÚ IndexErrorrr )rr=rUr rMÚ timestampÚentryrrrÚ acquire_entry‡s"  ÿ÷ $ôzMemoryStorage.acquire_entrycCs|j |t ¡¡S)z; :param key: the key to get the expiry for )r&r@rrSrrrÚ 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) cSr8r)r)rYrrrr:³r;z1MemoryStorage.get_moving_window..r<rLr)rr'r@rArBr)rr=rUr rXr'rDrrrÚget_moving_window§s ÿzMemoryStorage.get_moving_windowc Cs¦||krdSt ¡}| |||¡\}}| ||||¡\}} } } || || } t| ƒ||kr1dS|j|d||d} || || } t| ƒ|krQ| ||¡dSdS)NFé)rMT)rÚsliding_window_keysÚ_get_sliding_window_inforrPrR) rr=rUr rMÚnowÚ previous_keyÚ current_keyÚprevious_countÚ previous_ttlÚ current_countrÚweighted_countrrrÚacquire_sliding_window_entry¸s&û  z*MemoryStorage.acquire_sliding_window_entryrbrcraútuple[int, float, int, float]c Csb| |¡}| |¡}|dkrtdƒ}n d|||d|}d||d||}||||fS)NrrL)r@r ) rrbrcr rardrfreÚ current_ttlrrrr`Ùs    z&MemoryStorage._get_sliding_window_infocCs*t ¡}| |||¡\}}| ||||¡Sr)rr_r`©rr=r rarbrcrrrÚget_sliding_windowész MemoryStorage.get_sliding_windowcCs2t ¡}| |||¡\}}| |¡| |¡dSr)rr_rTrkrrrÚclear_sliding_windowðs z"MemoryStorage.clear_sliding_windowcCsdS)z- check if storage is healthy TrrGrrrÚchecköszMemoryStorage.checkú int | NonecCsBtt|jƒt|jƒƒ}|j ¡|j ¡|j ¡|j ¡|Sr)rQÚlenr"r'rTr&r%)rÚ num_itemsrrrÚresetýs     zMemoryStorage.reset)NF)rrrrrr )r r/)r3r/r r)r r)r rI)rL)r=r r r rMrNr rN)r=r rMrNr rN)r=r r rN)r=r r r) r=r rUrNr rNrMrNr r)r=r r r )r=r rUrNr rNr r\) rbr rcr r rNrar r ri)r=r r rNr ri)r=r r rNr r)r r)r ro)rrrÚ__doc__ÚSTORAGE_SCHEMErr4r7r)rOÚpropertyrKrPrRr@rTrZr[r]rhr`rlrmrnrrÚ __classcell__rrr-rrs0          û !   r)Ú __future__rrAr#rÚ collectionsrrÚmathrÚ limits.typingÚlimitsÚlimits.storage.baserrrr r rrrrrÚs   ÿ