Source code for compare_section_csv_table

# -*- coding: utf-8 -*-

__package__='odmltables'

import os.path
import csv

from .compare_section_table import CompareSectionTable


class CompareSectionCsvTable(CompareSectionTable):
    """
    class to write a CompareSectionTable to a csv-file
    """

    def __init__(self):

        CompareSectionTable.__init__(self)

    def write2file(self, save_to):
        """
        saves the table as a csv-file
        """

        # add csv extension if none present
        if os.path.splitext(save_to)[-1] == '':
            save_to += '.csv'

        properties, sections, table = self._build_table()

        with open(save_to, "w") as csvfile:
            csvwriter = csv.writer(csvfile, dialect='excel',
                                   quoting=csv.QUOTE_NONNUMERIC)

            if self.switch:
                csvwriter.writerow([''] + properties)

                for i, line in enumerate(table):
                    csvwriter.writerow([sections[i]] + line)

            else:
                csvwriter.writerow([''] + sections)
                if table:
                    for i in list(range(len(table[0]))):
                        cells = [properties[i]] + [table[j][i] for j in list(range(len(table)))]
                        csvwriter.writerow(cells)