o KDir&@sddlmZddlZddlZddlZddlmZmZddlm Z ddl m Z ddl Z ddlmZmZmZddlmZGdd d Ze d d Gd d d eeeeZdS)) annotationsN)Counter defaultdict)floor) versionadded)MovingWindowSupportSlidingWindowCounterSupportStorage)TimestampedSlidingWindowc@seZdZdddZdS) EntryexpiryintreturnNonecCst|_|j||_dSN)timeatimer )selfr rR/var/www/Datamplify/venv/lib/python3.10/site-packages/limits/aio/storage/memory.py__init__s zEntry.__init__N)r r rr)__name__ __module__ __qualname__rrrrrr sr z2.1)versioncseZdZdZdgZ dEdFfd d ZdGddZdHddZdIddZdIddZ e dJddZ dKdLd"d#Z dKdMd$d%Z dNd&d'ZdOd(d)Z dKdPd+d,ZdQd-d.ZdRd0d1Z dKdPd2d3ZdSd5d6ZdTd7d8ZdUdd?ZdWdAdBZdIdCdDZZS)X MemoryStoragez rate limit storage using :class:`collections.Counter` as an in memory storage for fixed & sliding window strategies, and a simple list to implement moving window strategy. z async+memoryNFuri str | Nonewrap_exceptionsbool_strrrc sBt|_ttj|_i|_i|_d|_t j |fd|i|dS)Nr) rstoragerasyncioLocklocks expirationseventstimersuperr)rrrr  __class__rrr*s  zMemoryStorage.__init__dict[str, limits.typing.Any]cCs|j}|d=|d=|S)Nr(r%)__dict__copyrstaterrr __getstate__4s zMemoryStorage.__getstate__r0cCs0|j|d|_ttj|_t|dSr) r-updater(rr#r$r% ensure_future_MemoryStorage__schedule_expiryr/rrr __setstate__:s  zMemoryStorage.__setstate__c s:ztt|jD]Y}|j|4IdHCtfdd|j|IdH}|j|gr<|j|d||j|<|j|dsQ|j|d|j|dWdIdHn 1IdHsawYq t|j D] }|j |tkr|j |d|j |d|j|dqnWdStj yYdSw)Ncstj| dddS)NcS|j Sr)r )eventrrrGzAMemoryStorage.__expire_events....key)bisect bisect_left)evtsnowrrr8Fs z/MemoryStorage.__expire_events..) rlistr'keysr%r# to_threadgetpopr&r"CancelledError)rr;cutoffrr?r__expire_events@s6  ( zMemoryStorage.__expire_eventscs*|jr |jrt||_dSdSr)r(doner# create_task_MemoryStorage__expire_eventsrrrr__schedule_expiryYszMemoryStorage.__schedule_expiry-type[Exception] | tuple[type[Exception], ...]cCstSr) ValueErrorrLrrrbase_exceptions]szMemoryStorage.base_exceptionsr;r floatamountr c s||IdH|IdH|j|4IdH$|j||7<|j||kr3t||j|<WdIdHn 1IdHsCwY|j||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 N)rDr4r%r"rr&)rr;r rSrrrincrcs(zMemoryStorage.incrc s||IdH|IdH|j|4IdHt|j||d|j|<WdIdHn 1IdHs7wY|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)rDr4r%maxr")rr;rSrrrdecrss(zMemoryStorage.decrcsP|j|dtkr!|j|d|j|d|j|d|j|dS)zB :param key: the key to get the counter value for rN)r&rDrr"rEr%rr;rrrrDs zMemoryStorage.getcs>|j|d|j|d|j|d|j|ddS)z> :param key: the key to clear rate limits for N)r"rEr&r'r%rWrrrclears zMemoryStorage.clearlimitc s||krdS|IdH|j|4IdHR|j|gt}z |j|||}Wn ty9d}Ynw|rO|j||krO WdIdHdSt|g||j|dd< WdIdHdS1IdHsowYdS)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)r4r%r' setdefaultr IndexErrorrr )rr;rYr rS timestampentryrrr acquire_entrys$   0zMemoryStorage.acquire_entrycs|j|tS)z; :param key: the key to get the expiry for )r&rDrrWrrr get_expiryszMemoryStorage.get_expirytuple[float, int]csPt}|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) cSr6r)r)r]rrrr8r9z1MemoryStorage.get_moving_window..r:rQr)rr'rDr<r=r)rr;rYr r\r'oldestrrrget_moving_windows zMemoryStorage.get_moving_windowc s||krdSt}||||\}}|||||IdH\}} } } || || } t| ||kr5dS|j|d||dIdH} || || } t| |kr[|||IdHdSdS)NF)rST)rsliding_window_keys_get_sliding_window_inforrTrV) rr;rYr rSr@ previous_key current_keyprevious_count previous_ttl current_countr weighted_countrrracquire_sliding_window_entrys( z*MemoryStorage.acquire_sliding_window_entrytuple[int, float, int, float]cs2t}||||\}}|||||IdHSr)rrdrerr;r r@rfrgrrrget_sliding_windows  z MemoryStorage.get_sliding_windowcs@t}||||\}}||IdH||IdHdSr)rrdrXrnrrrclear_sliding_windows z"MemoryStorage.clear_sliding_windowrfrgr@c sp||IdH}||IdH}|dkrtd}n d|||d|}d||d||}||||fS)NrrQ)rDrR) rrfrgr r@rhrjri current_ttlrrrres  z&MemoryStorage._get_sliding_window_infocsdS)z- check if storage is healthy TrrLrrrcheck szMemoryStorage.check int | NonecsDtt|jt|j}|j|j|j|j|Sr)rUlenr"r'rXr&r%)r num_itemsrrrresets    zMemoryStorage.resetcCsBz|jr|js|jWdSWdSWdSty YdSwr)r(rIcancel RuntimeErrorrLrrr__del__s zMemoryStorage.__del__)NF)rrrrr r!rr)rr,)r0r,rr)rr)rrN)rQ)r;r!r rRrSr rr )r;r!rSr rr )r;r!rr )r;r!rr) r;r!rYr r r rSr rr)r;r!rrR)r;r!rYr r r rr`)r;r!r r rrm)r;r!r r rr) rfr!rgr!r r r@rRrrm)rr)rrs)rrr__doc__STORAGE_SCHEMErr1r5rKr4propertyrPrTrVrDrXr^r_rbrlrorprerrrvry __classcell__rrr*rrs8          !    r) __future__rr#r<r collectionsrrmathrdeprecated.sphinxr limits.typinglimitslimits.aio.storage.baserrr limits.storage.baser r rrrrrs