博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
Flask下载文件
阅读量:6156 次
发布时间:2019-06-21

本文共 2838 字,大约阅读时间需要 9 分钟。

前言

由于最近在做文件管理模块的功能,所以难免会遇到文件上传下载这块的功能。不过文件上传那块是调用的OSS api,所以接触的不多。

文件的下载:

1. 接口返回真实的文件

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

核心代码如下:

from flask import send_file, send_from_directoryimport os@app.route("/download/
", 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里边出现中文的时候,会报如下错误:

image.png

解决办法:

使用flask自带的make_response
代码修改如下

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

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

2. 接口返回文件数据流

这种情况比较适合我现在的需求,因为我这边是用requests库,先请求一个oss链接,获取到文件的数据,然后我发现目前flask没有这样的api实现,这里还是使用make_response方法实现。

代码如下:

import mimetypes@app.route('/fileManager/download/
/
/
', methods=['GET'])def download_file(projId, id, filename): try: url = "your url" r = requests.get(url, timeout=500) if r.status_code != 200: raise Exception("Cannot connect with oss server or file is not existed") response = make_response(r.content) mime_type = mimetypes.guess_type(filename)[0] response.headers['Content-Type'] = mime_type response.headers['Content-Disposition'] = 'attachment; filename={}'.format(filename.encode().decode('latin-1')) return response except Exception as err: print('download_file error: {}'.format(str(err))) logging.exception(err) return Utils.beop_response_error(msg='Download oss files failed!')

解释一下:

make_response很强大,下载一个文件,需要在response的headers里边添加一些信息,比如文件的类型,文件的名字,是否以附件形式添加,这3个是比较关键的信息。
mime_type是文件的类型,我观察send_file的源代码发现里边用到了mimetypes.guess_type()这个方法,也就是猜测文件的类型,然后这里我就直接搬过来用了哈哈,r.content其实就是文件的数据流,之前我是通过

with open(filename, 'wb') as file:    file.write(r.content)

这样实现下载文件到本地的,所以其实r.content是一个文件数据流,也不清楚我的名词用的是否恰当哈哈。

之所以不用第一种方式,是因为我本地生成文件了之后,需要删除他,但是删除的时候总是会提示该文件已经被另一个程序使用,所以猜测是send_file这个api还在使用该文件,为了达到更好的效果,找到了第二种解决办法。

其实还有一种解决办法:

3. 发送静态文件

其实原来和第一种差不多,调用的api不一样,api是

from flask import appimport os@app.route("/download/
", methods=['GET'])def download_file(filepath): # 此处的filepath是文件的路径,但是文件必须存储在static文件夹下, 比如images\test.jpg return app.send_static_file(filepath)

转载地址:http://fwdfa.baihongyu.com/

你可能感兴趣的文章
在VMware网络测试“专用VLAN”功能
查看>>
使用Formik轻松开发更高质量的React表单(三)<Formik />解析
查看>>
也问腾讯:你把用户放在什么位置?
查看>>
CSS Sprites 样式生成工具(bg2css)
查看>>
[转]如何重构代码--重构计划
查看>>
类中如何对list泛型做访问器??
查看>>
C++解析XML--使用CMarkup类解析XML
查看>>
P2P应用层组播
查看>>
Sharepoint学习笔记—修改SharePoint的Timeouts (Execution Timeout)
查看>>
CSS引入的方式有哪些? link和@import的区别?
查看>>
Redis 介绍2——常见基本类型
查看>>
asp.net开发mysql注意事项
查看>>
(转)Cortex-M3 (NXP LPC1788)之EEPROM存储器
查看>>
ubuntu set defult jdk
查看>>
[译]ECMAScript.next:TC39 2012年9月会议总结
查看>>
【Xcode】编辑与调试
查看>>
用tar和split将文件分包压缩
查看>>
[BTS] Could not find stored procedure 'mp_sap_check_tid'
查看>>
PLSQL DBMS_DDL.ALTER_COMPILE
查看>>
Activity生命周期
查看>>