HistoryModel.h

// Author: Kang Lin <kl222@126.com>
 
#pragma once
 
#include <QDate>
#include <QAbstractTableModel>
#include "HistoryDatabase.h"
#include "ParameterWebBrowser.h"
 
class CHistoryModel : public QAbstractTableModel
{
    Q_OBJECT
 
public:
    explicit CHistoryModel(CParameterWebBrowser* pPara, QObject *parent = nullptr);
    ~CHistoryModel();
 
    enum Column {
        ColumnVisitTime = 0,
        ColumnTitle,
        ColumnUrl,
        ColumnCount,
    };
    Q_ENUM(Column)
 
    // Header
    QVariant headerData(int section,
                        Qt::Orientation orientation,
                        int role = Qt::DisplayRole) const override;
 
    // Basic functionality
    int rowCount(const QModelIndex &parent = QModelIndex()) const override;
    int columnCount(const QModelIndex &parent = QModelIndex()) const override;
    QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
    bool removeRows(int row, int count, const QModelIndex &parent = QModelIndex()) override;
 
    // 自定义方法
    void refresh();
    void refresh(const QDate &start, const QDate &end);
    HistoryItem getItem(const QModelIndex &index) const;
    bool removeDomainItems(const QString& szDomain);
    bool removeItems(const QString& url);
 
    bool importFromCSV(const QString& filename);
    bool exportToCSV(const QString &filename);
    bool importFromJson(const QString& filename);
    bool exportToJson(const QString& filename);
private:
    QList<HistoryItem> m_historyItems;
    CHistoryDatabase* m_pDatabase;
    CParameterWebBrowser* m_pPara;
};