diff options
Diffstat (limited to 'kodereviewer/models/file.py')
| -rw-r--r-- | kodereviewer/models/file.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/kodereviewer/models/file.py b/kodereviewer/models/file.py index cc3f8be..2e00755 100644 --- a/kodereviewer/models/file.py +++ b/kodereviewer/models/file.py @@ -158,21 +158,21 @@ class TreeFileModel(QAbstractItemModel): return QModelIndex() return self.createIndex(parent.row(), 0, parent) - def rowCount(self, index: QModelIndex = QModelIndex()) -> int: + def rowCount(self, parent: QModelIndex | QPersistentModelIndex = QModelIndex()) -> int: """Returns the number of rows under the given parent When the parent is valid it means that rowCount is returning the number of children of parent. """ - parent: FileItem - if index.isValid(): - parent = index.internalPointer() + node: FileItem + if parent.isValid(): + node = parent.internalPointer() else: - parent = self.root_node + node = self.root_node - return len(parent.children) + return len(node.children) - def columnCount(self, parent: QModelIndex = QModelIndex()) -> int: + def columnCount(self, parent: QModelIndex | QPersistentModelIndex = QModelIndex()) -> int: return 1 def data(self, index: QModelIndex | QPersistentModelIndex, role: int = Qt.ItemDataRole.DisplayRole) -> Any: |