9i2ddlmZddlZddlmZmZddlmZddlm Z ddl m Z ddl m Z mZmZmZmZmZddlmZ dd Zed Gd d ee ZGddeZGddeZy)) annotationsN)ABCabstractmethod) versionadded)errors)StorageRegistry)Any AwaitableCallablePRcast)LazyDependencycDtjdfd }|S)NcKtt|d} |i|d{S7#|j$r(}|jrt j ||d}~wwxYww)Nr)rStoragebase_exceptionswrap_exceptionsr StorageError)argskwargsinstanceexcfns J/var/www/html/venv/lib/python3.12/site-packages/limits/aio/storage/base.pyinnerz_wrap_errors..innersfa) T,V,, ,,'' ''))#.C7  s0A$ *(*A$*A!#AA!!A$)rzP.argsrzP.kwargsreturnr ) functoolswraps)rrs` r _wrap_errorsr s'__R Lz2.1)versionceZdZUdZded< d fd Z d dfd ZeeddZ edddZ eddZ edd Z edd Z edd Zedd ZxZS)rzJ Base class to extend when implementing an async storage backend. zlist[str] | NoneSTORAGE_SCHEMEc t|di|dD]"}t||tt ||$t|di|y)N>getincrcheckclearreset get_expiry)super__init_subclass__setattrr getattrclsrmethod __class__s rr.zStorage.__init_subclass__.sR !+F+  EF Cgc6.B!C D E !+F+r!c 0t|||_y)z :param wrap_exceptions: Whether to wrap storage exceptions in :exc:`limits.errors.StorageError` before raising it. N)r-__init__r)selfuriroptionsr4s rr6zStorage.__init__;s .r!ct)NNotImplementedErrorr7s rrzStorage.base_exceptionsHs "!r!cKtw)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;)r7keyexpiryamounts rr'z Storage.incrMs"! cKtw)zB :param key: the key to get the counter value for r;r7r?s rr&z Storage.getX "!rBcKtw)z; :param key: the key to get the expiry for r;rDs rr+zStorage.get_expiry_rErBcKtw)z- check if storage is healthy r;r=s rr(z Storage.checkfrErBcKtw)z/ reset storage to clear limits r;r=s rr*z Storage.resetmrErBcKtw)za resets the rate limit key :param key: the key to clear rate limits for r;rDs rr)z Storage.clearts"!rBrr rNone)NF)r8z str | Nonerboolr9zfloat | str | boolrrK)rz-type[Exception] | tuple[type[Exception], ...])r?strr@intrArPrrP)r?rOrrP)r?rOrfloat)rrL)rz int | None)r?rOrrK)__name__ __module__ __qualname____doc____annotations__r.r6propertyrrr'r&r+r(r*r) __classcell__r4s@rrr%s%$E , % /  / /& /  /"""""" "" "" "" ""r!r) metaclasscneZdZdZdfd Ze d ddZe ddZxZS) MovingWindowSupportzn Abstract base class for async storages that support the :ref:`strategies:moving window` strategy c rdD]"}t||tt||$t|di|y)N> acquire_entryget_moving_windowr,r/r r0r-r.r1s rr.z%MovingWindowSupport.__init_subclass__sE  F WS&12    !+F+r!cKtw)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;r7r?limitr@rAs rr^z!MovingWindowSupport.acquire_entrys"!rBcKtw)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;)r7r?rcr@s rr_z%MovingWindowSupport.get_moving_windows"!rBrJrM r?rOrcrPr@rPrArPrrL)r?rOrcrPr@rPrztuple[float, int]) rRrSrTrUr.rr^r_rXrYs@rr\r\~s ,?@ " "" ",/ "9< "  " " " "" ",/ "  " "r!r\c|eZdZdZdfd Ze d ddZe d dZed dZxZ S) SlidingWindowCounterSupportzw Abstract base class for async storages that support the :ref:`strategies:sliding window counter` strategy c rdD]"}t||tt||$t|di|y)N>get_sliding_windowclear_sliding_windowacquire_sliding_window_entryr,r`r1s rr.z-SlidingWindowCounterSupport.__init_subclass__sE F WS&12   !+F+r!cKtw)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;rbs rrkz8SlidingWindowCounterSupport.acquire_sliding_window_entrys""!rBcKtw)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;r7r?r@s rriz.SlidingWindowCounterSupport.get_sliding_windows"!rBc Kyw)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 Nr,rns rrjz0SlidingWindowCounterSupport.clear_sliding_windows  srJrMre)r?rOr@rPrztuple[int, float, int, float])r?rOr@rPrrK) rRrSrTrUr.rrkrirjrXrYs@rrgrgs ,  " "" "  "  ""$"" #" &""   r!rg)rCallable[P, Awaitable[R]]rrp) __future__rrabcrrdeprecated.sphinxrlimitsrlimits.storage.registryr limits.typingr r r r r r limits.utilrr rr\rgr,r!rrxsz"#*3' !   eU"nU"U"p*"#*"Z? #? r!