report.py 文件源码

 复制代码def get_csv(self):
        csvout = StringIO()
        csvwriter = csv.writer(
            csvout,
            dialect='excel',
            quoting=csv.QUOTE_NONNUMERIC
        )
        csvwriter.writerow(self.column_names)
        for row in self.rows:
            row_formatted = []
            for cell in row:
                if isinstance(cell, datetime.datetime):
                    cell = cell.strftime('%Y-%m-%d %H:%M')
                elif isinstance(cell, (int, long)):
                    cell = str(cell)
                elif isinstance(cell, (list, tuple)):
                    cell = str(cell)
                elif cell is None:
                    cell = ''
                else:
                    cell = cell.encode('utf8')
                row_formatted.append(cell)
            try:
                csvwriter.writerow(row_formatted)
            except Exception, e:
                raise Exception("%s: %s, %s"%(e, row, row_formatted))
        csvout.seek(0)
        return csvout.read()