102 lines
3.3 KiB
C++
102 lines
3.3 KiB
C++
|
#include "tableEditModel.h"
|
|||
|
|
|
|||
|
|
TableEditModel::TableEditModel(QObject *parent, const QSqlDatabase &db, const QStringList &visibleColumns)
|
|||
|
|
: QSqlTableModel(parent, db), m_visibleColumns(visibleColumns)
|
|||
|
|
{
|
|||
|
|
setEditStrategy(QSqlTableModel::OnManualSubmit); //对编辑数据手动进行提交
|
|||
|
|
|
|||
|
|
m_pageSize = 100;
|
|||
|
|
m_currentPage = 1;
|
|||
|
|
m_totalRecords = 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
TableEditModel::~TableEditModel()
|
|||
|
|
{}
|
|||
|
|
|
|||
|
|
//重写生成查询的方法
|
|||
|
|
QString TableEditModel::selectStatement() const
|
|||
|
|
{
|
|||
|
|
QString originalQuery = QSqlTableModel::selectStatement();
|
|||
|
|
QString newQuery;
|
|||
|
|
if(m_visibleColumns.isEmpty())
|
|||
|
|
newQuery = originalQuery;
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
int fromInex = originalQuery.indexOf(" FROM ");
|
|||
|
|
QString selectPart = "SELECT " + m_visibleColumns.join(", ");
|
|||
|
|
newQuery = selectPart + originalQuery + selectPart.mid(fromInex);
|
|||
|
|
}
|
|||
|
|
newQuery = newQuery + QString(" LIMIT %1 OFFSET %2").arg(m_pageSize, (m_currentPage - 1) * m_pageSize);
|
|||
|
|
|
|||
|
|
return newQuery;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
QVariant TableEditModel::data(const QModelIndex& index, int role) const
|
|||
|
|
{
|
|||
|
|
if(!index.isValid())
|
|||
|
|
return QVariant();
|
|||
|
|
|
|||
|
|
if(index.column() == 0) //第一列显示行号
|
|||
|
|
{
|
|||
|
|
int golbalRow = (m_currentPage - 1) * m_pageSize + index.row();
|
|||
|
|
switch (role)
|
|||
|
|
{
|
|||
|
|
case Qt::DisplayRole:
|
|||
|
|
{
|
|||
|
|
QString displayRow = QString::number(golbalRow + 1);
|
|||
|
|
if(m_editCache.keys().contains(index.row()))
|
|||
|
|
displayRow = "*" + displayRow;
|
|||
|
|
return displayRow;
|
|||
|
|
}
|
|||
|
|
case Qt::DecorationRole:
|
|||
|
|
return QVariant();
|
|||
|
|
case Qt::TextAlignmentRole: //行号居中展示
|
|||
|
|
return Qt::AlignCenter;
|
|||
|
|
default:
|
|||
|
|
return QVariant();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else if(role == Qt::UserRole + 100) //先以100来标识数据是否被编辑,在相关代理Delegate类实现中同步进行处理（文字加粗等效果）
|
|||
|
|
return isDirty(index);
|
|||
|
|
else
|
|||
|
|
return QSqlTableModel::data(createIndex(index.row(), index.column() - 1), role); //因为第一列显示行号，所以其它数据列要错位展示
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
bool TableEditModel::setData(const QModelIndex &index, const QVariant &value, int role)
|
|||
|
|
{
|
|||
|
|
if (!index.isValid() || index.column() == 0 || role != Qt::EditRole) //忽略行号列
|
|||
|
|
return false;
|
|||
|
|
|
|||
|
|
// if(index.column() == 0) //忽略行号列
|
|||
|
|
// return QSqlTableModel::setData(createIndex(index.row(), index.column()-1), value, role);
|
|||
|
|
|
|||
|
|
int row = index.row();
|
|||
|
|
if (row < 0 || row >= rowCount())
|
|||
|
|
return false;
|
|||
|
|
|
|||
|
|
//记录修改到缓存
|
|||
|
|
if(!m_editCache.contains(row))
|
|||
|
|
{
|
|||
|
|
EditRecord record;
|
|||
|
|
record.original = QSqlTableModel::record(row);
|
|||
|
|
m_editCache[row] = record;
|
|||
|
|
}
|
|||
|
|
m_editCache[row].modified.setValue(index.column() - 1, value);
|
|||
|
|
|
|||
|
|
emit dataChanged(index, index, {role});
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int TableEditModel::columnCount(const QModelIndex &parent) const
|
|||
|
|
{
|
|||
|
|
return QSqlTableModel::columnCount(parent) + 1; // 原始列数 + 行号列
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
Qt::ItemFlags TableEditModel::flags(const QModelIndex &index) const
|
|||
|
|
{
|
|||
|
|
if(index.column() == 0) //行号列不可编辑
|
|||
|
|
return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
|
|||
|
|
else return QSqlTableModel::flags(this->index(index.row(), index.column() - 1));
|
|||
|
|
}
|
|||
|
|
|