a a*j#@sddlmZddlZddlZddlmZmZddlmZddl m Z ddl Z ddl mZmZmZddlmZddl mZmZmZGd d d ejZe d d Gd ddeeeeZdS)) annotationsN)Counter defaultdict)floor) versionadded)MovingWindowSupportSlidingWindowCounterSupportStorage)TimestampedSlidingWindow)OptionalTypeUnioncs$eZdZdddfdd ZZS) LockableEntryintNone)expiryreturncs$t|_|j||_tdSN)timeatimersuper__init__)selfr __class__U/var/www/html/assistant/venv/lib/python3.9/site-packages/limits/aio/storage/memory.pyrs  zLockableEntry.__init__)__name__ __module__ __qualname__r __classcell__rrrrrsrz2.1)versioncsXeZdZdZdgZd;ddddd fd d Zed d ddZdd ddZdd ddZ ddddddd"d#d$Zdddd%d&Zdddd'd(d)Zdddd*d+d,d-Zd?dddddd"d.d/Zddd0d'd1d2Zddddd0d3d4d5Zdd 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. z async+memoryNFz Optional[str]boolstrr)uriwrap_exceptions_rc sBt|_ttj|_i|_i|_d|_t j |fd|i|dS)Nr&) rstoragerasyncioLocklocks expirationseventstimerrr)rr%r&r'rrrr+s  zMemoryStorage.__init__z3Union[Type[Exception], tuple[Type[Exception], ...]])rcCstSr) ValueErrorrrrrbase_exceptions5szMemoryStorage.base_exceptionsc s|jD]|}t|j|D]h}|4IdHB|jtkrZ||j|vrZ|j||WdIdHq1IdHsz0Yqq t|jD]@}|j|tkr|j|d|j|d|j |dqdSr) r-keyslistrrremover,r(popr+)rkeyeventrrrZ__expire_events;s>zMemoryStorage.__expire_eventscs$|jr|jr t||_dSr)r.doner)Z create_task_MemoryStorage__expire_eventsr0rrrZ__schedule_expiryHszMemoryStorage.__schedule_expiryfloatr)r6relastic_expiryamountrc s||IdH|IdH|j|4IdHL|j||7<|sV|j||krht||j|<WdIdHq1IdHs0Y|j||S)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 N)get_MemoryStorage__schedule_expiryr+r(rr,)rr6rr<r=rrrincrLs <zMemoryStorage.incr)r6r=rc s||IdH|IdH|j|4IdH0t|j||d|j|<WdIdHqv1IdHsl0Y|j||S)z decrements the counter for a given rate limit key. 0 is the minimum allowed value. :param amount: the number to increment by Nr)r>r?r+maxr()rr6r=rrrdecrbs DzMemoryStorage.decr)r6rcsN|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(r5r+rr6rrrr>os zMemoryStorage.getcs<|j|d|j|d|j|d|j|ddS)z> :param key: the key to clear rate limits for N)r(r5r,r-r+rCrrrclearzszMemoryStorage.clear)r6limitrr=rcs||kr dS|j|g|IdHt}z|j|||}Wnty\d}Yn0|rt|j|krtdSfddt|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)rr6rErr= timestampentryrrGr acquire_entrys   $zMemoryStorage.acquire_entrycs|j|tS)z; :param key: the key to get the expiry for )r,r>rrCrrr get_expiryszMemoryStorage.get_expiry)r6rrcs6t|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)rFkrrMrrrHrIz2MemoryStorage.get_num_acquired..r)rr-r>len)rr6rrrRrget_num_acquireds  zMemoryStorage.get_num_acquiredztuple[float, int])r6rErrcs\t}|||IdH}|j|gdddD] }|j||kr2|j|fSq2||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)rrTr-r>r)rr6rErrMZacquireditemrrrget_moving_windows zMemoryStorage.get_moving_windowc s||kr dSt}||||\}}|||||IdH\}} } } || || } t| ||krhdS|j|d||dIdH} || || } t| |kr|||IdHdSdSdS)NF)r=T)rsliding_window_keys_get_sliding_window_inforr@rB) rr6rErr=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]cs0t}||||\}}|||||IdHSr)rrYrZ)rr6rr[r\r]rrrget_sliding_windows z MemoryStorage.get_sliding_window)r\r]rr[rc sn||IdH}||IdH}|dkr2td}nd|||d|}d||d||}||||fS)Nrr:)r>r;) rr\r]rr[r^r`r_Z current_ttlrrrrZs z&MemoryStorage._get_sliding_window_infocsdS)z- check if storage is healthy Trr0rrrcheckszMemoryStorage.checkz Optional[int]csBtt|jt|j}|j|j|j|j|Sr)rArSr(r-rDr,r+)r num_itemsrrrreset s     zMemoryStorage.reset)NF)Fr:)r:)r:)r:)rrr__doc__ZSTORAGE_SCHEMErpropertyr1r9r?r@rBr>rDrOrPrTrWrarbrZrcrer rrrrr"s0     " r") __future__rr)r collectionsrrmathrZdeprecated.sphinxrZ limits.typingZlimitsZlimits.aio.storage.baserrr Zlimits.storage.baser r r r r*rr"rrrrs