Python Flask循环触发多个return redirect仅返回1个下载文件如何解决

错误原因 #

问题出在循环内部直接调用了return，Flask的视图函数只要执行到return语句就会直接终止运行，将响应返回给用户，因此循环只会执行第一次迭代，生成第一个文件后就跳转下载，剩余的迭代逻辑完全不会运行。

可行解决方案 #

方案1：将多个文件打包成压缩包返回（最推荐，用户仅需下载一次） #

先完成所有拆分文件的生成，再用Python内置的zipfile模块将所有文件打包为单个压缩包返回，示例代码如下：

import zipfile
import os
from flask import send_file

# 先完成所有文件的生成，全程不插入return
generated_files = []
for i in rank_list: 
    new = df[df["Rank"] == i]
    filename = f'file_{round(i)}.csv'
    file_path = os.path.join(DOWNLOAD_DIRECTORY, filename)
    new.to_csv(file_path)
    generated_files.append(file_path)

# 打包所有生成的文件为zip压缩包
zip_path = os.path.join(DOWNLOAD_DIRECTORY, 'split_result.zip')
with zipfile.ZipFile(zip_path, 'w') as zf:
    for file in generated_files:
        # arcname参数控制压缩包内的文件名，避免写入完整路径
        zf.write(file, arcname=os.path.basename(file))

# 统一返回压缩包下载响应
return send_file(zip_path, as_attachment=True, download_name='split_result.zip')

方案2：前端触发批量下载（适用于必须逐个下载的场景） #

如果业务要求不能打包，需逐个下载文件，可先由后端生成所有文件后返回下载链接列表，前端用JS循环触发每个链接的下载：

1. 后端生成所有拆分文件后，返回所有下载地址的列表
2. 前端接收到列表后，通过动态创建a标签触发下载，示例JS逻辑如下：

const downloadUrlList = ["/download/file_1.csv", "/download/file_2.csv", "/download/file_3.csv"]
downloadUrlList.forEach(url => {
  const a = document.createElement('a')
  a.href = url
  a.download = url.split('/').pop()
  document.body.appendChild(a)
  a.click()
  document.body.removeChild(a)
})

注意该方案可能会被浏览器的弹窗拦截规则限制，需要用户授权网站的弹窗/下载权限。

内容的提问来源于stack exchange，提问作者Vanassins