flask使用pymysql连接MySQL,生成xls文件并下载到本地
flask框架连接MySQL,我们使用pymsql这个工具,如下操作:
from flask import Flask import pymysql import xlwt from io import BytesIO from flask import make_response db = pymysql.connect(host='127.0.0.1',port=3306,user='user', passwd='pwd', db='db_name', charset='utf8') # 创建链接 def data_db(): # 去数据库取数据,假设数据库表只有4个字段 sql = """ select * from table_name; """ cc = db.cursor() # 其实就是用来获得python执行Mysql命令的方法,也就是
我们所说的操作游标 cc.execute(sql) # 真正执行sql语句 cn = cc.fetchall() # 接收全部的返回结果行row return cn # 返回 app = Flask(__name__) @app.route('/index_up') # 设置路由 def index_list(): # 执行视图函数 ret = data_db() # 获取数据 wb = xlwt.Workbook(encoding='utf-8') # 实例化,有encoding和style_compression参数 ws = wb.add_sheet("111", cell_overwrite_ok=True) # Workbook的方法,生成名为111.xls文件 row0 = ['字段1', '字段2', '字段3', '字段4'] # 指定xls文件的字段 for i in range(0, len(row0)): # 将这些字段写入111.xls文件 ws.write(0, i, row0[i]) k = 1 for i in ret: # 循环每一列 for j in range(4): # 在每列添加数据 ws.write(k, j, i[j])
k += 1 sio = BytesIO() # 将获取的数据在内存中写,有时会用到StringIO() wb.save(sio) # 将文件流保存 sio.seek(0) # 光标 response = make_response(sio.getvalue()) # response.headers['Content-type'] = 'application/vnd.ms-excel' # 指定返回的类型 response.headers['Transfer-Encoding'] = 'chunked' response.headers['Content-Disposition'] = 'attachment;filename=111.xls' # 设定用户浏览器显示的保存文件名 return response # 返回response,浏览器会出现如下效果,如果返回其他,比如None就会只下载不在浏览器提示。 if __name__ == '__main__': app.run()