--- a/src/eric7/WebBrowser/History/HistoryTreeModel.py Wed Jul 13 11:16:20 2022 +0200
+++ b/src/eric7/WebBrowser/History/HistoryTreeModel.py Wed Jul 13 14:55:47 2022 +0200
@@ -20,36 +20,36 @@
"""
Class implementing the history tree model.
"""
+
def __init__(self, sourceModel, parent=None):
"""
Constructor
-
+
@param sourceModel reference to the source model (QAbstractItemModel)
@param parent reference to the parent object (QObject)
"""
super().__init__(parent)
-
+
self.__sourceRowCache = []
self.__removingDown = False
-
+
self.setSourceModel(sourceModel)
-
- def headerData(self, section, orientation,
- role=Qt.ItemDataRole.DisplayRole):
+
+ def headerData(self, section, orientation, role=Qt.ItemDataRole.DisplayRole):
"""
Public method to get the header data.
-
+
@param section section number (integer)
@param orientation header orientation (Qt.Orientation)
@param role data role (Qt.ItemDataRole)
@return header data
"""
return self.sourceModel().headerData(section, orientation, role)
-
+
def data(self, index, role=Qt.ItemDataRole.DisplayRole):
"""
Public method to get data from the model.
-
+
@param index index of history entry to get data for (QModelIndex)
@param role data role (integer)
@return history entry data
@@ -66,104 +66,103 @@
return date.toString("yyyy-MM-dd")
if index.column() == 1:
return self.tr(
- "%n item(s)", "",
- self.rowCount(index.sibling(index.row(), 0)))
-
+ "%n item(s)", "", self.rowCount(index.sibling(index.row(), 0))
+ )
+
elif (
- role == Qt.ItemDataRole.DecorationRole and
- index.column() == 0 and
- not index.parent().isValid()
+ role == Qt.ItemDataRole.DecorationRole
+ and index.column() == 0
+ and not index.parent().isValid()
):
return UI.PixmapCache.getIcon("history")
-
+
elif (
- role == HistoryModel.DateRole and
- index.column() == 0 and
- index.internalId() == 0
+ role == HistoryModel.DateRole
+ and index.column() == 0
+ and index.internalId() == 0
):
offset = self.__sourceDateRow(index.row())
idx = self.sourceModel().index(offset, 0)
return idx.data(HistoryModel.DateRole)
-
+
return QAbstractProxyModel.data(self, index, role)
-
+
def columnCount(self, parent=None):
"""
Public method to get the number of columns.
-
+
@param parent index of parent (QModelIndex)
@return number of columns (integer)
"""
if parent is None:
parent = QModelIndex()
-
+
return self.sourceModel().columnCount(self.mapToSource(parent))
-
+
def rowCount(self, parent=None):
"""
Public method to determine the number of rows.
-
+
@param parent index of parent (QModelIndex)
@return number of rows (integer)
"""
if parent is None:
parent = QModelIndex()
-
+
if (
- parent.internalId() != 0 or
- parent.column() > 0 or
- self.sourceModel() is None
+ parent.internalId() != 0
+ or parent.column() > 0
+ or self.sourceModel() is None
):
return 0
-
+
# row count OF dates
if not parent.isValid():
if self.__sourceRowCache:
return len(self.__sourceRowCache)
-
+
currentDate = QDate()
rows = 0
totalRows = self.sourceModel().rowCount()
-
+
for row in range(totalRows):
- rowDate = self.sourceModel().index(row, 0).data(
- HistoryModel.DateRole)
+ rowDate = self.sourceModel().index(row, 0).data(HistoryModel.DateRole)
if rowDate != currentDate:
self.__sourceRowCache.append(row)
currentDate = rowDate
rows += 1
return rows
-
+
# row count FOR a date
start = self.__sourceDateRow(parent.row())
end = self.__sourceDateRow(parent.row() + 1)
return end - start
-
+
def __sourceDateRow(self, row):
"""
Private method to translate the top level date row into the offset
where that date starts.
-
+
@param row row number of the date (integer)
@return offset where that date starts (integer)
"""
if row <= 0:
return 0
-
+
if len(self.__sourceRowCache) == 0:
self.rowCount(QModelIndex())
-
+
if row >= len(self.__sourceRowCache):
if self.sourceModel() is None:
return 0
return self.sourceModel().rowCount()
-
+
return self.__sourceRowCache[row]
-
+
def mapToSource(self, proxyIndex):
"""
Public method to map an index to the source model index.
-
+
@param proxyIndex reference to a proxy model index (QModelIndex)
@return source model index (QModelIndex)
"""
@@ -172,12 +171,13 @@
return QModelIndex()
startDateRow = self.__sourceDateRow(offset - 1)
return self.sourceModel().index(
- startDateRow + proxyIndex.row(), proxyIndex.column())
-
+ startDateRow + proxyIndex.row(), proxyIndex.column()
+ )
+
def index(self, row, column, parent=None):
"""
Public method to create an index.
-
+
@param row row number for the index (integer)
@param column column number for the index (integer)
@param parent index of the parent item (QModelIndex)
@@ -185,15 +185,15 @@
"""
if parent is None:
parent = QModelIndex()
-
+
if (
- row < 0 or
- column < 0 or
- column >= self.columnCount(parent) or
- parent.column() > 0
+ row < 0
+ or column < 0
+ or column >= self.columnCount(parent)
+ or parent.column() > 0
):
return QModelIndex()
-
+
if not parent.isValid():
return self.createIndex(row, column, 0)
return self.createIndex(row, column, parent.row() + 1)
@@ -201,7 +201,7 @@
def parent(self, index):
"""
Public method to get the parent index.
-
+
@param index index of item to get parent (QModelIndex)
@return index of parent (QModelIndex)
"""
@@ -209,62 +209,63 @@
if offset == 0 or not index.isValid():
return QModelIndex()
return self.createIndex(offset - 1, 0, 0)
-
+
def hasChildren(self, parent=None):
"""
Public method to check, if an entry has some children.
-
+
@param parent index of the entry to check (QModelIndex)
@return flag indicating the presence of children (boolean)
"""
if parent is None:
parent = QModelIndex()
-
+
grandparent = parent.parent()
if not grandparent.isValid():
return True
return False
-
+
def flags(self, index):
"""
Public method to get the item flags.
-
+
@param index index of the item (QModelIndex)
@return flags (Qt.ItemFlags)
"""
return (
Qt.ItemFlag.NoItemFlags
- if not index.isValid() else
- (Qt.ItemFlag.ItemIsSelectable |
- Qt.ItemFlag.ItemIsEnabled |
- Qt.ItemFlag.ItemIsDragEnabled)
+ if not index.isValid()
+ else (
+ Qt.ItemFlag.ItemIsSelectable
+ | Qt.ItemFlag.ItemIsEnabled
+ | Qt.ItemFlag.ItemIsDragEnabled
+ )
)
-
+
def setSourceModel(self, sourceModel):
"""
Public method to set the source model.
-
+
@param sourceModel reference to the source model (QAbstractItemModel)
"""
if self.sourceModel() is not None:
self.sourceModel().modelReset.disconnect(self.__sourceReset)
self.sourceModel().layoutChanged.disconnect(self.__sourceReset)
- self.sourceModel().rowsInserted.disconnect(
- self.__sourceRowsInserted)
+ self.sourceModel().rowsInserted.disconnect(self.__sourceRowsInserted)
self.sourceModel().rowsRemoved.disconnect(self.__sourceRowsRemoved)
-
+
super().setSourceModel(sourceModel)
-
+
if self.sourceModel() is not None:
self.__loaded = False
self.sourceModel().modelReset.connect(self.__sourceReset)
self.sourceModel().layoutChanged.connect(self.__sourceReset)
self.sourceModel().rowsInserted.connect(self.__sourceRowsInserted)
self.sourceModel().rowsRemoved.connect(self.__sourceRowsRemoved)
-
+
self.beginResetModel()
self.endResetModel()
-
+
def __sourceReset(self):
"""
Private slot to handle a reset of the source model.
@@ -272,11 +273,11 @@
self.beginResetModel()
self.__sourceRowCache = []
self.endResetModel()
-
+
def __sourceRowsInserted(self, parent, start, end):
"""
Private slot to handle the insertion of data in the source model.
-
+
@param parent reference to the parent index (QModelIndex)
@param start start row (integer)
@param end end row (integer)
@@ -287,7 +288,7 @@
self.__sourceRowCache = []
self.endResetModel()
return
-
+
self.__sourceRowCache = []
treeIndex = self.mapFromSource(self.sourceModel().index(start, 0))
treeParent = treeIndex.parent()
@@ -295,40 +296,39 @@
self.beginInsertRows(QModelIndex(), 0, 0)
self.endInsertRows()
else:
- self.beginInsertRows(treeParent, treeIndex.row(),
- treeIndex.row())
+ self.beginInsertRows(treeParent, treeIndex.row(), treeIndex.row())
self.endInsertRows()
-
+
def mapFromSource(self, sourceIndex):
"""
Public method to map an index to the proxy model index.
-
+
@param sourceIndex reference to a source model index (QModelIndex)
@return proxy model index (QModelIndex)
"""
if not sourceIndex.isValid():
return QModelIndex()
-
+
if len(self.__sourceRowCache) == 0:
self.rowCount(QModelIndex())
-
+
try:
row = self.__sourceRowCache.index(sourceIndex.row())
except ValueError:
row = bisect.bisect_left(self.__sourceRowCache, sourceIndex.row())
if (
- row == len(self.__sourceRowCache) or
- self.__sourceRowCache[row] != sourceIndex.row()
+ row == len(self.__sourceRowCache)
+ or self.__sourceRowCache[row] != sourceIndex.row()
):
row -= 1
dateRow = max(0, row)
row = sourceIndex.row() - self.__sourceRowCache[dateRow]
return self.createIndex(row, sourceIndex.column(), dateRow + 1)
-
+
def removeRows(self, row, count, parent=None):
"""
Public method to remove entries from the model.
-
+
@param row row of the first entry to remove (integer)
@param count number of entries to remove (integer)
@param parent index of the parent entry (QModelIndex)
@@ -336,14 +336,10 @@
"""
if parent is None:
parent = QModelIndex()
-
- if (
- row < 0 or
- count <= 0 or
- row + count > self.rowCount(parent)
- ):
+
+ if row < 0 or count <= 0 or row + count > self.rowCount(parent):
return False
-
+
self.__removingDown = True
if parent.isValid() and self.rowCount(parent) == count - row:
self.beginRemoveRows(QModelIndex(), parent.row(), parent.row())
@@ -358,15 +354,14 @@
for i in range(row + count - 1, row - 1, -1):
dateParent = self.index(i, 0)
offset = self.__sourceDateRow(dateParent.row())
- if not self.sourceModel().removeRows(
- offset, self.rowCount(dateParent)):
+ if not self.sourceModel().removeRows(offset, self.rowCount(dateParent)):
return False
return True
-
+
def __sourceRowsRemoved(self, parent, start, end):
"""
Private slot to handle the removal of data in the source model.
-
+
@param parent reference to the parent index (QModelIndex)
@param start start row (integer)
@param end end row (integer)
@@ -376,7 +371,7 @@
self.__sourceRowCache = []
self.endResetModel()
return
-
+
if not parent.isValid():
if self.__sourceRowCache:
i = end
@@ -386,8 +381,8 @@
except ValueError:
ind = bisect.bisect_left(self.__sourceRowCache, i)
if (
- ind == len(self.__sourceRowCache) or
- self.__sourceRowCache[ind] != i
+ ind == len(self.__sourceRowCache)
+ or self.__sourceRowCache[ind] != i
):
ind -= 1
row = max(0, ind)
@@ -404,7 +399,7 @@
i -= 1
for j in range(row, len(self.__sourceRowCache)):
self.__sourceRowCache[j] -= 1
-
+
if self.__removingDown:
self.endRemoveRows()
self.__removingDown = False
