~wanghairui/rag-gateway.git

			@@ -1,21 +1,22 @@
			from fastapi import APIRouter, File, UploadFile
			from fastapi import APIRouter, File, UploadFile, Depends
			from fastapi.responses import JSONResponse, FileResponse
			from fastapi.exceptions import HTTPException
			from sqlalchemy.orm import Session
			from starlette.websockets import WebSocket, WebSocketDisconnect
			from werkzeug.utils import secure_filename
			from app.api import get_current_user_websocket
			from app.models.agent_model import AgentModel, AgentType
			from app.models.base_model import get_db
			from app.models.user_model import UserModel
			from app.utils.excelmerge.conformity import run_conformity
			from pathlib import Path
			import subprocess
			import shutil
			import os


			router = APIRouter()

			ALLOWED_EXTENSIONS = {'xlsx'}
			EXCEL_FILES_PATH = 'data/output'
			SOURCE_FILES_PATH = 'data/source'
			output_path_value = None


			def allowed_file(filename):
			return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
			@@ -26,23 +27,29 @@
			os.makedirs(path)


			@router.post('/excel/upload')
			async def upload_file(files: list[UploadFile] = File(...)):
			if not any(file.filename for file in files):
			return JSONResponse(content={"error": "没有文件部分"}, status_code=400)

			create_dir_if_not_exists(SOURCE_FILES_PATH)

			# 清空SOURCE_FILES_PATH目录
			for filename in os.listdir(SOURCE_FILES_PATH):
			file_path = os.path.join(SOURCE_FILES_PATH, filename)
			# 清理函数
			def clear_directory(path):
			for filename in os.listdir(path):
			file_path = os.path.join(path, filename)
			try:
			if os.path.isfile(file_path) or os.path.islink(file_path):
			os.unlink(file_path)
			elif os.path.isdir(file_path):
			shutil.rmtree(file_path)
			except Exception as e:
			return JSONResponse(content={"error": "文件处理出错"}, status_code=500)
			return {"error": "清空出错"}
			return {"message": "目录已清空"}


			@router.post('/excel/upload')
			async def upload_file(files: list[UploadFile] = File(...)):
			if not any(file.filename for file in files):
			return JSONResponse(content={"error": "没有文件部分"}, status_code=400)

			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)

			save_path_list = []
			for file in files:
			@@ -59,45 +66,49 @@
			return JSONResponse(content={"message": "文件上传成功", "paths": save_path_list}, status_code=201)


			@router.post('/excel/conformity')
			async def run_conformity_api():
			global output_path_value # 声明全局变量
			# ws://localhost:9201/api/document/ws/excel
			@router.websocket("/ws/excel")
			async def ws_excel(websocket: WebSocket):
			await websocket.accept()
			while True:
			data = await websocket.receive_text()
			try:
			if data == "合并Excel文件":
			clear_directory(EXCEL_FILES_PATH)
			output_file_path = run_conformity()
			await websocket.send_json({"step_message": "开始合并", "type": "stream", "files": []})
			elif data == "查询合并进度":
			files = os.listdir(EXCEL_FILES_PATH)
			if not files:
			await websocket.send_json({"step_message": "正在合并中", "type": "stream", "files": []})
			else:
			await websocket.send_json({"step_message": "文档合并成功！", "type": "stream", "files": []})
			elif data == "获取文件":
			files = os.listdir(EXCEL_FILES_PATH)
			if not files:
			await websocket.send_json({"error": "目录下没有生成的文件", "type": "stream", "files": []})
			else:
			first_file = files[0]
			file_name = os.path.basename(first_file)
			file_url = f"./api/document/download/{first_file}"
			await websocket.send_json({
			"step_message": "文档合并成功！",
			"type": "stream",
			"files": [{
			"file_name": file_name,
			"file_url": file_url
			}]
			})
			else:
			await websocket.send_json({"error": "未知指令"})
			except Exception as e:
			await websocket.send_json({"error": str(e)})


			@router.get("/download/{filename}")
			async def download_file(filename: str):
			try:
			create_dir_if_not_exists(EXCEL_FILES_PATH)

			# 清空EXCEL_FILES_PATH目录
			for filename in os.listdir(EXCEL_FILES_PATH):
			file_path = os.path.join(EXCEL_FILES_PATH, filename)
			try:
			if os.path.isfile(file_path) or os.path.islink(file_path):
			os.unlink(file_path)
			elif os.path.isdir(file_path):
			shutil.rmtree(file_path)
			except Exception as e:
			return JSONResponse(content={"error": "文件处理出错"}, status_code=500)

			# 运行方法
			output_path = run_conformity()
			output_path_value = output_path
			return JSONResponse(content={"message": "conformity.py 运行成功", "output_path": str(output_path)}, status_code=200)
			except Exception as e:
			return JSONResponse(content={"error": str(e)}, status_code=500)


			@router.get('/excel/file/status')
			async def get_file_status():
			try:
			return JSONResponse(content={"output_path": str(output_path_value)}, status_code=200)
			except Exception as e:
			return JSONResponse(content={"error": str(e)}, status_code=500)


			@router.get('/excel/download_excel')
			async def download_excel():
			try:
			files = os.listdir(EXCEL_FILES_PATH)
			first_file = files[0]
			return FileResponse(os.path.join(EXCEL_FILES_PATH, first_file), filename=first_file,
			return FileResponse(os.path.join(EXCEL_FILES_PATH, filename), filename=filename,
			media_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
			except FileNotFoundError:
			raise HTTPException(status_code=404, detail="文件不存在")