Qt数据库应用之实现数据分组导出

#include "frmdataout3.h"
#include "ui_frmdataout3.h"
#include "frmmain.h"
#include "quihelper.h"
#include "dataxls.h"
#include "dataprint.h"

frmDataOut3::frmDataOut3(QWidget *parent) : QWidget(parent), ui(new Ui::frmDataOut3)
{
    ui->setupUi(this);
    this->initForm();
}

frmDataOut3::~frmDataOut3()
{
    delete ui;
}

void frmDataOut3::showEvent(QShowEvent *)
{
    static bool isShow = false;
    if (!isShow) {
        isShow = true;
        QMetaObject::invokeMethod(this, "on_btnLoad_clicked");
    }
}

void frmDataOut3::setInfo(int type, int count)
{
    QString str1 = "生成数据";
    if (type == 1) {
        str1 = "导出数据";
    } else if (type == 2) {
        str1 = "打印数据";
    }

    ui->labInfo1->setText(str1);
    ui->labInfo2->setText(QString("共 %1 条").arg(count));
    QString msec = QString::number((float)time.elapsed() / 1000, 'f', 3);
    ui->labInfo3->setText(QString("用时 %1 秒").arg(msec));
}

void frmDataOut3::getContent(int maxCount, QStringList &content, QStringList &subTitle1, QStringList &subTitle2)
{
    content.clear();
    subTitle1.clear();
    subTitle2.clear();
    maxCount = maxCount >= rowCount ? rowCount : maxCount;

    QString sql = QString("select * from MsgInfo order by DeviceID asc,MsgTime asc limit %1").arg(maxCount);
    QSqlQuery query;
    if (!query.exec(sql)) {
        return;
    }

    int count = 0;
    int logID = 0;
    QString tempDeviceID = "0";
    QString temp = "";
    QString startTime = "";
    QString endTime = "";

    while (query.next()) {
        count++;
        logID++;

        QString deviceID = query.value(0).toString();
        QString deviceName = query.value(1).toString();
        QString deviceTel = query.value(2).toString();
        QString deviceWeight = query.value(3).toString();
        QString deviceTemp = query.value(4).toString();
        QString devicePressure = query.value(5).toString();
        QString msgType = query.value(6).toString();
        QString msgTime = query.value(7).toString();
        QString msgContent = query.value(8).toString();

        //如果是一个全新的设备,则添加子标题,添加上一个设备的数据集合后清空集合数据.
        if (deviceID != tempDeviceID) {
            subTitle1 << QString("设备编号 : %1    设备名称 : %2    设备号码 : %3").arg(deviceID).arg(deviceName).arg(deviceTel);
            //当前设备是一个新的设备,而且当前数据集合不为空,说明上一个设备数据集合已经准备完毕
            //此时添加上一个设备的子标题,因为此时才能计算得到结束时间
            if (!temp.isEmpty()) {
                content << temp.mid(0, temp.length() - 1);
                logID--;

                subTitle2 << QString("开始时间 : %1    结束时间 : %2    记录总数 : %3").arg(startTime).arg(endTime).arg(logID);
                temp = "";
                logID = 1;
            }

            tempDeviceID = deviceID;
            startTime = msgTime;
        }

        endTime =  msgTime;
        temp += QString("%1;%2;%3;%4;%5;%6;%7|").arg(logID).arg(deviceWeight).arg(deviceTemp).arg(devicePressure).arg(msgType).arg(msgTime).arg(msgContent);
    }

    //添加最后一个设备的数据
    if (!temp.isEmpty()) {
        content << temp.mid(0, temp.length() - 1);
        subTitle2 << QString("开始时间 : %1    结束时间 : %2    记录总数 : %3").arg(startTime).arg(endTime).arg(logID);
    }
}

DataContent frmDataOut3::getDataContent(int maxCount)
{
    DataContent dataContent;
    dataContent.title = "所有短信记录";
    dataContent.subTitle = QString("%1 导出短信记录").arg(DATETIME);

    QList<QString> columnNames;
    columnNames << "序号" << "重量" << "温度" << "压力" << "类型" << "接收时间" << "短信内容";
    QList<int> columnWidths;
    columnWidths << 70 << 70 << 70 << 70 << 100 << 180 << 260;
    dataContent.columnNames = columnNames;
    dataContent.columnWidths = columnWidths;

    QStringList content, subTitle1, subTitle2;
    getContent(maxCount, content, subTitle1, subTitle2);
    dataContent.content = content;
    dataContent.subTitle1 = subTitle1;
    dataContent.subTitle2 = subTitle2;

    //默认对齐方式
    dataContent.defaultAlignment = 1;
    //边框粗细
    dataContent.borderWidth = 1;
    //导出到xls有样式比如边框需要开启 cellStyle = true
    //dataContent.cellStyle = true;
    dataContent.randomColor = ui->ckRandomColor->isChecked();

    //下面表示第4列 (对应表中的 类型 字段) 值 包含 重量 关键字 则突出显示
    dataContent.checkColumn = ui->ckCheckColumn->isChecked() ? 4 : -1;
    dataContent.checkType = "contains";
    dataContent.checkValue = "重量";

    dataContent.stretchLast = ui->ckStretchLast->isChecked();
    dataContent.landscape = false;
    return dataContent;
}

void frmDataOut3::initForm()
{
    ui->frame->setFixedWidth(AppConfig::RightWidth);

    columnNames << "设备编号" << "设备名称" << "设备号码" << "重量" << "温度" << "压力" << "类型" << "接收时间" << "短信内容";
    columnWidths << 70 << 150 << 120 << 70 << 70 << 70 << 100 << 180 << 180;

    rowCount = frmMain::getCount();
    columnCount = columnNames.count();

    model = new QSqlTableModel;
    QUIHelper::initTableView(ui->tableView);
}

void frmDataOut3::on_btnLoad_clicked()
{
    time.restart();

    model->setTable("MsgInfo");
    model->setSort(0, Qt::AscendingOrder);
    model->select();
    ui->tableView->setModel(model);

    for (int i = 0; i < columnCount; ++i) {
        model->setHeaderData(i, Qt::Horizontal, columnNames.at(i));
        ui->tableView->setColumnWidth(i, columnWidths.at(i));
    }

    setInfo(0, rowCount);
}

void frmDataOut3::on_btnXls_clicked()
{
    time.restart();

    DataContent dataContent = getDataContent(AppConfig::CountXls);
    dataContent.fileName = QUIHelper::appPath() + "/db/dataout3.xls";
    dataContent.sheetName = "短信信息";
    DataXls::saveXls(dataContent);

    setInfo(1, qMin(rowCount, AppConfig::CountXls));
    QUIHelper::openFile(dataContent.fileName, "导出短信信息");
}

void frmDataOut3::on_btnPdf_clicked()
{
    time.restart();

    DataContent dataContent = getDataContent(AppConfig::CountPdf);
    dataContent.fileName = QUIHelper::appPath() + "/db/dataout3.pdf";
    DataPrint::savePdf(dataContent);

    setInfo(1, qMin(rowCount, AppConfig::CountPdf));
    QUIHelper::openFile(dataContent.fileName, "导出短信信息");
}

void frmDataOut3::on_btnPrint_clicked()
{
    DataContent dataContent = getDataContent(AppConfig::CountPrint);
    DataPrint::print(dataContent);
}