From f2305655d86728e3fa8fb3214dd9acf10fce6372 Mon Sep 17 00:00:00 2001 From: Matias Linares Date: Wed, 5 Mar 2025 17:48:15 -0300 Subject: models/file.py: Update some signatures --- kodereviewer/models/file.py | 14 +++++++------- 1 file 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: -- cgit v1.2.3-70-g09d2