a a*j@sddlmZddlZddlZddlmZmZddlmZddl m Z ddl m Z m Z mZmZmZmZmZmZddlmZddd d d ZGd d d ee dZGdddeZGdddeZGdddZdS)) annotationsN)ABCabstractmethod)errors)StorageRegistry)AnyCallableOptionalPRTypeUnioncast)LazyDependencyzCallable[P, R])fnreturncs$tddddfdd }|S)NzP.argszP.kwargsr )argskwargsrc s^tt|d}z|i|WS|jyX}z |jrBt||WYd}~n d}~00dS)Nr)rStoragebase_exceptionswrap_exceptionsrZ StorageError)rrinstanceexcrO/var/www/html/assistant/venv/lib/python3.9/site-packages/limits/storage/base.pyinners z_wrap_errors..inner) functoolswraps)rrrrr _wrap_errorss rcseZdZUdZded<dddfdd Zd(d d d dfdd ZeeddddZ ed)ddd dddddZ edddddZ eddddd Z ed dd!d"Z ed#dd$d%Zedddd&d'ZZS)*rzC Base class to extend when implementing a storage backend. zOptional[list[str]]ZSTORAGE_SCHEMErNonerrc s6dD]}t||tt||qtjfi|dS)N>clearcheckincrreset get_expirygetsetattrrgetattrsuper__init_subclass__clsrmethod __class__rrr,.szStorage.__init_subclass__NFz Optional[str]boolzUnion[float, str, bool])uriroptionsc st|_t||_dS)z :param wrap_exceptions: Whether to wrap storage exceptions in :exc:`limits.errors.StorageError` before raising it. N) threadingRLocklockr+__init__r)selfr3rr4r0rrr8:s  zStorage.__init__z3Union[Type[Exception], tuple[Type[Exception], ...]])rcCstdS)NNotImplementedErrorr9rrrrIszStorage.base_exceptionsstrint)keyexpiryelastic_expiryamountrcCstdS)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 Nr:)r9r@rArBrCrrrr$Ns z Storage.incr)r@rcCstdS)zB :param key: the key to get the counter value for Nr:r9r@rrrr']sz Storage.getfloatcCstdS)z; :param key: the key to get the expiry for Nr:rDrrrr&dszStorage.get_expirycCstdS)z- check if storage is healthy Nr:r<rrrr#ksz Storage.checkz Optional[int]cCstdS)z/ reset storage to clear limits Nr:r<rrrr%rsz Storage.resetcCstdS)za resets the rate limit key :param key: the key to clear rate limits for Nr:rDrrrr"ysz Storage.clear)NF)Fr=)__name__ __module__ __qualname____doc____annotations__r,r8propertyrrr$r'r&r#r%r" __classcell__rrr0rr&s, r) metaclasscs\eZdZdZdddfdd Zeddd d d d d d d Zedd d ddddZZS)MovingWindowSupportzh Abstract base class for storages that support the :ref:`strategies:moving window` strategy rr r!c s6dD]}t||tt||qtjfi|dS)N>get_moving_window acquire_entryr(r-r0rrr,s z%MovingWindowSupport.__init_subclass__r=r>r?r2r@limitrArCrcCstdS)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 Nr:r9r@rRrArCrrrrPsz!MovingWindowSupport.acquire_entryztuple[float, int])r@rRrArcCstdS)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) Nr:)r9r@rRrArrrrOs z%MovingWindowSupport.get_moving_window)r=) rFrGrHrIr,rrPrOrLrrr0rrNs   rNcsZeZdZdZdddfdd Zeddd d d d d d d Zedd ddddZZS)SlidingWindowCounterSupportzr Abstract base class for storages that support the :ref:`strategies:sliding window counter` strategy. rr r!c s6dD]}t||tt||qtjfi|dS)N>acquire_sliding_window_entryget_sliding_windowr(r-r0rrr,s z-SlidingWindowCounterSupport.__init_subclass__r=r>r?r2rQcCstdS)aU Acquire an entry if the weighted count of the current and previous windows is less than or equal to the limit :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 Nr:rSrrrrUs z8SlidingWindowCounterSupport.acquire_sliding_window_entryztuple[int, float, int, float])r@rArcCstdS)a Return the previous and current window information. :param key: the rate limit key :param expiry: the rate limit expiry, needed to compute the key in some implementations :return: a tuple of (int, float, int, float) with the following information: - previous window counter - previous window TTL - current window counter - current window TTL Nr:)r9r@rArrrrVsz.SlidingWindowCounterSupport.get_sliding_window)r=) rFrGrHrIr,rrUrVrLrrr0rrTs rTc@s(eZdZdZedddddddZd S) TimestampedSlidingWindowz\Helper class for storage that support the sliding window counter, with timestamp based keys.r>r?rEztuple[str, str])r@rAatrcCs0|dt||||dt||fS)a- returns the previous and the current window's keys. :param key: the key to get the window's keys from :param expiry: the expiry of the limit item, in seconds :param at: the timestamp to get the keys from. Default to now, ie ``time.time()`` Returns a tuple with the previous and the current key: (previous, current). Example: - key = "mykey" - expiry = 60 - at = 1738576292.6631825 The return value will be the tuple ``("mykey/28976271", "mykey/28976270")``. /)r?)r.r@rArXrrrsliding_window_keyssz,TimestampedSlidingWindow.sliding_window_keysN)rFrGrHrI classmethodrZrrrrrWsrW) __future__rrr5abcrrZlimitsrZlimits.storage.registryrZ limits.typingrrr r r r r rZ limits.utilrrrrNrTrWrrrrs   ( ])0