o ;h`@sddlmZddlZddlmZmZddlmZddlm Z ddl m Z m Z m Z mZmZddlmZdd d ZGd ddee dZGdddeZGdddeZGdddZdS)) annotationsN)ABCabstractmethod)errors)StorageRegistry)AnyCallablePRcast)LazyDependencyfnCallable[P, R]returncstd fdd }|S) NargsP.argskwargsP.kwargsrr c sNtt|d}z|i|WS|jy&}z |jr!t||d}~ww)Nr)r Storagebase_exceptionswrap_exceptionsrZ StorageError)rrinstanceexcr F/var/www/html/venv/lib/python3.10/site-packages/limits/storage/base.pyinners z_wrap_errors..inner)rrrrrr ) functoolswraps)r rrrr _wrap_errorss rcseZdZUdZded< d+fdd Z d,d-fdd Zeed.ddZ ed/d0ddZ ed1dd Z ed2d"d#Z ed3d$d%Z ed4d'd(Zed5d)d*ZZS)6rzC Base class to extend when implementing a storage backend. zlist[str] | NoneZSTORAGE_SCHEMErrrNonec 6dD] }t||tt||qtjdi|dS)N> get_expiryclearincrgetcheckresetrsetattrrgetattrsuper__init_subclass__clsrmethod __class__rrr,*szStorage.__init_subclass__NFuri str | Nonerbooloptionsfloat | str | boolc st||_dS)z :param wrap_exceptions: Whether to wrap storage exceptions in :exc:`limits.errors.StorageError` before raising it. N)r+__init__r)selfr2rr5r0rrr76s zStorage.__init__-type[Exception] | tuple[type[Exception], ...]cCt)NNotImplementedErrorr8rrrrDszStorage.base_exceptionskeystrexpiryintamountcCr:)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 r;)r8r?rArCrrrr$Is z Storage.incrcCr:)zB :param key: the key to get the counter value for r;r8r?rrrr%Tz Storage.getfloatcCr:)z; :param key: the key to get the expiry for r;rDrrrr"[rEzStorage.get_expirycCr:)z- check if storage is healthy r;r=rrrr&brEz Storage.check int | NonecCr:)z/ reset storage to clear limits r;r=rrrr'irEz Storage.resetcCr:)za resets the rate limit key :param key: the key to clear rate limits for r;rDrrrr#psz Storage.clearrrrr )NF)r2r3rr4r5r6)rr9r>)r?r@rArBrCrBrrB)r?r@rrB)r?r@rrF)rr4)rrG)r?r@rr )__name__ __module__ __qualname____doc____annotations__r,r7propertyrrr$r%r"r&r'r# __classcell__rrr0rr"s.     r) metaclasscs@eZdZdZdfdd ZeddddZedddZZS)MovingWindowSupportzh Abstract base class for storages that support the :ref:`strategies:moving window` strategy rrrr c r!)N> acquire_entryget_moving_windowrr(r-r0rrr,s z%MovingWindowSupport.__init_subclass__r>r?r@limitrBrArCr4cCr:)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 r;r8r?rUrArCrrrrSz!MovingWindowSupport.acquire_entrytuple[float, int]cCr:)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) r;)r8r?rUrArrrrTs z%MovingWindowSupport.get_moving_windowrHrI r?r@rUrBrArBrCrBrr4)r?r@rUrBrArBrrX) rJrKrLrMr,rrSrTrPrrr0rrRzs  rRcsPeZdZdZdfdd Ze ddddZedddZedddZZ S)SlidingWindowCounterSupportzr Abstract base class for storages that support the :ref:`strategies:sliding window counter` strategy. rrrr c r!)N>get_sliding_windowacquire_sliding_window_entryclear_sliding_windowrr(r-r0rrr,s z-SlidingWindowCounterSupport.__init_subclass__r>r?r@rUrBrArCr4cCr:)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 r;rVrrrr\s z8SlidingWindowCounterSupport.acquire_sliding_window_entrytuple[int, float, int, float]cCr:)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 r;r8r?rArrrr[sz.SlidingWindowCounterSupport.get_sliding_windowcCsdS)z Resets the rate limit key(s) for the sliding window :param key: the key to clear rate limits for :param expiry: the rate limit expiry, needed to compute the key in some implemenations Nrr_rrrr]rWz0SlidingWindowCounterSupport.clear_sliding_windowrHrIrY)r?r@rArBrr^)r?r@rArBrr ) rJrKrLrMr,rr\r[r]rPrrr0rrZs  rZc@seZdZdZed d d Zd S)TimestampedSlidingWindowz\Helper class for storage that support the sliding window counter, with timestamp based keys.r?r@rArBatrFrtuple[str, str]cCs0|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")``. /)rB)r.r?rArarrrsliding_window_keyss0z,TimestampedSlidingWindow.sliding_window_keysN)r?r@rArBrarFrrb)rJrKrLrM classmethodrdrrrrr`sr`)r rrr) __future__rrabcrrZlimitsrZlimits.storage.registryrZ limits.typingrrr r r Z limits.utilr rrrRrZr`rrrrs     X)>