o ;h2@sddlmZddlZddlmZmZddlmZddlm Z ddl m Z ddl m Z mZmZmZmZmZddlmZdd d ZeddGdddee dZGdddeZGdddeZdS)) annotationsN)ABCabstractmethod) versionadded)errors)StorageRegistry)Any AwaitableCallablePRcast)LazyDependencyfnCallable[P, Awaitable[R]]returncstd fdd }|S) NargsP.argskwargsP.kwargsrr c sVtt|d}z |i|IdHWS|jy*}z |jr%t||d}~ww)Nr)r Storagebase_exceptionswrap_exceptionsrZ StorageError)rrinstanceexcrJ/var/www/html/venv/lib/python3.10/site-packages/limits/aio/storage/base.pyinners z_wrap_errors..inner)rrrrrr ) functoolswraps)rrrrr _wrap_errorss r!z2.1)versioncseZdZUdZded< d+fdd Z d,d-fdd Zeed.ddZ ed/d0ddZ ed1dd Z ed2d"d#Z ed3d$d%Z ed4d'd(Zed5d)d*ZZS)6rzJ Base class to extend when implementing an async storage backend. zlist[str] | NoneZSTORAGE_SCHEMErrrNonec sHtjdi|dD] }t||tt||q tjdi|dS)N> get_expiryclearincrgetcheckresetr)super__init_subclass__setattrr!getattrclsrmethod __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)selfr3rr6r1rrr8;s zStorage.__init__-type[Exception] | tuple[type[Exception], ...]cCst)NNotImplementedErrorr9rrrrHszStorage.base_exceptionskeystrexpiryintamountct)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;)r9r?rArCrrrr&Ms z Storage.incrcrD)zB :param key: the key to get the counter value for r;r9r?rrrr'Xz Storage.getfloatcrD)z; :param key: the key to get the expiry for r;rErrrr$_rFzStorage.get_expirycrD)z- check if storage is healthy r;r=rrrr(frFz Storage.check int | NonecrD)z/ reset storage to clear limits r;r=rrrr)mrFz Storage.resetcrD)za resets the rate limit key :param key: the key to clear rate limits for r;rErrrr%tsz Storage.clearrrrr#)NF)r3r4rr5r6r7rr#)rr:r>)r?r@rArBrCrBrrB)r?r@rrB)r?r@rrG)rr5)rrH)r?r@rr#)__name__ __module__ __qualname____doc____annotations__r+r8propertyrrr&r'r$r(r)r% __classcell__rrr1rr%s.      r) metaclasscsBeZdZdZdfdd Ze ddddZedddZZS)MovingWindowSupportzn Abstract base class for async storages that support the :ref:`strategies:moving window` strategy rrrr#c 6dD] }t||tt||qtjdi|dS)N> acquire_entryget_moving_windowrr,r!r-r*r+r.r1rrr+s z%MovingWindowSupport.__init_subclass__r>r?r@limitrBrArCr5crD)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;r9r?rXrArCrrrrUs z!MovingWindowSupport.acquire_entrytuple[float, int]crD)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;)r9r?rXrArrrrVs z%MovingWindowSupport.get_moving_windowrIrJ r?r@rXrBrArBrCrBrr5)r?r@rXrBrArBrrZ) rKrLrMrNr+rrUrVrQrrr1rrS~s  rScsPeZdZdZdfdd Ze ddddZedddZedddZZ S)SlidingWindowCounterSupportzw Abstract base class for async storages that support the :ref:`strategies:sliding window counter` strategy rrrr#c rT)N>get_sliding_windowacquire_sliding_window_entryclear_sliding_windowrrWr.r1rrr+s z-SlidingWindowCounterSupport.__init_subclass__r>r?r@rXrBrArCr5crD)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;rYrrrr^sz8SlidingWindowCounterSupport.acquire_sliding_window_entrytuple[int, float, int, float]crD)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;r9r?rArrrr]sz.SlidingWindowCounterSupport.get_sliding_windowcsdS)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 Nrrarrrr_sz0SlidingWindowCounterSupport.clear_sliding_windowrIrJr[)r?r@rArBrr`)r?r@rArBrr#) rKrLrMrNr+rr^r]r_rQrrr1rr\s  r\)rrrr) __future__rrabcrrZdeprecated.sphinxrZlimitsrZlimits.storage.registryrZ limits.typingrr r r r r Z limits.utilrr!rrSr\rrrrs       X-