在国企,很多领导是按部就班的晋升上来的,他们的年龄及经历使他们至今还没有掌握电脑的基础知识,因此,纸质报表是他们掌握全局的依据,必备的工具.将报表下载下来,然后能稍许进行修饰,这是必须的.
项目描述
将页面原始呈现的数据及搜索后的数据以 excel 报表的形式下载下来.
知识难点
单一的 url 页面中会呈现出不同的数据,需要能够把不同的数据下载下来,只需要用到 ajax 的知识,在同一个页面中以 json 的形式传递数据.
实现页面模板
不再赘述,仅仅是在页面上加载一个下载按钮,详情看源码.
实现下载功能的试图函数
views.py 的内容如下:
很明显,通过 return 返回的 jsonify 数据,大家应该知道,我这是要用 json 了.
...
...
import os
from collections import OrderedDict
from config import basedir
from pyexcel_xls import save_data, get_data
...
...
@show.route('/<path:filename>', methods=['GET', 'POST'])
def download_xls(filename):
data = OrderedDict()
data_path = os.path.join(excel_path, filename)
num = 0
if '01.xls' in filename:
header_data = ['序号', '部门', '角色', '员工', '电话', '客户', '月份',
'本月资产余额', '上月资产余额', '新增余额']
body_data = [header_data]
for t in json.loads(request.get_data()):
num += 1
body_data.append([num, t["department"], t["role"], t["staff_name"], t["phone"], t["guest_name"],
t["month"], t["balance"], t["last_balance"],
float('%.2f' % t["balance"]) - float('%.2f' % t["last_balance"])])
data.update({'01 报表': body_data})
if '02.xls' in filename:
header_data = ['序号', '部门', '角色', '员工', '电话', '月份', '管户数'
'本月资产余额', '上月资产余额', '新增余额']
body_data = [header_data]
for t in json.loads(request.get_data()):
num += 1
body_data.append([num, t["department"], t["role"], t["staff_name"], t["staff_phone"],
t["month"], t["balance"], t["last_balance"],
float('%.2f' % t["balance"]) - float('%.2f' % t["last_balance"])])
data.update({'02 报表': body_data})
save_data(data_path, data)
return jsonify({"data": "ok"})
而其中的两个 filename 的判断,我是因为要做两个页面的下载链接,所以对 filename 进行了一个判断.
代码中使用了 collections 的有序字典 OrderedDict, pyexcel_xls 的相关模块,同时也设置了要生成的 xls 文件的生成路径及文件名--
实现 ajax 功能
data_path = os.path.join(excel_path, filename)
.
既然是 ajax 那就是 javascript 咯,还得返回 html 模板.在 flask 项目中,模板的变量都是封装在 jinja2 中,我们写 javascript 的时候,不能单独再建立一个 js 文件,导入到模板中,因为,无法获得 Jinja2 变量.还得直接在 html 中写 javasript.
01.html 中实现 ajax 下载
详情请看注释.
{% block js %}
{{ super() }} <!-- 继承 block js 里的原始 js 文件 -->
<script>
$SCRIPT_ROOT = {{ request.script_root|tojson|safe }}; <!-- flask ajax 需要进行该项设置 -->
function downloadxls() { <!-- 创建 downloadxls 函数,页面上的导出报表按钮需要调用这个函数 -->
var database = new Object();
database = {{ database|safe }}; <!-- 调用 jinja2 中的变量,使用 safe ,防止出现多重冒号. -->
var data = JSON.stringify(database);
$.ajax({
type: "POST",
url: "{{ url_for('show.download_xls', filename='%s_01.xls'% current_user.phone) }}",<!-- 生成 xls 文件的 url 链接 -->
contentType: "application/json; charset=utf-8",
data: data, <!-- 返回到后台的 json 数据 -->
success: function (msg) {
if (msg.data === 'ok'){ <!-- 后台送来的'ok' 数据 -->
//window.open("{{ url_for('show.static', filename='excel_files/%s_01.xls'% current_user.phone) }}");<!-- 用 js 打开 xls 文件,可以进行下载,但是会出现页面闪现的现象,界面不算友好. -->
var $eleForm = $("<form method='get'></form>");<!-- 使用 form 表单的形式进行文件下载,不会出现页面闪现的现象. -->
$eleForm.attr("action", "{{ url_for('show.static', filename='excel_files/%s_01.xls'% current_user.phone) }}");
$(document.body).append($eleForm);
// 提交表单,实现下载
$eleForm.submit();
}
}
})
}
</script>
{% endblock %}
02.html 中实现 ajax 下载
不在赘述,基本上是和 02.html 页面一样.
至此,你可以实验一下,你的项目是否可以实现报表下载了.
很不幸,我们忘了一个重要的事情,那就是 js 中 database 是继承自 jinja2 变量,它是否是你所想的那种 json 数据呢.答案是否定的.结果是这样的:
018.png
那么我们需要继续工作咯.
使 sqlalchemy 数据 json 化
models.py 中建立一个类:
详细功能请看注释.
class AlchemyJsonEncoder(json.JSONEncoder):
def default(self, obj):
# 判断是否是 Query
if isinstance(obj, Query):
# 定义一个字典数组
fields = []
# 检索结果集的行记录
for rec in obj:
# 定义一个字典对象
record = {}
# 检索记录中的成员
for field in [x for x in dir(rec) if
# 过滤属性
not x.startswith('_')
# 过滤掉方法属性
and hasattr(rec.__getattribute__(x), '__call__') == False
# 过滤掉不需要的属性
and x != 'metadata']:
try:
record[field] = rec.__getattribute__(field)
except TypeError:
record[field] = None
fields.append(record)
# 返回字典数组
return fields
# 其他类型的数据按照默认的方式序列化成 JSON
return json.JSONEncoder.default(self, obj)
在 01 , 02 两个 url 链接的试图函数中,加入如下内容:
views.py 中增加
return render_template('show/01.html', data=data, searchForm=search_form, database=json.dumps(database, cls=AlchemyJsonEncoder))
此时,在查看我们要传输的 json 数据:
019.png
至此,你可以使用下载功能了,不管你的页面中的数据如何变化,你下载的报表都会显示你当前页面的数据.
源码下载
提示: 让这么多的数据在页面中使用明文传输,是一种极其不安全的行为,这是网络爬虫的肥沃矿场.当然,如果你使用的环境是我这样的内部 web 环境,那就另当别论.不过如果你有更好的解决方法,那么可以告诉我,不胜感激.
来源: http://www.jianshu.com/p/4498b726247e