This commit is contained in:
dataeaseShu
2025-07-28 10:10:04 +08:00
4 changed files with 13 additions and 27 deletions
+2 -2
View File
@@ -36,7 +36,7 @@ class TokenMiddleware(BaseHTTPMiddleware):
request.state.current_user = validator[1]
request.state.assistant = validator[2]
return await call_next(request)
return JSONResponse({"msg": f"Unauthorized:[{validator[1]}]"}, status_code=401, headers={"Access-Control-Allow-Origin": "*"})
return JSONResponse(f"Unauthorized:[{validator[1]}]", status_code=401, headers={"Access-Control-Allow-Origin": "*"})
#validate pass
tokenkey = settings.TOKEN_KEY
token = request.headers.get(tokenkey)
@@ -44,7 +44,7 @@ class TokenMiddleware(BaseHTTPMiddleware):
if validate_pass:
request.state.current_user = data
return await call_next(request)
return JSONResponse({"msg": f"Unauthorized:[{data}]"}, status_code=401, headers={"Access-Control-Allow-Origin": "*"})
return JSONResponse(f"Unauthorized:[{data}]", status_code=401, headers={"Access-Control-Allow-Origin": "*"})
def is_options(self, request: Request):
return request.method == "OPTIONS"
+5 -16
View File
@@ -14,7 +14,8 @@ class ResponseMiddleware(BaseHTTPMiddleware):
if isinstance(response, JSONResponse) or request.url.path == f"{settings.API_V1_STR}/openapi.json":
return response
if response.status_code != 200:
return response
if response.headers.get("content-type") == "application/json":
try:
body = b""
@@ -51,11 +52,7 @@ class ResponseMiddleware(BaseHTTPMiddleware):
SQLBotLogUtil.error(f"Response processing error: {str(e)}", exc_info=True)
return JSONResponse(
status_code=500,
content={
"code": 500,
"data": None,
"msg": str(e)
},
content=str(e),
headers={
k: v for k, v in response.headers.items()
if k.lower() not in ("content-length", "content-type")
@@ -71,11 +68,7 @@ class exception_handler():
SQLBotLogUtil.error(f"HTTP Exception: {exc.detail}", exc_info=True)
return JSONResponse(
status_code=exc.status_code,
content={
"code": exc.status_code,
"msg": exc.detail,
"data": None
},
content=exc.detail,
headers={"Access-Control-Allow-Origin": "*"}
)
@@ -85,11 +78,7 @@ class exception_handler():
SQLBotLogUtil.error(f"Unhandled Exception: {str(exc)}", exc_info=True)
return JSONResponse(
status_code=500,
content={
"code": 500,
"msg": str(exc),
"data": None
},
content=str(exc),
headers={"Access-Control-Allow-Origin": "*"}
)
+4 -7
View File
@@ -171,9 +171,9 @@ class HttpService {
errorMessage = 'Invalid request parameters'
break
case 401:
// eslint-disable-next-line @typescript-eslint/ban-ts-comment
// @ts-ignore
errorMessage = error.response?.data?.data?.msg || 'Unauthorized, please login again'
errorMessage = error.response?.data
? error.response.data.toString()
: 'Unauthorized, please login again'
// Redirect to login page if needed
ElMessage({
message: errorMessage,
@@ -199,10 +199,7 @@ class HttpService {
errorMessage = `Server responded with error: ${error.response.status}`
}
if (error?.response?.data) {
const msgData: any = error.response.data
if (msgData?.msg) {
errorMessage = msgData.msg
}
errorMessage = error.response.data.toString()
}
} else if (error.request) {
errorMessage = 'No response from server'
@@ -76,7 +76,7 @@ const advancedSettingPagination = computed(() => {
return advancedSetting.value.slice(currentPage.value * 5 - 5, currentPage.value * 5)
})
const handelCurrentChange = (val: any) => {
const handleCurrentChange = (val: any) => {
currentPage.value = val
}
@@ -318,7 +318,7 @@ defineExpose({
:default-page-size="5"
layout="prev, pager, next"
:total="advancedSetting.length"
@current-change="handelCurrentChange"
@current-change="handleCurrentChange"
/>
</div>
</div>