mirror of
https://github.com/keepassxreboot/keepassxc.git
synced 2025-04-05 21:47:38 +03:00
125 lines
3.6 KiB
C++
125 lines
3.6 KiB
C++
/*
|
|
* Copyright (C) 2016 Enrico Mariotti <enricomariotti@yahoo.it>
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 2 or (at your option)
|
|
* version 3 of the License.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "CsvParserModel.h"
|
|
|
|
CsvParserModel::CsvParserModel(QObject *parent)
|
|
: QAbstractTableModel(parent)
|
|
, m_skipped(0)
|
|
{}
|
|
|
|
CsvParserModel::~CsvParserModel()
|
|
{}
|
|
|
|
void CsvParserModel::setFilename(const QString& filename) {
|
|
m_filename = filename;
|
|
}
|
|
|
|
QString CsvParserModel::getFileInfo(){
|
|
QString a(QString::number(getFileSize()).append(tr(" byte, ")));
|
|
a.append(QString::number(getCsvRows())).append(tr(" rows, "));
|
|
a.append(QString::number(qMax(0, getCsvCols()-1))).append(tr(" columns"));
|
|
return a;
|
|
}
|
|
|
|
bool CsvParserModel::parse() {
|
|
bool r;
|
|
beginResetModel();
|
|
m_columnMap.clear();
|
|
if (CsvParser::isFileLoaded()) {
|
|
r = CsvParser::reparse();
|
|
} else {
|
|
QFile csv(m_filename);
|
|
r = CsvParser::parse(&csv);
|
|
}
|
|
for (int i = 0; i < columnCount(); ++i)
|
|
m_columnMap.insert(i,0);
|
|
addEmptyColumn();
|
|
endResetModel();
|
|
return r;
|
|
}
|
|
|
|
void CsvParserModel::addEmptyColumn() {
|
|
for (int i = 0; i < m_table.size(); ++i) {
|
|
CsvRow r = m_table.at(i);
|
|
r.prepend(QString(""));
|
|
m_table.replace(i, r);
|
|
}
|
|
}
|
|
|
|
void CsvParserModel::mapColumns(int csvColumn, int dbColumn) {
|
|
if ((csvColumn < 0) || (dbColumn < 0))
|
|
return;
|
|
beginResetModel();
|
|
if (csvColumn >= getCsvCols())
|
|
m_columnMap[dbColumn] = 0; //map to the empty column
|
|
else
|
|
m_columnMap[dbColumn] = csvColumn;
|
|
endResetModel();
|
|
}
|
|
|
|
void CsvParserModel::setSkippedRows(int skipped) {
|
|
m_skipped = skipped;
|
|
QModelIndex topLeft = createIndex(skipped,0);
|
|
QModelIndex bottomRight = createIndex(m_skipped+rowCount(), columnCount());
|
|
emit dataChanged(topLeft, bottomRight);
|
|
emit layoutChanged();
|
|
}
|
|
|
|
void CsvParserModel::setHeaderLabels(QStringList l) {
|
|
m_columnHeader = l;
|
|
}
|
|
|
|
int CsvParserModel::rowCount(const QModelIndex &parent) const {
|
|
if (parent.isValid())
|
|
return 0;
|
|
return getCsvRows();
|
|
}
|
|
|
|
int CsvParserModel::columnCount(const QModelIndex &parent) const {
|
|
if (parent.isValid())
|
|
return 0;
|
|
return m_columnHeader.size();
|
|
}
|
|
|
|
QVariant CsvParserModel::data(const QModelIndex &index, int role) const {
|
|
if ((index.column() >= m_columnHeader.size())
|
|
|| (index.row()+m_skipped >= rowCount())
|
|
|| !index.isValid()) {
|
|
return QVariant();
|
|
}
|
|
if (role == Qt::DisplayRole)
|
|
return m_table.at(index.row()+m_skipped).at(m_columnMap[index.column()]);
|
|
return QVariant();
|
|
}
|
|
|
|
QVariant CsvParserModel::headerData(int section, Qt::Orientation orientation, int role) const {
|
|
if (role == Qt::DisplayRole) {
|
|
if (orientation == Qt::Horizontal) {
|
|
if ((section < 0) || (section >= m_columnHeader.size()))
|
|
return QVariant();
|
|
return m_columnHeader.at(section);
|
|
} else if (orientation == Qt::Vertical) {
|
|
if (section+m_skipped >= rowCount())
|
|
return QVariant();
|
|
return QString::number(section+1);
|
|
}
|
|
}
|
|
return QVariant();
|
|
}
|
|
|
|
|