Last active
April 1, 2016 17:51
-
-
Save PabloLefort/21c97202af933608c916 to your computer and use it in GitHub Desktop.
Django utf-8 csv export
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
''' | |
Using the same example on Python 2.7 docs | |
''' | |
import csv, codecs, cStringIO | |
class UnicodeWriter: | |
""" | |
A CSV writer which will write rows to CSV file "f", | |
which is encoded in the given encoding. | |
""" | |
def __init__(self, f, dialect=csv.excel, encoding="utf-8", **kwds): | |
# Redirect output to a queue | |
self.queue = cStringIO.StringIO() | |
self.writer = csv.writer(self.queue, dialect=dialect, **kwds) | |
self.stream = f | |
self.encoder = codecs.getincrementalencoder(encoding)() | |
def writerow(self, row): | |
self.writer.writerow([s.encode("utf-8") for s in row]) | |
# Fetch UTF-8 output from the queue ... | |
data = self.queue.getvalue() | |
data = data.decode("utf-8") | |
# ... and reencode it into the target encoding | |
data = self.encoder.encode(data) | |
# write to the target stream | |
self.stream.write(data) | |
# empty queue | |
self.queue.truncate(0) | |
def writerows(self, rows): | |
for row in rows: | |
self.writerow(row) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from django.shortcuts import HttpResponse | |
from .django-csv-export-utf8 import UnicodeWriter | |
def csvexport(request): | |
response = HttpResponse(content_type='text/csv') | |
name = 'foo.csv' | |
response['Content-Disposition'] = 'attachment; filename='+name | |
''' | |
Here we combine Django example: | |
https://docs.djangoproject.com/es/1.9/howto/outputting-csv/ | |
with UnicodeWriter class | |
''' | |
writer = UnicodeWriter(response) | |
writer.writerow(["Bar", "1", "2"]) | |
return response |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment