xuyonghao
2024-12-17 2dc4a7392eef26fdadd00fde1baf8b471ab25ca5
app/api/excel.py
@@ -56,8 +56,7 @@
        if file.filename == '':
            return JSONResponse(content={"error": "没有选择文件"}, status_code=400)
        if file and allowed_file(file.filename):
            filename = secure_filename(file.filename)
            save_path = os.path.join(SOURCE_FILES_PATH, filename)
            save_path = os.path.join(SOURCE_FILES_PATH, file.filename)
            with open(save_path, 'wb') as buffer:
                shutil.copyfileobj(file.file, buffer)
            save_path_list.append(save_path)
@@ -73,16 +72,12 @@
    create_dir_if_not_exists(SOURCE_FILES_PATH)
    create_dir_if_not_exists(EXCEL_FILES_PATH)
    clear_directory(SOURCE_FILES_PATH)
    clear_directory(EXCEL_FILES_PATH)
    while True:
        data = await websocket.receive_text()
        try:
            if data == "\"合并Excel\"":
                clear_directory(EXCEL_FILES_PATH)
                output_file_path = run_conformity()
                clear_directory(EXCEL_FILES_PATH)
                output_file_path = run_conformity()
                files = os.listdir(EXCEL_FILES_PATH)
                if files:
@@ -90,7 +85,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 +93,7 @@
                        }]
                    })
                    await websocket.send_json({
                        "step_message": "",
                        "message": "文档合并成功!",
                        "type": "close",
                    })
                else: