9i`ddlmZddlZddlmZmZddlmZddlm Z ddl m Z m Z m Z mZmZddlmZ ddZGd d ee ZGd d eZGddeZGddZy)) annotationsN)ABCabstractmethod)errors)StorageRegistry)AnyCallablePRcast)LazyDependencycDtjdfd }|S)Nctt|d} |i|S#|j$r(}|jrt j ||d}~wwxYw)Nr)r Storagebase_exceptionswrap_exceptionsr StorageError)argskwargsinstanceexcfns F/var/www/html/venv/lib/python3.12/site-packages/limits/storage/base.pyinnerz_wrap_errors..inners]a) t&v& &'' ''))#.C7  sA#AA)rzP.argsrzP.kwargsreturnr ) functoolswraps)rrs` r _wrap_errorsrs'__R LceZdZUdZded< d fd Z d dfd ZeeddZ edddZ eddZ edd Z edd Z edd Zedd ZxZS)rzC Base class to extend when implementing a storage backend. zlist[str] | NoneSTORAGE_SCHEMEc rdD]"}t||tt||$t|di|y)N>getincrcheckclearreset get_expirysetattrrgetattrsuper__init_subclass__clsrmethod __class__s rr.zStorage.__init_subclass__*sB  EF Cgc6.B!C D E !+F+rc 0t|||_y)z :param wrap_exceptions: Whether to wrap storage exceptions in :exc:`limits.errors.StorageError` before raising it. N)r-__init__r)selfuriroptionsr2s rr4zStorage.__init__6s .rct)NNotImplementedErrorr5s rrzStorage.base_exceptionsDs "!rct)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 r9)r5keyexpiryamounts rr$z Storage.incrIs "!rct)zB :param key: the key to get the counter value for r9r5r=s rr#z Storage.getT "!rct)z; :param key: the key to get the expiry for r9rAs rr(zStorage.get_expiry[rBrct)z- check if storage is healthy r9r;s rr%z Storage.checkbrBrct)z/ reset storage to clear limits r9r;s rr'z Storage.resetirBrct)za resets the rate limit key :param key: the key to clear rate limits for r9rAs rr&z Storage.clearps "!rrrrNone)NF)r6z str | Nonerboolr7zfloat | str | bool)rz-type[Exception] | tuple[type[Exception], ...])r=strr>intr?rMrrM)r=rLrrM)r=rLrfloat)rrI)rz int | None)r=rLrrH)__name__ __module__ __qualname____doc____annotations__r.r4propertyrrr$r#r(r%r'r& __classcell__r2s@rrr"s%$E , % /  / /& /"""""" "" "" "" ""rr) metaclasscJeZdZdZdfd ZedddZeddZxZS) MovingWindowSupportzh Abstract base class for storages that support the :ref:`strategies:moving window` strategy c rdD]"}t||tt||$t|di|y)N> acquire_entryget_moving_windowr)r*r/s rr.z%MovingWindowSupport.__init_subclass__sE  F WS&12    !+F+rct)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 r9r5r=limitr>r?s rr[z!MovingWindowSupport.acquire_entrys "!rct)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) r9)r5r=r_r>s rr\z%MovingWindowSupport.get_moving_windows "!rrGrJ r=rLr_rMr>rMr?rMrrI)r=rLr_rMr>rMrztuple[float, int]) rOrPrQrRr.rr[r\rUrVs@rrYrYzs5 ,"" " "rrYc|eZdZdZdfd Ze d ddZe d dZed dZxZ S) SlidingWindowCounterSupportzr Abstract base class for 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*r/s rr.z-SlidingWindowCounterSupport.__init_subclass__sE F WS&12   !+F+rct)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 r9r^s rrgz8SlidingWindowCounterSupport.acquire_sliding_window_entrys "!rct)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 r9r5r=r>s rrez.SlidingWindowCounterSupport.get_sliding_windows "!rcy)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)rjs rrfz0SlidingWindowCounterSupport.clear_sliding_windows rrGrJra)r=rLr>rMrztuple[int, float, int, float])r=rLr>rMrrH) rOrPrQrRr.rrgrerfrUrVs@rrcrcs ,?@ " "" ",/ "9< "  " """ #" &""   rrcc"eZdZdZeddZy)TimestampedSlidingWindowz\Helper class for storage that support the sliding window counter, with timestamp based keys.cT|dt||z |z |dt||z fS)a- returns the previous and the current window's keys. :param key: the key to get the window's keys from :param expiry: the expiry of the limit item, in seconds :param at: the timestamp to get the keys from. Default to now, ie ``time.time()`` Returns a tuple with the previous and the current key: (previous, current). Example: - key = "mykey" - expiry = 60 - at = 1738576292.6631825 The return value will be the tuple ``("mykey/28976271", "mykey/28976270")``. /)rM)r0r=r>ats rsliding_window_keysz,TimestampedSlidingWindow.sliding_window_keyss>$aR&[F2345#aBK@P?Q7RRRrN)r=rLr>rMrprNrztuple[str, str])rOrPrQrR classmethodrqr)rrrmrmsfSSrrm)rCallable[P, R]rrs) __future__rrabcrrlimitsrlimits.storage.registryr limits.typingrr r r r limits.utilr rrrYrcrmr)rrrzso"#3'    U"nU"p&"#&"R; #; |SSr