64 lines
2.0 KiB
Python
64 lines
2.0 KiB
Python
from fastapi import FastAPI, Request
|
|
from starlette.middleware.base import BaseHTTPMiddleware, RequestResponseEndpoint
|
|
from starlette.responses import Response
|
|
|
|
from utils.client_ip_util import ClientIPUtil
|
|
from utils.log_util import logger
|
|
from utils.response_util import ResponseUtil
|
|
|
|
|
|
class DemoModeMiddleware(BaseHTTPMiddleware):
|
|
"""
|
|
演示模式中间件
|
|
"""
|
|
|
|
async def dispatch(self, request: Request, call_next: RequestResponseEndpoint) -> Response:
|
|
"""
|
|
演示模式下拦截指定请求
|
|
"""
|
|
url_path = str(request.url)
|
|
method = request.method.lower()
|
|
intercept_url_list = [
|
|
'system/user',
|
|
'system/role',
|
|
'system/menu',
|
|
'system/dept',
|
|
'system/post',
|
|
'system/dict',
|
|
'system/config',
|
|
'system/notice',
|
|
'monitor/operlog',
|
|
'monitor/logininfor',
|
|
'monitor/online',
|
|
'monitor/job',
|
|
'monitor/jobLog',
|
|
'monitor/cache',
|
|
'ai/model',
|
|
'ai/chat',
|
|
]
|
|
|
|
for item in intercept_url_list:
|
|
if (url_path.startswith(f'{request.base_url!s}{item}') and method != 'get') or url_path.startswith(
|
|
(
|
|
f'{request.base_url!s}common',
|
|
f'{request.base_url!s}register',
|
|
f'{request.base_url!s}tool/gen/createTable',
|
|
)
|
|
):
|
|
operate_ip = ClientIPUtil.get_client_ip(request)
|
|
logger.warning(
|
|
'请求IP:{}||请求API:{}||请求方法:{}||请求结果:演示模式，不允许操作！', operate_ip, url_path, method
|
|
)
|
|
return ResponseUtil.failure(msg='演示模式，不允许操作！')
|
|
response = await call_next(request)
|
|
return response
|
|
|
|
|
|
def add_demo_mode_middleware(app: FastAPI) -> None:
|
|
"""
|
|
添加演示模式中间件
|
|
|
|
:param app: FastAPI对象
|
|
"""
|
|
app.add_middleware(DemoModeMiddleware)
|