Flask-实现下载功能

1. 接口返回真实的文件

这种情况比较简单， flask里带有此类api, 可以用send_from_directory和send_file.

核心代码如下:

from flask import send_file, send_from_directory
import os


@app.route("/download/<filename>", methods=['GET'])
def download_file(filename):
    # 需要知道2个参数, 第1个参数是本地目录的path, 第2个参数是文件名(带扩展名)
    directory = os.getcwd()  # 假设在当前目录
    return send_from_directory(directory, filename, as_attachment=True)

后边那个as_attachment参数需要赋值为True，不过此种办法有个问题，就是当filename里边出现中文的时候，会报如下错误:

解决办法:
使用flask自带的make_response
代码修改如下

from flask import send_file, send_from_directory
import os
from flask import make_response


@app.route("/download/<filename>", methods=['GET'])
def download_file(filename):
    # 需要知道2个参数, 第1个参数是本地目录的path, 第2个参数是文件名(带扩展名)
    directory = os.getcwd()  # 假设在当前目录
    response = make_response(
        send_from_directory(excel_path, filename.encode('utf-8').decode('utf-8'), as_attachment=True))
    response.headers["Content-Disposition"] = "attachment; filename={}".format(filename.encode().decode('latin-1'))
    return response

使用make_response函数建立一个response对象，然后将filename编码转为latin-1，可以看到server.py里边会严格按照latin-1编码来解析filename，所以我这里的做法是先将utf8编码的中文文件名默认转为latin-1编码。