修复数据库调用问题
This commit is contained in:
8
app.py
8
app.py
@@ -2,6 +2,9 @@ from fastapi import FastAPI
|
||||
from fastapi.responses import FileResponse
|
||||
from fastapi import Request, HTTPException
|
||||
from contextlib import asynccontextmanager
|
||||
from slowapi import Limiter, _rate_limit_exceeded_handler
|
||||
from slowapi.util import get_remote_address
|
||||
from slowapi.errors import RateLimitExceeded
|
||||
from routes import (session, admin, object)
|
||||
import model.database
|
||||
import os, asyncio
|
||||
@@ -30,6 +33,11 @@ app.include_router(admin.Router)
|
||||
app.include_router(session.Router)
|
||||
app.include_router(object.Router)
|
||||
|
||||
# 挂载Slowapi限流中间件
|
||||
limiter = Limiter(key_func=get_remote_address)
|
||||
app.state.limiter = limiter
|
||||
app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)
|
||||
|
||||
@app.get("/")
|
||||
def read_root():
|
||||
if not os.path.exists("dist/index.html"):
|
||||
|
||||
Reference in New Issue
Block a user