FrederickSundeep commited on
Commit
2080245
·
1 Parent(s): aafdd78

update file 020

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -8,7 +8,7 @@ import uuid
8
  import sqlite3
9
  import csv
10
  import zipfile
11
- from io import BytesIO
12
 
13
  app = Flask(__name__)
14
 
@@ -176,13 +176,17 @@ def export_csv():
176
  rows = c.fetchall()
177
  conn.close()
178
 
179
- output = BytesIO()
 
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", headers={"Content-Disposition": "attachment;filename=tickets.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():