Ih DdZddlmZddlmZddlmZdZGddZy) z.Token bucket implementation for rate limiting.) annotations)deque) monotonic) TokenBucketcLeZdZdZdZdZdZd dZdZdZ dZ d dZ d d Z d Z y) raToken Bucket Algorithm. See Also -------- https://en.wikipedia.org/wiki/Token_Bucket Most of this code was stolen from an entry in the ASPN Python Cookbook: https://code.activestate.com/recipes/511490/ Warning: ------- Thread Safety: This implementation is not thread safe. Access to a `TokenBucket` instance should occur within the critical section of any multithreaded code. Nct||_||_t||_t |_t |_yN)floatcapacity_tokens fill_rater timestamprcontents)selfrr s K/var/www/html/planif/env/lib/python3.12/site-packages/kombu/utils/limits.py__init__zTokenBucket.__init__%s2h  y)" c:|jj|yr )rappend)ritems raddzTokenBucket.add,s T"rc6|jjSr )rpopleftrs rpopzTokenBucket.pop/s}}$$&&rc8|jjyr )rclearrs r clear_pendingzTokenBucket.clear_pending2s rcV||jkr|xj|zc_yy)aCheck if one or more tokens can be consumed. Returns ------- bool: true if the number of tokens can be consumed from the bucket. If they can be consumed, a call will also consume the requested number of tokens from the bucket. Calls will only consume `tokens` (the number requested) or zero tokens -- it will never consume a partial number of tokens. TF) _get_tokensr )rtokenss r can_consumezTokenBucket.can_consume5s) T%%' ' LLF "Lrc^|j}t||}||z |jz S)z~Return estimated time of token availability. Returns ------- float: the time in seconds. )r!maxr)rr"r s r expected_timezTokenBucket.expected_timeFs2""$VW% DNN22rc|j|jkrUt}|j||jz z}t |j|j|z|_||_|jSr )r r rrrmin)rnowdeltas rr!zTokenBucket._get_tokensQs\ <<$-- '+CNNcDNN&:;Et}}dllU.BCDL DN||r)r)__name__ __module__ __qualname____doc__rr rrrrrr#r&r!rrrr s?"IHI #'" 3rrN) r. __future__r collectionsrtimer__all__rr/rrr4s"4" LLr