Skip to content

Commit 6d406f8

Browse files
committed
feat: import Sample SQL
1 parent baf8c93 commit 6d406f8

File tree

4 files changed

+17
-7
lines changed

4 files changed

+17
-7
lines changed

backend/apps/data_training/api/data_training.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
from apps.data_training.models.data_training_model import DataTrainingInfo
1717
from common.core.config import settings
1818
from common.core.deps import SessionDep, CurrentUser, Trans
19+
from common.core.file import FileRequest
1920
from common.utils.data_format import DataFormat
2021

2122
router = APIRouter(tags=["DataTraining"], prefix="/system/data-training")
@@ -214,11 +215,12 @@ def inner():
214215
return await asyncio.to_thread(inner)
215216

216217

217-
@router.get("/download-fail-info/{filename}")
218-
async def download_excel(filename: str, trans: Trans):
218+
@router.post("/download-fail-info")
219+
async def download_excel(req: FileRequest):
219220
"""
220221
根据文件路径下载 Excel 文件
221222
"""
223+
filename = req.file
222224
file_path = os.path.join(path, filename)
223225

224226
# 检查文件是否存在

backend/common/core/file.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
from pydantic import BaseModel
2+
3+
class FileRequest(BaseModel):
4+
file: str

frontend/src/api/training.ts

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,12 @@ export const trainingApi = {
1616
requestOptions: { customError: true },
1717
}),
1818
downloadError: (path: any) =>
19-
request.get(`/system/data-training/download-fail-info/${path}`, {
20-
responseType: 'blob',
21-
requestOptions: { customError: true },
22-
}),
19+
request.post(
20+
`/system/data-training/download-fail-info`,
21+
{ file: path },
22+
{
23+
responseType: 'blob',
24+
requestOptions: { customError: true },
25+
}
26+
),
2327
}

frontend/src/views/system/training/index.vue

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,7 @@ const onSuccess = (response: any) => {
134134
})
135135
const link = document.createElement('a')
136136
link.href = URL.createObjectURL(blob)
137-
link.download = `${t('training.data_training')}_error.xlsx`
137+
link.download = response.data.error_excel_filename
138138
document.body.appendChild(link)
139139
link.click()
140140
document.body.removeChild(link)

0 commit comments

Comments
 (0)