Spaces:
Sleeping
Sleeping
File size: 918 Bytes
5564ecb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from math import ceil
from fastapi import FastAPI, Request, Response
from fastapi.routing import APIRoute
from common.exception import errors
def ensure_unique_route_names(app: FastAPI) -> None:
"""
:param app: FastAPI
:return:
"""
temp_routes = set()
for route in app.routes:
if isinstance(route, APIRoute):
if route.name in temp_routes:
raise ValueError(f'Non-unique route name: {route.name}')
temp_routes.add(route.name)
async def http_limit_callback(request: Request, response: Response, expire: int) -> None:
"""
:param request: FastAPI
:param response: FastAPI
:param expire:
:return:
"""
expires = ceil(expire / 1000)
raise errors.HTTPError(code=429, msg='The request is too frequent, please try again later', headers={'Retry-After': str(expires)})
|