mirror of
https://github.com/jwinterm/modoboa.git
synced 2026-10-19 05:58:34 +02:00
22 lines
499 B
Python
22 lines
499 B
Python
"""Custom renderers for DRF."""
|
|
|
|
import csv
|
|
from io import StringIO
|
|
|
|
from rest_framework import renderers
|
|
|
|
|
|
class CSVRenderer(renderers.BaseRenderer):
|
|
"""Custom CSV renderer."""
|
|
|
|
media_type = "text/csv"
|
|
format = "csv"
|
|
|
|
def render(self, data, media_type=None, renderer_context=None):
|
|
with StringIO() as fp:
|
|
csvwriter = csv.writer(fp)
|
|
for item in data:
|
|
csvwriter.writerow(item)
|
|
content = fp.getvalue()
|
|
return content
|