用Flask生成word文件

前面我们已经实现了利用python-docx库自动生成表格简化工作流程的python脚本。具体请看:生活中的python-利用python-docx自动生成表格简化工作流程

但是python脚本的缺点就是,如果想让其他人使用,必须在其电脑上安装相应的python环境。这对于电脑不熟练的人来说还是相当有难度的。于是就有了这篇文章---用Flask生成我们需要的word文件,之后将web程序部署到公网服务器上,这样任何人都可以轻松的使用了。

整体思路很简单,就是做一个web表单,然后在Flask里处理最后生成word供下载即可

关于如何生成word部分直接看上面的文章即可,这里说明下Flask部分需要注意的地方

一、使用BYtesIO读写数据

BYtesIO实现在内存中读写数据,我们这里可以先将生成的word文件放在服务器的内存上,然后供下载,代码如下:

f = io.BytesIO()
document.save(f)

f就是我们生成的word文件并且放在了服务器的内存中。

二、Flask下载中文名文件

正常情况下,我们使用Flask的send_file下载文件没有问题,但是如果要下载的文件名是中文的话,就会出现编码错误。

解决方法就是从urllib.parse引入quote,首先对文件名进行编码,然后send_file中作为attachment_filename的参数,这时候能成功下载文件,但是文件名是编码后的名字,要解码的话,我们需要在headers里面声明编码格式,即:

filename = quote(name + '清单.doc')
rv = send_file(f, as_attachment=True, attachment_filename=filename)
rv.headers['Content-Disposition'] += "; filename*=utf-8''{}".format(filename)

这样,对文件名进行UTF-8解码,Flask就可以正常下载中文名文件了

三、同时下载两个(多个)文件问题

因为我的需求是输入一次信息,生成两个不同的word文件。而浏览器似乎无法实现一个submit按钮同时下载两个文件。所以,只能用两个submit按钮下载不同的word文件。这里我的思路是,将两个submit按钮设置成同样的name值,不同的value值,然后通过request.form[]来获取数据:

if request.form['key'] == '申请表下载':
    pass
elif request.form['key'] == '清单下载':
    pass

说明

源代码已经上传至:https://github.com/juventusryp/flask_create_word

下一篇将讲述如何将web程序部署到公网服务器上。

参考链接:
StringIO和BytesIO
Flask 下载中文名文件

你可能感兴趣的:(用Flask生成word文件)