62 lines
1.8 KiB
C++
62 lines
1.8 KiB
C++
#ifndef TABLEEDITMODEL_H
|
||
#define TABLEEDITMODEL_H
|
||
|
||
/**
|
||
* @brief 用来加载显示数据库中数据表的Model类
|
||
*
|
||
* 基本功能包括：
|
||
* 1、可以自定义显示数据表中的哪些列
|
||
* 2、可以分页显示并自定义每页展示数量
|
||
* 3、编辑的数据可以突出展示（如加粗、改色）
|
||
* 4、最前方加入一个用于展示行号的列（行号从1开始）
|
||
* 5、点击行号列可以实现选中标识（通过icon）
|
||
* 6、哪一行被编辑，改行的行号指示可以加*进行标识
|
||
*
|
||
*/
|
||
|
||
#include <QSqlTableModel>
|
||
#include <QSqlRecord>
|
||
|
||
class TableEditModel : public QSqlTableModel
|
||
{
|
||
Q_OBJECT
|
||
public:
|
||
explicit TableEditModel(QObject* parent = nullptr, const QSqlDatabase& db = QSqlDatabase(), const QStringList &visibleColumns = {});
|
||
~TableEditModel();
|
||
|
||
QVariant data(const QModelIndex& index, int role) const override;
|
||
bool setData(const QModelIndex &index, const QVariant &value, int role = Qt::EditRole) override;
|
||
int columnCount(const QModelIndex& parent = QModelIndex()) const override;
|
||
Qt::ItemFlags flags(const QModelIndex &index) const override;
|
||
|
||
//分页控制
|
||
void setPageSize(int);
|
||
int pageSize() const;
|
||
void setCurrentPage(int);
|
||
int currentPage() const;
|
||
int totalPages() const;
|
||
|
||
//展示列控制
|
||
void setVisibleColumns(const QStringList& columns);
|
||
|
||
protected:
|
||
QString selectStatement() const override; //调用select()时会使用该函数重写后返回的语句
|
||
|
||
private:
|
||
struct EditRecord //编辑记录
|
||
{
|
||
QSqlRecord original;
|
||
QSqlRecord modified;
|
||
bool isNew = false;
|
||
bool isDeleted = false;
|
||
};
|
||
|
||
int m_pageSize;
|
||
int m_currentPage;
|
||
int m_totalRecords;
|
||
QStringList m_visibleColumns;
|
||
QHash<int, EditRecord> m_editCache; //key:当前页中的row
|
||
};
|
||
|
||
#endif //TABLEEDITMODEL_H
|