gitlab_request_rate_limit.py

from functools import wraps
from threading import Timer, Semaphore

import requests

def service_request_rate_limit(per_minute: int):
    assert isinstance(per_minute, int), per_minute
    assert per_minute > 0, per_minute
    lock = Semaphore(per_minute)
 
    def decorator(fn):
        nonlocal lock
        @wraps(fn)
        def wrapper(*a, **k):
            nonlocal lock
            lock.acquire()
            output = fn(*a, **k)
            Timer(60.0, lock.release).start()
            return output
        return wrapper
    return decorator

gitlab_ratelimit_requests_per_minute = 1000

gitlab_request_rate_limit = service_request_rate_limit(per_minute=gitlab_ratelimit_requests_per_minute)

gitlab_get = gitlab_request_rate_limit(requests.get)
gitlab_put = gitlab_request_rate_limit(requests.put)
gitlab_post = gitlab_request_rate_limit(requests.post)
gitlab_delete = gitlab_request_rate_limit(requests.delete)