1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
#include "selectabledbmodel.h"
#include "dbtree/dbtreeitem.h"
#include "dbtree/dbtreemodel.h"
SelectableDbModel::SelectableDbModel(QObject* parent) :
QSortFilterProxyModel(parent)
{
}
QVariant SelectableDbModel::data(const QModelIndex& index, int role) const
{
if (role != Qt::CheckStateRole)
return QSortFilterProxyModel::data(index, role);
DbTreeItem* item = getItemForProxyIndex(index);
if (!item)
return QSortFilterProxyModel::data(index, role);
DbTreeItem::Type type = item->getType();
if (type != DbTreeItem::Type::DB)
return QSortFilterProxyModel::data(index, role);
return checkedDatabases.contains(item->text(), Qt::CaseInsensitive) ? Qt::Checked : Qt::Unchecked;
}
bool SelectableDbModel::setData(const QModelIndex& index, const QVariant& value, int role)
{
if (role != Qt::CheckStateRole)
return QSortFilterProxyModel::setData(index, value, role);
DbTreeItem* item = getItemForProxyIndex(index);
if (!item)
return QSortFilterProxyModel::setData(index, value, role);
DbTreeItem::Type type = item->getType();
if (type != DbTreeItem::Type::DB)
return QSortFilterProxyModel::setData(index, value, role);
if (value.toBool())
checkedDatabases << item->text();
else
checkedDatabases.removeOne(item->text());
emit dataChanged(index, index, {Qt::CheckStateRole});
return true;
}
Qt::ItemFlags SelectableDbModel::flags(const QModelIndex& index) const
{
Qt::ItemFlags itemFlags = QSortFilterProxyModel::flags(index);
DbTreeItem* item = getItemForProxyIndex(index);
if (!item)
return itemFlags;
DbTreeItem::Type type = item->getType();
if (item->getDb() && item->getDb()->getVersion() == disabledVersion)
itemFlags ^= Qt::ItemIsEnabled;
else if (type == DbTreeItem::Type::DB)
itemFlags |= Qt::ItemIsUserCheckable;
return itemFlags;
}
void SelectableDbModel::setDatabases(const QStringList& databases)
{
beginResetModel();
checkedDatabases = databases;
endResetModel();
}
QStringList SelectableDbModel::getDatabases() const
{
return checkedDatabases;
}
bool SelectableDbModel::filterAcceptsRow(int srcRow, const QModelIndex& srcParent) const
{
QModelIndex idx = sourceModel()->index(srcRow, 0, srcParent);
DbTreeItem* item = dynamic_cast<DbTreeItem*>(dynamic_cast<DbTreeModel*>(sourceModel())->itemFromIndex(idx));
switch (item->getType())
{
case DbTreeItem::Type::DIR:
case DbTreeItem::Type::DB:
return true;
default:
return false;
}
return false;
}
DbTreeItem* SelectableDbModel::getItemForProxyIndex(const QModelIndex& index) const
{
QModelIndex srcIdx = mapToSource(index);
DbTreeItem* item = dynamic_cast<DbTreeItem*>(dynamic_cast<DbTreeModel*>(sourceModel())->itemFromIndex(srcIdx));
return item;
}
int SelectableDbModel::getDisabledVersion() const
{
return disabledVersion;
}
void SelectableDbModel::setDisabledVersion(int value)
{
disabledVersion = value;
}
|
