17 lines
362 B
Python
17 lines
362 B
Python
|
def rate_limit(limit: int, key=None):
|
||
|
|
"""
|
||
|
|
Decorator for configuring rate limit and key in different functions.
|
||
|
|
|
||
|
|
:param limit:
|
||
|
|
:param key:
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
|
||
|
|
def decorator(func):
|
||
|
|
setattr(func, 'throttling_rate_limit', limit)
|
||
|
|
if key:
|
||
|
|
setattr(func, 'throttling_key', key)
|
||
|
|
return func
|
||
|
|
|
||
|
|
return decorator
|