o KDi"@sddlmZddlZddlZddlZddlmZmZddlm Z ddl Z ddl m Z mZmZmZGdddZGdd d ee eeZdS) ) annotationsN)Counter defaultdict)floor)MovingWindowSupportSlidingWindowCounterSupportStorageTimestampedSlidingWindowc@seZdZdddZdS) EntryexpiryfloatreturnNonecCst|_|j||_dSN)timeatimer )selfr rN/var/www/Datamplify/venv/lib/python3.10/site-packages/limits/storage/memory.py__init__s zEntry.__init__N)r r r r)__name__ __module__ __qualname__rrrrrr sr cseZdZdZdgZdCdDfd d ZdEddZdFddZdGddZdGddZ e dHddZ dIdJd"d#Z dIdKd$d%Z dLd&d'ZdMd(d)ZdIdNd+d,ZdOd-d.ZdPd0d1Z dIdNd2d3ZdQd8d9ZdRd:d;ZdSdd?ZdUdAdBZZS)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. memoryNFuri str | Nonewrap_exceptionsbool_strc sVt|_ttj|_i|_i|_td|j |_ |j t j |fd|i|dS)N{Gz?r)rstorager threadingRLocklocks expirationseventsTimer_MemoryStorage__expire_eventstimerstartsuperr)rrrr __class__rrr$s  zMemoryStorage.__init__r dict[str, limits.typing.Any]cCs|j}|d=|d=|S)Nr*r%)__dict__copyrstaterrr __getstate__-s zMemoryStorage.__getstate__r3rcCs6|j|ttj|_td|j|_|j dSNr!) r0updaterr#r$r%r(r)r*r+r2rrr __setstate__3s  zMemoryStorage.__setstate__c Cst|jD]G}|j|8|j|g}r1tj|t ddd}|j|d||j|<|j|ds?|j|dWdn1sIwYqt|j D] }|j |tkrv|j |d|j |d|j|dqVdS)NcS|j Sr)r )eventrrr>z/MemoryStorage.__expire_events..key) listr'keysr%getbisect bisect_leftrpopr&r")rr=r'oldestrrr__expire_events9s& zMemoryStorage.__expire_eventscCs,|jstd|j|_|jdSdSr5)r*is_aliver#r(r)r+rrrr__schedule_expiryIs zMemoryStorage.__schedule_expiry-type[Exception] | tuple[type[Exception], ...]cCstSr) ValueErrorrGrrrbase_exceptionsNszMemoryStorage.base_exceptionsr=r r amountintcCs|||||j|!|j||7<|j||kr(t||j|<Wdn1s2wY|j|dS)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 rMrrrincrTs  zMemoryStorage.incrcCsd||||j|t|j||d|j|<Wdn1s&wY|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@rOr%maxr")rr=rMrrrdecrds  zMemoryStorage.decrcCsN|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"rCr%rr=rrrr@rs zMemoryStorage.getcCs<|j|d|j|d|j|d|j|ddS)z> :param key: the key to clear rate limits for N)r"rCr&r'r%rSrrrclear~szMemoryStorage.clearlimitc Cs||krdS||j|L|j|gt}z |j|||}Wn ty1d}Ynw|rD|j||krD WddSt|g||j|dd< WddS1s^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 timestampentryrrr acquire_entrys"   $zMemoryStorage.acquire_entrycCs|j|tS)z; :param key: the key to get the expiry for )r&r@rrSrrr get_expiryszMemoryStorage.get_expirytuple[float, int]cCsNt}|j|g}r#tj||| ddd}||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'rDrrrget_moving_windows zMemoryStorage.get_moving_windowc Cs||krdSt}||||\}}|||||\}} } } || || } t| ||kr1dS|j|d||d} || || } t| |krQ|||dSdS)NF)rMT)rsliding_window_keys_get_sliding_window_inforrPrR) rr=rUr rMnow previous_key current_keyprevious_count previous_ttl current_countrweighted_countrrracquire_sliding_window_entrys&  z*MemoryStorage.acquire_sliding_window_entryrbrcratuple[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 rarbrcrrrget_sliding_windowsz MemoryStorage.get_sliding_windowcCs2t}||||\}}||||dSr)rr_rTrkrrrclear_sliding_windows z"MemoryStorage.clear_sliding_windowcCsdS)z- check if storage is healthy TrrGrrrcheckszMemoryStorage.check int | NonecCsBtt|jt|j}|j|j|j|j|Sr)rQlenr"r'rTr&r%)r num_itemsrrrresets     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)rOpropertyrKrPrRr@rTrZr[r]rhr`rlrmrnrr __classcell__rrr-rrs0          !   r) __future__rrAr#r collectionsrrmathr limits.typinglimitslimits.storage.baserrrr r rrrrrs