# copyright 2003-2011 LOGILAB S.A. (Paris, FRANCE), all rights reserved.

# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr

#

# This file is part of CubicWeb.

#

# CubicWeb is free software: you can redistribute it and/or modify it under the

# terms of the GNU Lesser General Public License as published by the Free

# Software Foundation, either version 2.1 of the License, or (at your option)

# any later version.

#

# CubicWeb is distributed in the hope that it will be useful, but WITHOUT

# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS

# FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more

# details.

#

# You should have received a copy of the GNU Lesser General Public License along

# with CubicWeb.  If not, see <http://www.gnu.org/licenses/>.

"""csv export views"""

__docformat__ = "restructuredtext en"

from cubicweb import _

from six import PY2

from six.moves import range

from cubicweb.schema import display_name

from cubicweb.predicates import any_rset, empty_rset

from cubicweb.uilib import UnicodeCSVWriter

from cubicweb.view import EntityView, AnyRsetView

class CSVMixIn(object):

    """mixin class for CSV views"""

    templatable = False

    content_type = "text/comma-separated-values"

    binary = PY2 # python csv module is unicode aware in py3k

    csv_params = {'dialect': 'excel',

                  'quotechar': '"',

                  'delimiter': ';',

                  'lineterminator': '\n'}

    def set_request_content_type(self):

        """overriden to set a .csv filename"""

        self._cw.set_content_type(self.content_type, filename='cubicwebexport.csv')

    def csvwriter(self, **kwargs):

        params = self.csv_params.copy()

        params.update(kwargs)

        return UnicodeCSVWriter(self.w, self._cw.encoding, **params)

class CSVRsetView(CSVMixIn, AnyRsetView):

    """dumps raw result set in CSV"""

    __regid__ = 'csvexport'

    __select__ = any_rset()

    title = _('csv export')

    def call(self):

        writer = self.csvwriter()

        writer.writerow(self.columns_labels())

        rset, descr = self.cw_rset, self.cw_rset.description

        eschema = self._cw.vreg.schema.eschema

        for rowindex, row in enumerate(rset):

            csvrow = []

            for colindex, val in enumerate(row):

                etype = descr[rowindex][colindex]

                if val is not None and not eschema(etype).final:

                    # csvrow.append(val) # val is eid in that case

                    content = self._cw.view('textincontext', rset,

                                            row=rowindex, col=colindex)

                else:

                    content = self._cw.view('final', rset,

                                            format='text/plain',

                                            row=rowindex, col=colindex)

                csvrow.append(content)

            writer.writerow(csvrow)

class CSVEntityView(CSVMixIn, EntityView):

    """dumps rset's entities (with full set of attributes) in CSV

    the generated CSV file will have a table per entity type found in the

    resultset. ('table' here only means empty lines separation between table

    contents)

    """

    __regid__ = 'ecsvexport'

    __select__ = EntityView.__select__ | empty_rset()

    title = _('csv export (entities)')

    def call(self):

        req = self._cw

        rows_by_type = {}

        writer = self.csvwriter()

        rowdef_by_type = {}

        for index in range(len(self.cw_rset)):

            entity = self.cw_rset.complete_entity(index)

            if entity.e_schema not in rows_by_type:

                rowdef_by_type[entity.e_schema] = [rs for rs, at in entity.e_schema.attribute_definitions()

                                                   if at != 'Bytes']

                rows_by_type[entity.e_schema] = [[display_name(req, rschema.type)

                                                  for rschema in rowdef_by_type[entity.e_schema]]]

            rows = rows_by_type[entity.e_schema]

            rows.append([entity.printable_value(rs.type, format='text/plain')

                         for rs in rowdef_by_type[entity.e_schema]])

        for rows in rows_by_type.values():

            writer.writerows(rows)

            # use two empty lines as separator

            writer.writerows([[], []])