a a*j^@sddlmZddlZddlmZmZddlmZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZmZmZddlmZd d d d d Zed dGdddee dZGdddeZGdddeZdS)) annotationsN)ABCabstractmethod) versionadded)errors)StorageRegistry) Any AwaitableCallableOptionalPRTypeUnioncast)LazyDependencyzCallable[P, Awaitable[R]])fnreturncs$tddddfdd }|S)NzP.argszP.kwargsr )argskwargsrc sdtt|d}z|i|IdHWS|jy^}z |jrHt||WYd}~n d}~00dS)Nr)rStoragebase_exceptionswrap_exceptionsrZ StorageError)rrinstanceexcrS/var/www/html/assistant/venv/lib/python3.9/site-packages/limits/aio/storage/base.pyinners z_wrap_errors..inner) functoolswraps)rrrrr _wrap_errorss r!z2.1)versioncseZdZUdZded<dddfdd Zd(d d d ddfdd ZeeddddZ ed)ddd dddddZ edddddZ eddddd Z ed dd!d"Z ed#dd$d%Zedddd&d'ZZS)*rzJ Base class to extend when implementing an async storage backend. zOptional[list[str]]ZSTORAGE_SCHEMErNonerrc sHtjfi|dD]}t||tt||qtjfi|dS)N>clearcheckincrreset get_expiryget)super__init_subclass__setattrr!getattrclsrmethod __class__rrr,1szStorage.__init_subclass__NFz Optional[str]boolzUnion[float, str, bool])uriroptionsrc st||_dS)z :param wrap_exceptions: Whether to wrap storage exceptions in :exc:`limits.errors.StorageError` before raising it. N)r+__init__r)selfr5rr6r2rrr7>s zStorage.__init__z3Union[Type[Exception], tuple[Type[Exception], ...]])rcCstdS)NNotImplementedErrorr8rrrrKszStorage.base_exceptionsstrint)keyexpiryelastic_expiryamountrcstdS)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 Nr9)r8r?r@rArBrrrr'Ps z Storage.incr)r?rcstdS)zB :param key: the key to get the counter value for Nr9r8r?rrrr*_sz Storage.getfloatcstdS)z; :param key: the key to get the expiry for Nr9rCrrrr)fszStorage.get_expirycstdS)z- check if storage is healthy Nr9r;rrrr&msz Storage.checkz Optional[int]cstdS)z/ reset storage to clear limits Nr9r;rrrr(tsz Storage.resetcstdS)za resets the rate limit key :param key: the key to clear rate limits for Nr9rCrrrr%{sz Storage.clear)NF)Fr<)__name__ __module__ __qualname____doc____annotations__r,r7propertyrrr'r*r)r&r(r% __classcell__rrr2rr(s,  r) metaclasscs\eZdZdZdddfdd Zeddd d d d d d d Zedd d ddddZZS)MovingWindowSupportzn Abstract base class for async storages that support the :ref:`strategies:moving window` strategy rr#r$c s6dD]}t||tt||qtjfi|dS)N>get_moving_window acquire_entryr-r!r.r+r,r/r2rrr,s z%MovingWindowSupport.__init_subclass__r<r=r>r4r?limitr@rBrcstdS)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 Nr9r8r?rRr@rBrrrrOs z!MovingWindowSupport.acquire_entryztuple[float, int])r?rRr@rcstdS)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) Nr9)r8r?rRr@rrrrNs z%MovingWindowSupport.get_moving_window)r<) rErFrGrHr,rrOrNrKrrr2rrMs  rMcsZeZdZdZdddfdd Zeddd d d d d d d Zedd ddddZZS)SlidingWindowCounterSupportzw Abstract base class for async storages that support the :ref:`strategies:sliding window counter` strategy rr#r$c s6dD]}t||tt||qtjfi|dS)N>acquire_sliding_window_entryget_sliding_windowrPr/r2rrr,s z-SlidingWindowCounterSupport.__init_subclass__r<r=r>r4rQcstdS)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 Nr9rSrrrrUsz8SlidingWindowCounterSupport.acquire_sliding_window_entryztuple[int, float, int, float])r?r@rcstdS)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 Nr9)r8r?r@rrrrVsz.SlidingWindowCounterSupport.get_sliding_window)r<) rErFrGrHr,rrUrVrKrrr2rrTs rT) __future__rrabcrrZdeprecated.sphinxrZlimitsrZlimits.storage.registryrZ limits.typingrr r r r r rrrZ limits.utilrr!rrMrTrrrrs    , \-