+
−# 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([[], []])