o ;hZ&@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 rL/var/www/html/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%Z ensure_future_MemoryStorage__schedule_expiryr/rrr __setstate__:s  zMemoryStorage.__setstate__c s:ztt|jD]Y}tfdd|j|IdH}|j|4IdH3|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 )eventrrrFzAMemoryStorage.__expire_events....key)bisect bisect_left)Zevtsnowrrr7Es z/MemoryStorage.__expire_events..) rlistr'keysr#Z to_threadr%getpopr&r"ZCancelledError)rr:cutoffrr=rZ__expire_events@s6  (zMemoryStorage.__expire_eventscs*|jr |jrt||_dSdSr)r(doner#Z create_task_MemoryStorage__expire_eventsrrrrZ__schedule_expiryYszMemoryStorage.__schedule_expiry-type[Exception] | tuple[type[Exception], ...]cCstSr) ValueErrorrFrrrbase_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)rAr3r%r"rr&)rr:r rLrrrincrcs(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)rAr3r%maxr")rr:rLrrrdecrss(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&rArr"rBr%rr:rrrrAs zMemoryStorage.getcs>|j|d|j|d|j|d|j|ddS)z> :param key: the key to clear rate limits for N)r"rBr&r'r%rPrrrclears 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)r3r%r' setdefaultr IndexErrorrr )rr:rRr rL timestampentryrrr acquire_entrys$   0zMemoryStorage.acquire_entrycs|j|tS)z; :param key: the key to get the expiry for )r&rArrPrrr 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) cSr5r)r)rVrrrr7r8z1MemoryStorage.get_moving_window..r9rJr)rr'rAr;r<r)rr:rRr rUr'Zoldestrrrget_moving_windows zMemoryStorage.get_moving_windowc s||krdSt}||||\}}|||||IdH\}} } } || || } t| ||kr5dS|j|d||dIdH} || || } t| |kr[|||IdHdSdS)NF)rLT)rsliding_window_keys_get_sliding_window_inforrMrO) rr:rRr rLr> previous_key current_keyprevious_count previous_ttl current_countr Zweighted_countrrracquire_sliding_window_entrys( z*MemoryStorage.acquire_sliding_window_entrytuple[int, float, int, float]cs2t}||||\}}|||||IdHSr)rr\r]rr:r r>r^r_rrrget_sliding_windows  z MemoryStorage.get_sliding_windowcs@t}||||\}}||IdH||IdHdSr)rr\rQrerrrclear_sliding_windows z"MemoryStorage.clear_sliding_windowr^r_r>c sp||IdH}||IdH}|dkrtd}n d|||d|}d||d||}||||fS)NrrJ)rArK) rr^r_r r>r`rbraZ current_ttlrrrr]s  z&MemoryStorage._get_sliding_window_infocsdS)z- check if storage is healthy TrrFrrrcheck szMemoryStorage.check int | NonecsDtt|jt|j}|j|j|j|j|Sr)rNlenr"r'rQr&r%)r num_itemsrrrresets    zMemoryStorage.resetcCsBz|jr|js|jWdSWdSWdSty YdSwr)r(rDcancel RuntimeErrorrFrrr__del__s zMemoryStorage.__del__)NF)rrrrr r!rr)rr,)r0r,rr)rr)rrG)rJ)r:r!r rKrLr rr )r:r!rLr rr )r:r!rr )r:r!rr) r:r!rRr r r rLr rr)r:r!rrK)r:r!rRr r r rrY)r:r!r r rrd)r:r!r r rr) r^r!r_r!r r r>rKrrd)rr)rri)rrr__doc__ZSTORAGE_SCHEMErr1r4rEr3propertyrIrMrOrArQrWrXrZrcrfrgr]rhrlro __classcell__rrr*rrs8          !    r) __future__rr#r;r collectionsrrmathrZdeprecated.sphinxrZ limits.typingZlimitsZlimits.aio.storage.baserrr Zlimits.storage.baser r rrrrrs