^f LdZddlmZddlmZddlmZdZGddZdS) z.Token bucket implementation for rate limiting.) annotations)deque) monotonic) TokenBucketcNeZdZdZdZdZdZd dZdZdZ dZ d dZ d d Z d Z dS) 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 |_dSN)floatcapacity_tokens fill_rater timestamprcontents)selfrr s D/var/www/html/env/lib/python3.11/site-packages/kombu/utils/limits.py__init__zTokenBucket.__init__%s?h  y))" c:|j|dSr )rappend)ritems raddzTokenBucket.add,s T"""""rc4|jSr )rpopleftrs rpopzTokenBucket.pop/s}$$&&&rc8|jdSr )rclearrs r clear_pendingzTokenBucket.clear_pending2s rcZ||kr|xj|zc_dSdS)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_consume5s5 T%%'' ' ' LLF "LL4urcd|}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_timeFs6""$$VW%% DN22rc|j|jkrIt}|j||jz z}t |j|j|z|_||_|jSr )r r rrrmin)rnowdeltas rr!zTokenBucket._get_tokensQsV <$- ' '++CNcDN&:;Et}dlU.BCCDL DN|r)r)__name__ __module__ __qualname____doc__rr rrrrrr#r&r!rrrr s"IHI    ###'''" 3 3 3 3rrN) r. __future__r collectionsrtimer__all__rr/rrr4s44"""""" LLLLLLLLLLr