h @dZddlmZddlmZdZGddZdS)z.Token bucket implementation for rate limiting.)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)selfr r s N/var/www/html/Sam_Eipo/venv/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_pending.s 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_consume1s5 T%%'' ' ' LLF "LL4urcd|}t||}||z |jz S)zoReturn estimated time of token availability. Returns: float: the time in seconds. )r maxr )rr!r s r expected_timezTokenBucket.expected_timeAs6 ""$$VW%% DN22rc|j|jkrIt}|j||jz z}t |j|j|z|_||_|jSr )r r rr rmin)rnowdeltas rr zTokenBucket._get_tokensKsV <$- ' '++CNcDN&:;Et}dlU.BCCDL DN|r)r)__name__ __module__ __qualname____doc__r r rrrrrr"r%r rrrr s  IHI    ###''' 3333rrN)r- collectionsrtimer__all__rr.rrr2sp44 HHHHHHHHHHr