o wvXjå ã@sDdZddlmZmZddlmZddlmZdZGdd„de ƒZ dS) z.Token bucket implementation for rate limiting.é)Úabsolute_importÚunicode_literals)Ú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úO/var/www/html/myproject/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_pending0szTokenBucket.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_consume3s zTokenBucket.can_consumecCs | ¡}t||ƒ}|||jS)zoReturn estimated time of token availability. Returns: float: the time in seconds. )rÚmaxr )rrr rrrÚ expected_timeCs zTokenBucket.expected_timecCsB|j|jkrtƒ}|j||j}t|j|j|ƒ|_||_|jSr)r r rr r Úmin)rÚnowÚdeltarrrrMs zTokenBucket._get_tokens)r)Ú__name__Ú __module__Ú __qualname__Ú__doc__r r r rrrrrrrrrrrr s    rN) r&Ú __future__rrÚ collectionsrÚ kombu.fiverÚ__all__ÚobjectrrrrrÚs