CSVRenderer

Complexity

Total Complexity

4

Size/Duplication

Total Lines	23
Duplicated Lines	0 %

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__init__()	0	2	1
A	__call__()	0	19	3

# http://docs.pylonsproject.org/projects/pyramid-cookbook/en/latest/templates/customrenderers.html

import csv
import io

class CSVRenderer(object):
   def __init__(self, info):
      pass

   def __call__(self, value, system):
      """ Returns a plain CSV-encoded string with content-type
      ``text/csv``. The content-type may be overridden by
      setting ``request.response.content_type``."""

      request = system.get('request')
      if request is not None:
         response = request.response
         ct = response.content_type
         if ct == response.default_content_type:
            response.content_type = 'text/csv'

      fout = io.StringIO()
      writer = csv.writer(fout, delimiter=',', quotechar=',', quoting=csv.QUOTE_MINIMAL)

      writer.writerow(value.get('header', []))
      writer.writerows(value.get('rows', []))

      return fout.getvalue()