Skip to content

Instantly share code, notes, and snippets.

@h2rd
Forked from ChrisTM/throttle.py
Created March 3, 2017 19:39
Show Gist options
  • Save h2rd/3eb87b144a212497eab49079ea9d797a to your computer and use it in GitHub Desktop.
Save h2rd/3eb87b144a212497eab49079ea9d797a to your computer and use it in GitHub Desktop.
Python decorator for throttling function calls.
class throttle(object):
"""
Decorator that prevents a function from being called more than once every
time period.
To create a function that cannot be called more than once a minute:
@throttle(minutes=1)
def my_fun():
pass
"""
def __init__(self, seconds=0, minutes=0, hours=0):
self.throttle_period = timedelta(
seconds=seconds, minutes=minutes, hours=hours
)
self.time_of_last_call = datetime.min
def __call__(self, fn):
@wraps(fn)
def wrapper(*args, **kwargs):
now = datetime.now()
time_since_last_call = now - self.time_of_last_call
if time_since_last_call > self.throttle_period:
self.time_of_last_call = now
return fn(*args, **kwargs)
return wrapper
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment