this package help you
to limit the call's [frequent] rates of any callable in python.
- default built-in memory is :
expiring dict memorythat not recommended for production usage
pip install rlimit
from rlimit import Limiter, Per
from time import sleep
@Limiter(times=2, per=2 * Per.SECOND)
def func_a():
return True
func_a() # True
func_a() # True
func_a() # OverflowError
sleep(2)
func_a() # True
func_a() # True
func_a() # OverflowErroruseful when uses in class or creating embed key
from rlimit import Limiter, Per
@Limiter(
times=2, per=2 * Per.SECOND,
key=lambda name, age: str(hash((name, age)))
)
def func_a(name: str, age: int):
return f'hi2{name}-{age}'
func_a('gzuz', 13)
func_a('gzuz', 14)
func_a('gzuz', 13)
func_a('gzuz', 15)
func_a('gzuz', 14)
func_a('gzuz', 14) # OverflowError
func_a('gzuz', 13) # OverflowErrorfrom rlimit import Limiter, Per
from fastapi import APIRouter, HTTPException, Request
router = APIRouter(...)
@router.get(...)
@Limiter(
times=2, per=4 * Per.SECOND,
key=lambda r: r.client.host,
exception=HTTPException(429)
)
def test_ratelimit(request: Request):
return 'no limit'to implement new memory class,
you have to extend BaseMemory abstract class