Missing column headers in saved file

Question

Is it possible I get assistance here? My function saves a file with a space after every row, also the column headers are not saved. Here is the function.

def download_results(self):
    try:
        path = QFileDialog.getSaveFileName(MainWindow, 'Save results', os.getenv('HOME'), 'CSV(*.csv)')
        if path[0] != '':
            with open(path[0], 'w') as csv_file:
                writer = csv.writer (csv_file, dialect = 'excel', delimiter = ',')
                for row in range(self.analysis_table.rowCount()):
                    row_data = []
                    for column in range(self.analysis_table.columnCount()):
                        item = self.analysis_table.item(row, column)
                        if item is not None:
                            row_data.append(item.text())
                        else:
                            row_data.append('')
                    writer.writerow(row_data)
        QMessageBox.information(MainWindow,"Success","Succeeded")
    except:
        QMessageBox.warning(MainWindow,"Failed","Operation failed.")

"My function saves a file with a space after every row", See https://stackoverflow.com/q/3348460/235698. — Mark Tolonen, May 06 '20 at 16:09
FYI, `dialect='excel'` is the default, and `delimiter=','` is the delimiter for the `excel` dialect, so neither are required. — Mark Tolonen, May 06 '20 at 16:19

eyllanesc · Accepted Answer · 2020-05-06T16:33:11.663

You have to extract the text from the QTableWidgetItem using horizontalHeaderItem() method:

def download_results(self):
    try:
        path, _ = QFileDialog.getSaveFileName(
            MainWindow, "Save results", os.getenv("HOME"), "CSV(*.csv)"
        )
        if path:
            with open(path, "w", newline='') as csv_file:
                writer = csv.writer(csv_file)
                headers = []
                for c in range(self.analysis_table.columnCount()):
                    it = self.analysis_table.horizontalHeaderItem(c)
                    if it is not None:
                        headers.append(it.text())
                    else:
                        headers.append("")
                writer.writerow(headers)
                for row in range(self.analysis_table.rowCount()):
                    row_data = []
                    for column in range(self.analysis_table.columnCount()):
                        item = self.analysis_table.item(row, column)
                        if item is not None:
                            row_data.append(item.text())
                        else:
                            row_data.append("")
                    writer.writerow(row_data)
        QMessageBox.information(MainWindow, "Success", "Succeeded")
    except:
        QMessageBox.warning(MainWindow, "Failed", "Operation failed.")

`csv.writer` doesn't have `fieldnames`. That's `DictWriter`. — Mark Tolonen, May 06 '20 at 16:12
It returns this.Traceback (most recent call last): File "D:\Python\PyQt5\Proper_1.py", line 1499, in download_results with open(path, "w", newline='') as csv_file: TypeError: expected str, bytes or os.PathLike object, not tuple — Ptar, May 06 '20 at 16:28

score 0 · Answer 2 · answered May 06 '20 at 16:52

This did it. I'm gratefull to the contributors

def download_results(self):
    try:
        path = QFileDialog.getSaveFileName(MainWindow, "Save results", os.getenv("HOME"), "CSV(*.csv)")
        if path != "":
            with open(path[0], "w", newline='') as csv_file:
                writer = csv.writer(csv_file)
                headers = []
                for c in range(self.analysis_table.columnCount()):
                    it = self.analysis_table.horizontalHeaderItem(c)
                    if it is not None:
                        headers.append(it.text())
                    else:
                        headers.append("")
                writer.writerow(headers)
                for row in range(self.analysis_table.rowCount()):
                    row_data = []
                    for column in range(self.analysis_table.columnCount()):
                        item = self.analysis_table.item(row, column)
                        if item is not None:
                            row_data.append(item.text())
                        else:
                            row_data.append("")
                    writer.writerow(row_data)
        QMessageBox.information(MainWindow, "Success", "Succeeded")
    except:
        QMessageBox.warning(MainWindow, "Failed", "Operation failed.")

Missing column headers in saved file

2 Answers2