Commit ·
2080245
1
Parent(s): aafdd78
update file 020
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ import uuid
|
|
| 8 |
import sqlite3
|
| 9 |
import csv
|
| 10 |
import zipfile
|
| 11 |
-
from io import
|
| 12 |
|
| 13 |
app = Flask(__name__)
|
| 14 |
|
|
@@ -176,13 +176,17 @@ def export_csv():
|
|
| 176 |
rows = c.fetchall()
|
| 177 |
conn.close()
|
| 178 |
|
| 179 |
-
|
|
|
|
| 180 |
writer = csv.writer(output)
|
| 181 |
writer.writerow(["ID", "First Name", "Last Name", "Email", "Issue", "Priority", "Date", "Generated"])
|
| 182 |
writer.writerows(rows)
|
| 183 |
|
|
|
|
| 184 |
output.seek(0)
|
| 185 |
-
return Response(output, mimetype="text/csv",
|
|
|
|
|
|
|
| 186 |
|
| 187 |
@app.route("/export/zip")
|
| 188 |
def export_zip():
|
|
|
|
| 8 |
import sqlite3
|
| 9 |
import csv
|
| 10 |
import zipfile
|
| 11 |
+
from io import StringIO
|
| 12 |
|
| 13 |
app = Flask(__name__)
|
| 14 |
|
|
|
|
| 176 |
rows = c.fetchall()
|
| 177 |
conn.close()
|
| 178 |
|
| 179 |
+
# Use StringIO here
|
| 180 |
+
output = StringIO()
|
| 181 |
writer = csv.writer(output)
|
| 182 |
writer.writerow(["ID", "First Name", "Last Name", "Email", "Issue", "Priority", "Date", "Generated"])
|
| 183 |
writer.writerows(rows)
|
| 184 |
|
| 185 |
+
# Encode string to bytes before returning
|
| 186 |
output.seek(0)
|
| 187 |
+
return Response(output.getvalue(), mimetype="text/csv",
|
| 188 |
+
headers={"Content-Disposition": "attachment;filename=tickets.csv"})
|
| 189 |
+
|
| 190 |
|
| 191 |
@app.route("/export/zip")
|
| 192 |
def export_zip():
|