文件上传

zhangxiao

2024-10-23 ab21f16ec2a4054ed1ca9ccea76ad192bd1e5c7e

 app/api/excel.py

@@ -56,15 +56,15 @@
        if file.filename == '':
            return JSONResponse(content={"error": "没有选择文件"}, status_code=400)
        if file and allowed_file(file.filename):
            filename = secure_filename(file.filename)
            filename = file.filename
            save_path = os.path.join(SOURCE_FILES_PATH, filename)
            with open(save_path, 'wb') as buffer:
                shutil.copyfileobj(file.file, buffer)
            save_path_list.append(save_path)
        else:
            return JSONResponse(content={"error": "不允许的文件类型"}, status_code=400)
    return JSONResponse(content={"code": 200, "msg": "", "data": {}}, status_code=200)

    return JSONResponse(content={"code": 200, "msg": "文件上传成功", "data": {"file_paths": save_path_list}},
                        status_code=200)

# ws://localhost:9201/api/document/ws/excel
@router.websocket("/ws/excel")
@@ -90,7 +90,7 @@
                    file_name = os.path.basename(first_file)
                    file_url = f"./api/document/download/{first_file}"
                    await websocket.send_json({
                        "step_message": "文档合并成功！",
                        "message": "文档合并成功！",
                        "type": "stream",
                        "files": [{
                            "file_name": file_name,
@@ -98,7 +98,7 @@
                        }]
                    })
                    await websocket.send_json({
                        "step_message": "",
                        "message": "文档合并成功！",
                        "type": "close",
                    })
                else: