1 回答
TA贡献1847条经验 获得超7个赞
首先让每个函数执行最小的任务,以便调试很容易,例如在你的情况下,没有必要在那里使用 QFIleDialog 来选择文件,而只接收路径,然后在另一个函数中进行选择。另一方面,所使用的信息存储在模型中而不是视图中,因此可以通过保存模型而不是 QTableWidget 来概括它,因此您也可以将它用于任何类型的 QTableView。
def write_model_to_csv(self, model, filename):
with open(filename, "w") as stream:
writer = csv.writer(stream, dialect="excel", delimiter=",")
headers = []
for column in range(model.columnCount()):
text = model.headerData(column, Qt.Horizontal)
if text:
headers.append(text)
else:
headers.append("Column {}".format(column))
writer.writerow(headers)
for row in range(model.rowCount()):
rowdata = []
for column in range(model.columnCount()):
text = model.index(row, column).data()
if text:
rowdata.append(text)
else:
rowdata.append("")
writer.writerow(rowdata)
path, _ = QFileDialog.getSaveFileName(
self, "Save File", QDir.homePath() + "/export.csv", "CSV Files(*.csv *.txt)"
)
if path:
self.write_model_to_csv(self.tableWidget_show.model(), path)
添加回答
举报