Skip to main content

choke: simple implementation of throttling mechanism.

Project description

License: MIT Build Status

choke is a package implementing a trivial to use general purpose throttling mechanism. The basic workflow with choke is as follows:

  1. Create manager - an object responsible for keeping track of timestamps when some events (think: calls to your functions) occur.

  2. Instruct manager to “choke” some callables, i.e. define maximum number of calls that can occur per given time window.

  3. Use your callables as usual, keeping in mind that when the above defined limit is exceeded, the choked callable will raise CallLimitExceededError.

Here is an example containing everything you need to use choke:

from time import sleep
from redis import StrictRedis
from choke import RedisChokeManager, CallLimitExceededError

REDIS = StrictRedis() # Tweak this to reflect your setup
CHOKE_MANAGER = RedisChokeManager(redis=REDIS)

# Example configuration: enforce limit of no more than 10 calls in two seconds window
@CHOKE_MANAGER.choke(limit=10, window_length=2)
def foo(x, y):
    """Just print something to show that foo was called."""
    print(f'foo called with ({x}, {y})')

if __name__ == '__main__':
    # We expect pattern of 10 successes followed by 10 failures followed again by 10 successes
    # Some deviations from this pattern may obviously occur as calling foo takes nonzero time
    for i in range(30):
            foo(i, y=i ** 2)
        except CallLimitExceededError:
            print('Foo not called. Limit exceeded!')

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

choke-0.1.2.tar.gz (7.1 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page