FormatCSVAction.run()

Complexity

Conditions

5

Size

Total Lines

19

Duplication

Lines	0
Ratio	0 %

Importance

Changes	1
Bugs	0	Features	1

import csv
    def run(self, data, delimiter=',', quote_char='"'):
        if len(data) == 0:
            raise InvalidActionParameterException("data has no rows")
        if not isinstance(data, list):
            raise InvalidActionParameterException("data must be a list")
        if not isinstance(data[0], dict):
            raise InvalidActionParameterException("data must be a list of dict")

        fieldnames = data[0].keys()
        sh = StringIO()
        writer = csv.DictWriter(sh, fieldnames=fieldnames)

        writer.writeheader()
        for row in data:
            writer.writerow(row)

        out = sh.get_value()
        sh.close()
        return out