jKjddlZddlZddlZddlmZmZmZmZmZdZ ejdZ ee e fZdedededefd ZGd d ZGd d Zy)N)CallableDictHashableOptionalUniong@posthog min_value max_valuelabelreturnc t|tst|ttfstj |d|d|S||kDr tj |d|d|d|S||kr tj |d|d|d|S|S)Nz# must be a number. Using max value .z cannot be greater than z. Using z cannot be less than ) isinstanceboolintfloatlogwarning)valuer r r s n/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/posthog/bucketed_rate_limiter.py_clamp_to_rangers%je &E ug@ 1MN y ug5i[ STUV y ug29+Xi[PQRS Lc eZdZdZdedefdZy)_Buckettokens last_accessrrc ||_||_yNr)selfrrs r__init__z_Bucket.__init__!s &rN)__name__ __module__ __qualname__ __slots__Numberrr!rrrrs)I'v'E'rrc eZdZdZdej fdedededeee gdfdege ff dZ d e d e d dfd Z d e d efdZddZy)BucketedRateLimiteruToken bucket rate limiter that tracks a separate bucket per key. Each key starts with a full bucket of ``bucket_size`` tokens and every call to :meth:`consume_rate_limit` consumes one token. ``refill_rate`` tokens are restored per elapsed ``refill_interval_seconds`` (whole intervals only, fractional elapsed time is carried over), capped at ``bucket_size``. The call that empties a bucket is itself reported as rate limited — a burst over a fresh bucket lets ``bucket_size - 1`` events through before limiting kicks in — and ``on_bucket_rate_limited`` fires once each time a bucket is drained. Thread-safe. ``clock`` must return seconds and is injectable for tests. N bucket_size refill_raterefill_interval_secondson_bucket_rate_limitedclockct|ddd|_t|d|jd|_t|dtd|_||_||_i|_tj|_ y)Nrdr*r+r,) r _bucket_size _refill_rateONE_DAY_IN_SECONDS_refill_interval_on_bucket_rate_limited_clock_buckets threadingLock_lock)r r*r+r,r-r.s rr!zBucketedRateLimiter.__init__7ss,KCO+ D--} !0 #Q(:$ 13 ^^% rbucketnowr c`|jdkr|j|_||_y||jz }t ||jz}|dkDrZ||j z}t |j|z|j|_|xj||jzz c_yy)Nr)r4r1rrrr2min)r r;r<elapsedrefill_intervals tokens_to_adds r _apply_refillz!BucketedRateLimiter._apply_refillKs  A % --FM!$F  ***w$*?*??@ a ,t/@/@@M =t?P?PQFM   "2T5J5J"J J  rkeycd}|j5|j}|jj|}|'t |j |}||j|<n|j |||jdkr dddy|xjdzc_|jdk}|r |j}ddd|||S#1swYxYw)zrTst  <<g " sEz  f   PV ''Q"Q"r