mirror of
https://github.com/MemeMeow-Studio/MemeMeow.git
synced 2026-10-17 04:41:01 +08:00
fQwQf
8d37784edf
- 重构 api_settings.py，使用 load_config 函数动态加载配置 - 优化保护模式中间件，规范化路径比较并改进错误响应 - 在 api.py 中更新配置加载方式，使用新的 load_config 函数
30 lines
1.0 KiB
Python
30 lines
1.0 KiB
Python
from fastapi import Request, HTTPException
|
|
from fastapi.responses import JSONResponse
|
|
from starlette.middleware.base import BaseHTTPMiddleware
|
|
|
|
class ProtectedModeMiddleware(BaseHTTPMiddleware):
|
|
def __init__(self, app, config):
|
|
super().__init__(app)
|
|
self.config = config
|
|
|
|
async def dispatch(self, request: Request, call_next):
|
|
|
|
# 检查保护模式状态
|
|
if self.config.protected_mode:
|
|
# 标准化路径：移除末尾斜杠
|
|
path = request.url.path.rstrip('/')
|
|
allowed_paths = [p.rstrip('/') for p in self.config.allowed_endpoints]
|
|
|
|
# 放行白名单端点
|
|
if path in allowed_paths:
|
|
return await call_next(request)
|
|
|
|
# 拦截其他请求
|
|
return JSONResponse(
|
|
status_code=403,
|
|
content={
|
|
"detail": f"Protected mode blocked request to {path}"
|
|
}
|
|
)
|
|
|
|
return await call_next(request) |