【Qt】将数据导出到csv文件

实际项目中有存储大量数据到表格的需求。最开始我使用的是Qtxlsx,将数据导出为excel表格。Qtxlsx插件的好处在于跨平台,相比于Qt自带的excel表格操作类,Qtxlsx不要求宿主机上安装有office,同时其读写速度远超过Qt原生excel操作类。
但Qtxlsx也有缺点,一是需要编译、移植库等一系列操作后才可以使用;二是存储大量数据(几十万条数量级)时,其对内存的占用会迅速爬升,十万条100字段的数据,轻轻松松占到2G内存。这如果移植到嵌入式平台将是致命的缺点。
幸好有csv这般轻量级、易操作的表格存储手段,具体介绍见伟大的百度百科https://baike.baidu.com/item/CSV/10739?fr=aladdin,其精髓就是以逗号、引号分隔列,以换行符分隔行,从而形成表格。下面在上两篇博文例程【传送门】的基础上再加入一个功能:导出数据库内容到csv文件。
最新的【读取该表内所有记录】按钮的作用是读取指定表的数据并显示在tableview中,那么就在这里加入保存为csv文件的功能吧。
在之前的基础上包含头文件

#include 
#include 
#include 

在上两篇的基础上,在void MainWindow::on_pushButton_4_clicked()中增加相应代码。

//显示所选表所有内容
void MainWindow::on_pushButton_4_clicked()
{
    QString cmd;
    QSqlQueryModel *model;
    model = new QSqlQueryModel();
    cmd += "select * from ";
    cmd += ui->lineEdit->text();
    model->setQuery(cmd);
    ui->tableView->setModel(model);

    //以下是保存csv
    int row,col,i,j;
    QFile file;
    QString fileName;
    QModelIndex index;
    row = model->rowCount();            //获取当前tableview行列数
    col = model->columnCount();
    fileName = QFileDialog::getSaveFileName(this,
                                            tr("保存表格"),
                                            QDir::currentPath() + "/" +\
                                            ui->lineEdit->text() + ".csv",
                                            tr("csv File(*.csv)"));
    if(fileName.isNull())
    {
        return;
    }

    file.setFileName(fileName);
    if(!file.open(QIODevice::WriteOnly | QIODevice::Text))
    {
        qDebug()<<"open file fail";
        return;
    }
    QTextStream out(&file);             //创建一个文本流,向保存文件中写入文本

    //写表头
    for(i = 0;i < col; i++)
    {
        out<headerData(i,Qt::Horizontal,Qt::DisplayRole).toString()<<",";
        if(i == (col - 1))
        {
            out<<"\n";
        }
    }
    //写数据
    for(i = 0;i < row; i++)
    {
        for(j = 0;j < col; j++)
        {
            index = model->index(i,j);
            out<data(index).toString()<<",";
            if(j == (col - 1))
            {
                out<<"\n";
            }
        }
    }
    file.close();

}

测试中,选择好要保存的文件后,程序能成功保存为csv文件,用office打开如下,于预期相符。
【Qt】将数据导出到csv文件_第1张图片

你可能感兴趣的:(Qt,数据库,qt)