^g'ddlmZddlZddlmZmZddlmZddlm 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 ^/var/www/html/suriana-translation/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 e d ddZ 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)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 r;)r7keyexpiryelastic_expiryamounts rr'z Storage.incrMs"! cKtw)zB :param key: the key to get the counter value for r;r7r?s rr&z Storage.get\ "!rCcKtw)z; :param key: the key to get the expiry for r;rEs rr+zStorage.get_expirycrFrCcKtw)z- check if storage is healthy r;r=s rr(z Storage.checkjrFrCcKtw)z/ reset storage to clear limits r;r=s rr*z Storage.resetqrFrCcKtw)za resets the rate limit key :param key: the key to clear rate limits for r;rEs rr)z Storage.clearxs"!rCrr rNone)NF)r8z str | Nonerboolr9zfloat | str | boolrrL)rz-type[Exception] | tuple[type[Exception], ...])F) r?strr@intrArMrBrPrrP)r?rOrrP)r?rOrfloat)rrM)rz int | None)r?rOrrL)__name__ __module__ __qualname____doc____annotations__r.r6propertyrrr'r&r+r(r*r) __classcell__r4s@rrr%s %$E , % /  / /& /  /""QR " " # "59 "KN "  " """ "" "" "" ""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@rBs rr^z!MovingWindowSupport.acquire_entrys"!rCcKtw)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"!rCrKrN r?rOrcrPr@rPrBrPrrM)r?rOrcrPr@rPrztuple[float, int]) rRrSrTrUr.rr^r_rXrYs@rr\r\s ,?@ " "" ",/ "9< "  " " " "" ",/ "  " "r!r\cjeZdZdZdfd Ze d ddZe ddZxZS) 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_windowacquire_sliding_window_entryr,r`r1s rr.z-SlidingWindowCounterSupport.__init_subclass__sAL 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""!rCcKtw)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 rrjz.SlidingWindowCounterSupport.get_sliding_windows"!rCrKrerf)r?rOr@rPrztuple[int, float, int, float]) rRrSrTrUr.rrkrjrXrYs@rrhrhs ,  " "" "  "  ""$"" #" &""r!rh)rCallable[P, Awaitable[R]]rrn) __future__rrabcrrdeprecated.sphinxrlimitsrlimits.storage.registryr limits.typingr r r r r r limits.utilrr rr\rhr,r!rrvsz"#*3' !   eY"nY"Y"x*"#*"Z1"#1"r!