Greedy throttling for function calls

throttle_greedy.py

import time
from functools import wraps


def throttle_greedy(wait):
    def decorator(fn):
        state = {'last_call_at': None}

        @wraps(fn)
        def wrapper(*args, **kwargs):
            last_call_at = state.get('last_call_at')
            now = long(time.time())
            if last_call_at:
                diff = now - last_call_at
                if diff < wait:
                    time.sleep(wait - diff)
            result = fn(*args, **kwargs)
            state['last_call_at'] = long(time.time())
            return result
        return wrapper
    return decorator