增加下载功能
增加包
from flask import Flask, render_template, request,redirect,url_for, Response
from openpyxl import Workbook
增加代码
<a href="/download" class="btn btn-success">点击下载</a>
@app.route('/download')
def download():
message = []
with sqlite3.connect("stu.db") as con: #建立与database.db数据库的连接
cur = con.cursor() #获取游标
cur.execute("select * from students;" ) #添加数据,执行单条的sql语句
message = cur.fetchall()
print(message)
#return message
workbook = Workbook()
worksheet = workbook.active
worksheet.title = "学生管理系统"
worksheet.append([ '姓名', '地址', '城市', '邮编']) #在下一行添加数据
for i in message:
id ,na,add,city,code = i
worksheet.append([na,add,city,code])
workbook.save('huolong.xlsx')
# 保存工作簿到字节流中
import io
output = io.BytesIO()
workbook.save(output)
output.seek(0) # 移动到文件开头
# 创建一个响应对象,将字节流作为内容返回
response = Response(output.getvalue(), mimetype='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
response.headers['Content-disposition'] = 'attachment; filename=student_management.xlsx'
return response
—— 本文来自火龙信奥(义乌睿码科技):义乌青少年信息学奥赛与编程教育平台,专注 CSP-J/S、NOIP、GESP 竞赛培训,线上线下融合教学,助力编程升学。网址:hlcoding.com