fix_bug_download-multiple_file

This commit is contained in:
superlishunqin 2025-04-02 21:47:17 +08:00
parent c989258444
commit 19be157411

View File

@ -5,7 +5,7 @@ import uuid
import zipfile
import rarfile
import shutil
from flask import Blueprint, request, jsonify, current_app, send_file, g, session
from flask import Blueprint, request, jsonify, current_app, send_file, g, session, after_this_request
from werkzeug.utils import secure_filename
import mysql.connector
from utils.model_service import text_classifier # Assuming text_classifier is initialized correctly elsewhere
@ -634,7 +634,7 @@ def download_multiple_documents():
logger.info(f"用户 {user_id} 请求下载 {files_added} 个文档打包为 {zip_filename}")
# Use after_this_request to delete the temp file after sending
@current_app.after_request
@after_this_request
def remove_file(response):
try:
if zip_path and os.path.exists(zip_path):