o ¨D®f÷ ã@s>dZddlmZddlmZddlmZdZGdd„dƒZdS) z.Token bucket implementation for rate limiting.é)Ú annotations)Údeque)Ú monotonic)Ú TokenBucketc@sZeZdZdZdZdZdZddd„Zdd„Zdd „Z d d „Z dd d „Z ddd„Z dd„Z dS)raÀToken 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. NécCs.t|ƒ|_||_t|ƒ|_tƒ|_tƒ|_dS©N)ÚfloatÚcapacityÚ_tokensÚ fill_raterÚ timestamprÚcontents)Úselfr r ©rúK/home/ubuntu/webapp/venv/lib/python3.10/site-packages/kombu/utils/limits.pyÚ__init__%s   zTokenBucket.__init__cCs|j |¡dSr)r Úappend)rÚitemrrrÚadd,szTokenBucket.addcCs |j ¡Sr)r Úpopleft©rrrrÚpop/s zTokenBucket.popcCs|j ¡dSr)r ÚclearrrrrÚ clear_pending2szTokenBucket.clear_pendingcCs"|| ¡kr|j|8_dSdS)aÕCheck 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 )rÚtokensrrrÚ can_consume5s zTokenBucket.can_consumecCs | ¡}t||ƒ}|||jS)z~Return estimated time of token availability. Returns ------- float: the time in seconds. )rÚmaxr )rrr rrrÚ expected_timeFs zTokenBucket.expected_timecCsB|j|jkrtƒ}|j||j}t|j|j|ƒ|_||_|jSr)r r rr r Úmin)rÚnowÚdeltarrrrQs zTokenBucket._get_tokens)r)Ú__name__Ú __module__Ú __qualname__Ú__doc__r r r rrrrrrrrrrrr s    rN) r%Ú __future__rÚ collectionsrÚtimerÚ__all__rrrrrÚs