add basic rate limiting
This commit is contained in:
@ -1,6 +1,7 @@
|
||||
import importlib
|
||||
from fastapi import FastAPI
|
||||
from fastapi import FastAPI, Depends
|
||||
from fastapi.responses import JSONResponse
|
||||
from fastapi_throttle import RateLimiter
|
||||
from typing import Optional, Union
|
||||
from .constructors import ValidYTSearchRequest
|
||||
|
||||
@ -22,7 +23,9 @@ class YT(FastAPI):
|
||||
|
||||
for endpoint, handler in self.endpoints.items():
|
||||
app.add_api_route(
|
||||
f"/{endpoint}", handler, methods=["POST"], include_in_schema=True
|
||||
f"/{endpoint}", handler,
|
||||
methods=["POST"], include_in_schema=True,
|
||||
dependencies=[Depends(RateLimiter(times=2, seconds=2))]
|
||||
)
|
||||
|
||||
async def yt_video_search_handler(self, data: ValidYTSearchRequest) -> JSONResponse:
|
||||
|
Reference in New Issue
Block a user