Function-Limiter provides call rate limitation of callable function.
pip install Function-Limiter
Add the rate limiter to your function as decorator. The following example uses the default in memory implementation. Limiter()
create instance of limiter. By using limiter.limit()
call rate of callable function become limited. limiter.limit(limitation, key)
limitation get the limitation can be assigned number per one of these keywords (second, minute, hour, day, month, year). Limitation applied on defined key.
from function_limiter.limiter import Limiter
from function_limiter.limiter import RateLimitExceeded
import time
limiter = Limiter()
@limiter.limit('3/second', 'key')
def function():
print('hello world!')
from function_limiter.limiter import Limiter
from function_limiter.limiter import RateLimitExceeded
import time
storage_uri = 'redis://ip:port/'
limiter = Limiter(
storage_uri=storage_uri
)
There are a few ways of using this decorator depending on your preference and use-case.
The limit string can be a single limit or a delimiter separated string
@limiter.limit('3/second;10 per minute', 'key')
def function():
print('hello world!')
You can implement your own function to retrieve the value of rate limit config.
def limitation():
return '5/second'
def key():
return 'custom key'
@limiter.limit(limitation, key=key)
def function():
print('hello world!')
Redis storage can be involved to lunch multiple instance of application.
limiter = Limiter(
storage_uri='redis://ip:port/'
)
@limiter.limit('3/minute', 'key')
def func():
pass
Exempt key can be used to exempt defined keys. If key and exempt key matched it ignores the limitations
limiter = Limiter()
@limiter.limit('3/minute', 'key', exempt='key')
def func():
pass
You can define rate limit default value when the Limiter instance was initialized. By defining default rate limit values if there isn't any value for the specific key it applies the default value.
limiter = Limiter(
default_limitations='3/minute',
default_key='key',
default_exempt='key'
)
@limiter.limit()
def func():
pass