422 lines
12 KiB
C++
422 lines
12 KiB
C++
#ifndef DTE_H
|
|
#define DTE_H
|
|
#include <qwidget.h>
|
|
#include <qcalendarwidget.h>
|
|
#include <qlayout.h>
|
|
#include <qpushbutton.h>
|
|
#include <qlineedit.h>
|
|
#include <uiModule/uiModule_tableView.h>
|
|
#include <sqliteController/sqlControler.h>
|
|
#include <qfile.h>
|
|
#include <qmessagebox.h>
|
|
#include <QComboBox>
|
|
#include <QDateTimeEdit>
|
|
#include <QTime>
|
|
#define cacheSqlFileName "sqlite-database-copy.db"
|
|
//#include "xlsxdocument.h"
|
|
class TimeRangeSelectorWidget : public QWidget {
|
|
|
|
public:
|
|
TimeRangeSelectorWidget(QWidget* parent = nullptr) : QWidget(parent) {
|
|
setupUi();
|
|
}
|
|
|
|
std::pair<int, int>getIntTime()
|
|
{
|
|
int time1 = startTimeEdit->time().hour() * 60 + startTimeEdit->time().minute();
|
|
int time2 = endTimeEdit->time().hour() * 60 + startTimeEdit->time().minute();
|
|
return { std::min(time1,time2),std::max(time1,time2) };
|
|
}
|
|
|
|
QTime getStartTime() const {
|
|
return startTimeEdit->time();
|
|
}
|
|
|
|
QTime getEndTime() const {
|
|
return endTimeEdit->time();
|
|
}
|
|
|
|
|
|
private:
|
|
void setupUi() {
|
|
QVBoxLayout* layout = new QVBoxLayout(this);
|
|
|
|
QLabel* startLabel = new QLabel("开始时间:", this);
|
|
startTimeEdit = new QDateTimeEdit(this);
|
|
startTimeEdit->setDisplayFormat("HH:mm");
|
|
|
|
QLabel* endLabel = new QLabel("结束时间:", this);
|
|
endTimeEdit = new QDateTimeEdit(this);
|
|
endTimeEdit->setDisplayFormat("HH:mm");
|
|
|
|
layout->addWidget(startLabel);
|
|
layout->addWidget(startTimeEdit);
|
|
layout->addWidget(endLabel);
|
|
layout->addWidget(endTimeEdit);
|
|
|
|
setLayout(layout);
|
|
}
|
|
|
|
QDateTimeEdit* startTimeEdit;
|
|
QDateTimeEdit* endTimeEdit;
|
|
};
|
|
|
|
class dumpToExcel
|
|
{
|
|
public:
|
|
QWidget* mainWidget;
|
|
sqlTableDatas globalSqlDatas;
|
|
QCalendarWidget* calendarW;
|
|
QPushButton* showMainWidgetBtn;
|
|
QPushButton* dumpToCsvBtn;
|
|
QPushButton* dumpToCsvWithDateLimitBtn;
|
|
QPushButton* openPathSelectBtn;
|
|
QLineEdit* filePathLineEdit;
|
|
TimeRangeSelectorWidget* timeSelector;
|
|
dumpToExcel()
|
|
{
|
|
showMainWidgetBtn = new QPushButton(u8"导出数据为表格");
|
|
showMainWidgetBtn->setStyleSheet("color: white;");
|
|
mainWidget = new QWidget;
|
|
QHBoxLayout* layout = new QHBoxLayout;
|
|
QVBoxLayout* VLayout = new QVBoxLayout;
|
|
mainWidget->setLayout(VLayout);
|
|
VLayout->addLayout(layout);
|
|
QWidget* dateWIdget;
|
|
{
|
|
calendarW = new QCalendarWidget();
|
|
dateWIdget = new QWidget();
|
|
QVBoxLayout* layoutTemp = new QVBoxLayout();
|
|
dateWIdget->setLayout(layoutTemp);
|
|
QLabel* labelTemp = new QLabel(u8"请选择日期");
|
|
labelTemp->setAlignment(Qt::AlignCenter);
|
|
layoutTemp->addWidget(labelTemp);
|
|
layoutTemp->addWidget(calendarW);
|
|
layout->addWidget(dateWIdget);
|
|
}
|
|
QWidget* controlPanel;
|
|
{
|
|
controlPanel = new QWidget();
|
|
|
|
QVBoxLayout* layoutTemp = new QVBoxLayout();
|
|
controlPanel->setLayout(layoutTemp);
|
|
|
|
QLabel* title = new QLabel(u8"导出选项");
|
|
title->setAlignment(Qt::AlignCenter);
|
|
|
|
QLabel* space = new QLabel();
|
|
QLabel* selectFolderLabel = new QLabel(u8"选择导出路径");
|
|
|
|
QHBoxLayout* layoutTemp2 = new QHBoxLayout();
|
|
openPathSelectBtn = new QPushButton("...");
|
|
openPathSelectBtn->setFixedWidth(30);
|
|
|
|
timeSelector = new TimeRangeSelectorWidget();
|
|
|
|
layoutTemp2->addWidget(openPathSelectBtn);
|
|
filePathLineEdit = new QLineEdit("D://");
|
|
layoutTemp2->addWidget(filePathLineEdit);
|
|
|
|
dumpToCsvWithDateLimitBtn = new QPushButton(u8"导出指定时间段数据为csv表格");
|
|
dumpToCsvBtn = new QPushButton(u8"导出数据为CSV表格");
|
|
|
|
layoutTemp->addWidget(title);
|
|
//layoutTemp->addWidget(data);
|
|
layoutTemp->addWidget(space);
|
|
layoutTemp->addWidget(selectFolderLabel);
|
|
layoutTemp->addLayout(layoutTemp2);
|
|
layoutTemp->addWidget(dumpToCsvBtn);
|
|
layoutTemp->addWidget(timeSelector);
|
|
layoutTemp->addWidget(dumpToCsvWithDateLimitBtn);
|
|
layoutTemp->setAlignment(Qt::AlignTop);
|
|
}
|
|
layout->addWidget(controlPanel);
|
|
tV = new tableView();
|
|
VLayout->addWidget(tV->tableWidget);
|
|
QDate selectedDate = calendarW->selectedDate();
|
|
switchDate(selectedDate.toString("yyyy-MM-dd").toStdString(), "D://");
|
|
#pragma region
|
|
QObject::connect(dumpToCsvWithDateLimitBtn, &QPushButton::clicked, [=]() {
|
|
QDate selectedDate = calendarW->selectedDate();
|
|
std::pair<int, int>times = timeSelector->getIntTime();
|
|
saveSqlTableDataToCSV(globalSqlDatas, filePathLineEdit->text() + selectedDate.toString("yyyy-MM-dd") + ".csv",times.first,times.second);
|
|
});
|
|
QObject::connect(openPathSelectBtn, &QPushButton::clicked, [=]() {
|
|
filePathLineEdit->setText(fileManager::openFolderDialog()+"/");
|
|
});
|
|
QObject::connect(dumpToCsvBtn, &QPushButton::clicked, [=]() {
|
|
QDate selectedDate = calendarW->selectedDate();
|
|
saveSqlTableDataToCSV(globalSqlDatas, filePathLineEdit->text() + selectedDate.toString("yyyy-MM-dd") + ".csv");
|
|
});
|
|
QObject::connect(showMainWidgetBtn, &QPushButton::clicked, [=](){
|
|
mainWidget->show();
|
|
mainWidget->raise();
|
|
mainWidget->isActiveWindow();
|
|
});
|
|
QObject::connect(calendarW, &QCalendarWidget::selectionChanged, [=]() {
|
|
QDate selectedDate = calendarW->selectedDate();
|
|
switchDate(selectedDate.toString("yyyy-MM-dd").toStdString(),"D://");
|
|
});
|
|
#pragma endregion 信号连接
|
|
}
|
|
void switchDate(std::string dateStr,std::string excelPath)
|
|
{
|
|
sqlController _sqlController;
|
|
QString sqlCopiedFilePath;
|
|
#pragma region
|
|
QString path(sqlDefaultDatabasePath);
|
|
QFile sqlFile(path);
|
|
if (sqlFile.open(QIODevice::ReadOnly)) {
|
|
QFileInfo fileInfo(sqlFile);
|
|
sqlCopiedFilePath = fileInfo.absolutePath() + cacheSqlFileName;
|
|
QFile destinationFile(sqlCopiedFilePath);
|
|
if (destinationFile.open(QIODevice::WriteOnly)) {
|
|
QByteArray data = sqlFile.readAll();
|
|
destinationFile.write(data);
|
|
sqlFile.close();
|
|
destinationFile.close();
|
|
qDebug() << "文件复制成功";
|
|
}
|
|
else {
|
|
qDebug() << "无法打开目标文件";
|
|
return;
|
|
}
|
|
}
|
|
else {
|
|
qDebug() << "无法打开源文件";
|
|
return;
|
|
}
|
|
#pragma endregion 复制文件
|
|
#pragma region
|
|
_sqlController.initPara(sqlCopiedFilePath.toStdString(), false);
|
|
//globalSqlDatas = _sqlController.selectAllData(dateStr);
|
|
_sqlController.dbPtr->Select(dateStr, {}, "", "", globalSqlDatas);
|
|
#pragma endregion sqlite读数据
|
|
#pragma region
|
|
if (globalSqlDatas.rowDatas.size() == 0)
|
|
{
|
|
tV->clearAllData();
|
|
tV->Init({u8"搜索结果"});
|
|
std::vector<QString> strVec({ "找不到数据" });
|
|
tV->addRowDataValue(strVec);
|
|
return;
|
|
}
|
|
#pragma endregion 检索不到数据的对策
|
|
#pragma region
|
|
tV->clearAllData();
|
|
std::vector<char>avoidVec;
|
|
std::vector<QString>dataNames;
|
|
//设置表头
|
|
for (int i = 0; i < globalSqlDatas.rowDatas[0].size(); i++)
|
|
{
|
|
QString dataName(QString::fromStdString(globalSqlDatas.rowDatas[0][i].key));
|
|
auto it = bridge.back.json.sqlDataTranslate.datas.find(dataName);
|
|
if (it != bridge.back.json.sqlDataTranslate.datas.end())
|
|
{
|
|
|
|
dataNames.push_back(it->second);
|
|
if (it->second.size() == 0)
|
|
avoidVec.push_back(i);
|
|
}
|
|
else
|
|
dataNames.push_back(dataName);
|
|
}
|
|
tV->Init(dataNames);
|
|
//插入数据
|
|
for (auto data : globalSqlDatas.rowDatas)
|
|
{
|
|
std::vector<QString>datas;
|
|
for (auto d : data)
|
|
datas.push_back(QString::fromStdString(d.value));
|
|
tV->addRowDataValue(datas);
|
|
}
|
|
//隐藏特定列
|
|
for(auto index:avoidVec)
|
|
tV->tableWidget->setColumnHidden(index, true);
|
|
|
|
#pragma endregion 将数据插入tableWidget
|
|
remove(sqlCopiedFilePath.toLocal8Bit());
|
|
}
|
|
int StringTimeToInt(const std::string input)
|
|
{
|
|
size_t firstColonPos = input.find(':');
|
|
int timeReturn = 0;
|
|
if (firstColonPos != std::string::npos) {
|
|
std::string time1 = input.substr(0, firstColonPos);
|
|
timeReturn += std::stoi(time1) * 60;
|
|
size_t secondColonPos = input.find(':', firstColonPos + 1);
|
|
if (secondColonPos != std::string::npos) {
|
|
std::string time2 = input.substr(firstColonPos + 1, secondColonPos - firstColonPos - 1);
|
|
timeReturn += std::stoi(time2);
|
|
}
|
|
}
|
|
return timeReturn;
|
|
}
|
|
void saveSqlTableDataToCSV(const sqlTableDatas& tableData, const QString& fileName, int time1 = 0, int time2 = 0)
|
|
{
|
|
#pragma region
|
|
if (tableData.rowDatas.size() == 0)
|
|
{
|
|
QMessageBox::critical(nullptr, "Error", u8"检索不到数据！");
|
|
return;
|
|
}
|
|
#pragma endregion 没有数据的对策
|
|
QFile csvFile(fileName);
|
|
#pragma region
|
|
if (time1 != 0 || time2 != 0)
|
|
{
|
|
int updateTimeIndex = -1;
|
|
for (int i = 0; i < tableData.rowDatas[0].size(); i++)
|
|
{
|
|
if (tableData.rowDatas[0][i].key == "updateTime")
|
|
updateTimeIndex = i;
|
|
}
|
|
if (!csvFile.exists()) {
|
|
// 如果文件不存在，创建文件
|
|
if (csvFile.open(QIODevice::WriteOnly | QIODevice::Text)) {
|
|
QTextStream stream(&csvFile);
|
|
//顶部标题
|
|
for (auto data : tableData.rowDatas[0])
|
|
{
|
|
QString title;
|
|
auto it = bridge.back.json.sqlDataTranslate.datas.find(QString::fromStdString(data.key));
|
|
if (it != bridge.back.json.sqlDataTranslate.datas.end())
|
|
title = it->second;
|
|
else
|
|
title = it->first;
|
|
stream << title << ",";
|
|
}
|
|
stream << "\n";
|
|
for (const auto& rowDatas : tableData.rowDatas)
|
|
{
|
|
int thisTime = StringTimeToInt(rowDatas[updateTimeIndex].value);
|
|
if (thisTime < time1 || thisTime > time2)
|
|
{
|
|
continue;
|
|
}
|
|
for (const auto& data : rowDatas)
|
|
stream << QString::fromStdString(data.value) << ",";
|
|
stream << "\n";
|
|
}
|
|
qDebug() << "CSV文件保存成功";
|
|
}
|
|
else {
|
|
qDebug() << "无法创建CSV文件";
|
|
QMessageBox::critical(nullptr, "Error", u8"没有权限或空间不足！");
|
|
}
|
|
}
|
|
else {
|
|
// 如果文件已存在，追加内容
|
|
if (csvFile.open(QIODevice::Append | QIODevice::Text)) {
|
|
QTextStream stream(&csvFile);
|
|
for (const auto& rowDatas : tableData.rowDatas)
|
|
{
|
|
int thisTime = StringTimeToInt(rowDatas[updateTimeIndex].value);
|
|
if (thisTime < time1 || thisTime > time2)
|
|
{
|
|
continue;
|
|
}
|
|
for (const auto& data : rowDatas)
|
|
stream << QString::fromStdString(data.value) << ",";
|
|
stream << "\n";
|
|
}
|
|
#pragma region
|
|
std::string command = "explorer /select, \"" + fileName.toStdString() + "\"";
|
|
size_t pos = 0;
|
|
while ((pos = command.find("//", pos)) != std::string::npos) {
|
|
command.replace(pos, 2, "\\");
|
|
}
|
|
system(command.c_str());
|
|
#pragma endregion 打开文件夹
|
|
QMessageBox::information(nullptr, "Success", u8"成功!");
|
|
qDebug() << "CSV文件保存成功";
|
|
return;
|
|
}
|
|
else {
|
|
qDebug() << "无法打开CSV文件";
|
|
#pragma region
|
|
std::string command = "explorer /select, \"" + fileName.toStdString() + "\"";
|
|
size_t pos = 0;
|
|
while ((pos = command.find("//", pos)) != std::string::npos) {
|
|
command.replace(pos, 2, "\\");
|
|
}
|
|
system(command.c_str());
|
|
#pragma endregion 打开文件夹
|
|
QMessageBox::critical(nullptr, "Error", u8"文件已存在且无法被写入！");
|
|
}
|
|
}
|
|
}
|
|
#pragma endregion 需要判断时间的循环
|
|
#pragma region
|
|
if (!csvFile.exists()) {
|
|
// 如果文件不存在，创建文件
|
|
if (csvFile.open(QIODevice::WriteOnly | QIODevice::Text)) {
|
|
QTextStream stream(&csvFile);
|
|
//顶部标题
|
|
for (auto data : tableData.rowDatas[0])
|
|
{
|
|
QString title;
|
|
auto it = bridge.back.json.sqlDataTranslate.datas.find(QString::fromStdString(data.key));
|
|
if (it != bridge.back.json.sqlDataTranslate.datas.end())
|
|
title = it->second;
|
|
else
|
|
title = it->first;
|
|
stream << title<<",";
|
|
}
|
|
stream << "\n";
|
|
for (const auto& rowDatas : tableData.rowDatas) {
|
|
for (const auto& data : rowDatas)
|
|
stream << QString::fromStdString(data.value) << ",";
|
|
stream << "\n";
|
|
}
|
|
qDebug() << "CSV文件保存成功";
|
|
}
|
|
else {
|
|
qDebug() << "无法创建CSV文件";
|
|
QMessageBox::critical(nullptr, "Error", u8"没有权限或空间不足！");
|
|
}
|
|
}
|
|
else {
|
|
// 如果文件已存在，追加内容
|
|
if (csvFile.open(QIODevice::Append | QIODevice::Text)) {
|
|
QTextStream stream(&csvFile);
|
|
for (const auto& rowDatas : tableData.rowDatas) {
|
|
for (const auto& data : rowDatas)
|
|
stream << QString::fromStdString(data.value) << ",";
|
|
stream << "\n";
|
|
}
|
|
#pragma region
|
|
std::string command = "explorer /select, \"" + fileName.toStdString() + "\"";
|
|
size_t pos = 0;
|
|
while ((pos = command.find("//", pos)) != std::string::npos) {
|
|
command.replace(pos, 2, "\\");
|
|
}
|
|
system(command.c_str());
|
|
#pragma endregion 打开文件夹
|
|
QMessageBox::information(nullptr, "Success", u8"成功!");
|
|
qDebug() << "CSV文件保存成功";
|
|
return;
|
|
}
|
|
else {
|
|
qDebug() << "无法打开CSV文件";
|
|
#pragma region
|
|
std::string command = "explorer /select, \"" + fileName.toStdString() + "\"";
|
|
size_t pos = 0;
|
|
while ((pos = command.find("//", pos)) != std::string::npos) {
|
|
command.replace(pos, 2, "\\");
|
|
}
|
|
system(command.c_str());
|
|
#pragma endregion 打开文件夹
|
|
QMessageBox::critical(nullptr, "Error", u8"文件已存在且无法被写入！");
|
|
}
|
|
}
|
|
#pragma endregion 不需要判断时间的循环
|
|
}
|
|
public slots:
|
|
|
|
private:
|
|
tableView* tV;
|
|
};
|
|
|
|
#pragma once
|
|
#endif |