# vim: set fileencoding=utf-8 sw=4 ts=4 et :

# Copyright (C) 2007-2020 CS GROUP - France

# License: GNU GPL v2 <http://www.gnu.org/licenses/gpl-2.0.html>

"""Fonction d'export des alertes au format CSV."""

import csv

from cStringIO import StringIO

from tg import config

def export(page, plugins_data):

    buf = StringIO()

    quoting = config.get('csv_quoting', 'ALL').upper()

    if quoting not in ('ALL', 'MINIMAL', 'NONNUMERIC', 'NONE'):

        quoting = 'ALL'

    csv_writer = csv.DictWriter(buf,

        config['csv_columns'],

        extrasaction='ignore',

        delimiter=config.get("csv_delimiter_char", ';'),

        escapechar=config.get("csv_escape_char", '\\'),

        quotechar=config.get("csv_quote_char", '"'),

        quoting=getattr(csv, 'QUOTE_%s' % quoting))

    csv_writer.writerow(dict(zip(config['csv_columns'], config['csv_columns'])))

    for item in page.items:

        values = {}

        for plugin_name, plugin_instance in config['columns_plugins']:

            if plugins_data[plugin_name]:

                values[plugin_name] = repr(plugins_data[plugin_name])

            else:

                for data_key, data_value in \

                    plugin_instance.get_data(item).iteritems():

                    # Pour les valeurs en unicode, on convertit en UTF-8.

                    if isinstance(data_value, unicode):

                        values[data_key] = data_value.encode('utf-8')

                    # Pour le reste, on suppose qu'on peut en obtenir une

                    # représentation adéquate dont l'encodage ne posera pas

                    # de problème.

                    else:

                        values[data_key] = data_value

        csv_writer.writerow(values)

    return buf.getvalue()