ZeIi2ddlmZddlZddlmZmZddlmZddlm Z ddl m Z ddl m Z mZmZmZmZmZddlmZdd Zed Gddee ZGddeZGddeZdS)) annotationsN)ABCabstractmethod) versionadded)errors)StorageRegistry)Any AwaitableCallablePRcast)LazyDependencyfnCallable[P, Awaitable[R]]returncHtjdfd }|S) NargsP.argskwargsP.kwargsrr cKtt|d} |i|d{VS#|j$r"}|jrt j||d}~wwxYw)Nr)rStoragebase_exceptionswrap_exceptionsr StorageError)rrinstanceexcrs \/var/www/html/MCyber-Diagnostic/venv/lib/python3.11/site-packages/limits/aio/storage/base.pyinnerz_wrap_errors..innersa)) T,V,,,,,,,, ,'   ' 8)#..C7  s . AAA)rrrrrr ) functoolswraps)rr s` r _wrap_errorsr#s?_R Lz2.1)versionceZdZUdZded< d"fd Z d#d$fd Zeed%dZ ed&d'dZ ed(dZ ed)dZ ed*dZ ed+d Zed,d!ZxZS)-rzJ Base class to extend when implementing an async storage backend. zlist[str] | NoneSTORAGE_SCHEMErr rNonec tjdi|dD].}t||tt ||/tjdi|dS)N>getincrcheckclearreset get_expiry)super__init_subclass__setattrr#getattrclsrmethod __class__s rr2zStorage.__init_subclass__.s!!++F+++  E EF Cgc6.B.B!C!C D D D D!!++F+++++r$NFuri str | Nonerbooloptionsfloat | str | boolc Vt||_dS)z :param wrap_exceptions: Whether to wrap storage exceptions in :exc:`limits.errors.StorageError` before raising it. N)r1__init__r)selfr9rr<r8s rr?zStorage.__init__;s) .r$-type[Exception] | tuple[type[Exception], ...]ct)NNotImplementedErrorr@s rrzStorage.base_exceptionsHs "!r$keystrexpiryintamountcKt)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 rC)r@rGrIrKs rr+z Storage.incrMs"!r$cKt)zB :param key: the key to get the counter value for rCr@rGs rr*z Storage.getX "!r$floatcKt)z; :param key: the key to get the expiry for rCrNs rr/zStorage.get_expiry_rOr$cKt)z- check if storage is healthy rCrEs rr,z Storage.checkfrOr$ int | NonecKt)z/ reset storage to clear limits rCrEs rr.z Storage.resetmrOr$cKt)za resets the rate limit key :param key: the key to clear rate limits for rCrNs rr-z Storage.clearts"!r$rr rr()NF)r9r:rr;r<r=rr()rrArF)rGrHrIrJrKrJrrJ)rGrHrrJ)rGrHrrP)rr;)rrS)rGrHrr()__name__ __module__ __qualname____doc____annotations__r2r?propertyrrr+r*r/r,r.r- __classcell__r8s@rrr%st%$$$E , , , , , , % / / / / / / /"""^X"""""^""""^" """^" """^" """^" """^"""""r$r) metaclasscXeZdZdZdfd Ze dddZeddZxZS)MovingWindowSupportzn Abstract base class for async storages that support the :ref:`strategies:moving window` strategy rr rr(c dD].}t||tt||/tjdi|dS)N> acquire_entryget_moving_windowr0r3r#r4r1r2r5s rr2z%MovingWindowSupport.__init_subclass__sm   F WS&1122     "!++F+++++r$rFrGrHlimitrJrIrKr;cKt)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 rCr@rGrgrIrKs rrdz!MovingWindowSupport.acquire_entrys"!r$tuple[float, int]cKt)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) rC)r@rGrgrIs rrez%MovingWindowSupport.get_moving_windows"!r$rVrW rGrHrgrJrIrJrKrJrr;)rGrHrgrJrIrJrrj) rXrYrZr[r2rrdrer^r_s@rrbrb~s , , , , , ,?@ " " " "^ " " " "^ " " " " "r$rbcpeZdZdZdfd Ze dddZeddZeddZxZ S)SlidingWindowCounterSupportzw Abstract base class for async storages that support the :ref:`strategies:sliding window counter` strategy rr rr(c dD].}t||tt||/tjdi|dS)N>get_sliding_windowclear_sliding_windowacquire_sliding_window_entryr0rfr5s rr2z-SlidingWindowCounterSupport.__init_subclass__sm  F WS&1122     "!++F+++++r$rFrGrHrgrJrIrKr;cKt)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 rCris rrrz8SlidingWindowCounterSupport.acquire_sliding_window_entrys""!r$tuple[int, float, int, float]cKt)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 rCr@rGrIs rrpz.SlidingWindowCounterSupport.get_sliding_windows"!r$c KdS)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 Nr0rvs rrqz0SlidingWindowCounterSupport.clear_sliding_windows  r$rVrWrl)rGrHrIrJrrt)rGrHrIrJrr() rXrYrZr[r2rrrrprqr^r_s@rrnrns , , , , , ,  """"^"$"""^"    ^     r$rn)rrrr) __future__rr!abcrrdeprecated.sphinxrlimitsrlimits.storage.registryr limits.typingr r r r r r limits.utilrr#rrbrnr0r$rrs""""""########******333333'&&&&&     eU"U"U"U"U"nU"U"U"U"p*"*"*"*"*"#*"*"*"Z? ? ? ? ? #? ? ? ? ? r$