Fix 14 QAbstractTableModel example

jone7875 · May 18, 2022 · f23cbb2 · f23cbb2
1 parent 6ba2eb7
commit f23cbb2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/14 QAbstractTableModel example/README.md b/src/14 QAbstractTableModel example/README.md
@@ -31,12 +31,12 @@ class TableModel(QAbstractTableModel):
         # How many columns?
         return len(headers)
     def data(self, index, role):
-        if role != Qt.DisplayRole:
+        if role != Qt.ItemDataRole.DisplayRole:
             return QVariant()
         # What's the value of the cell at the given index?
         return rows[index.row()][index.column()]
     def headerData(self, section, orientation, role):
-        if role != Qt.DisplayRole or orientation != Qt.Horizontal:
+        if role != Qt.ItemDataRole.DisplayRole or orientation != Qt.Orientation.Horizontal:
             return QVariant()
         # What's the header for the given column?
         return headers[section]

diff --git a/src/14 QAbstractTableModel example/main.py b/src/14 QAbstractTableModel example/main.py
@@ -12,11 +12,11 @@ def rowCount(self, parent):
     def columnCount(self, parent):
         return len(headers)
     def data(self, index, role):
-        if role != Qt.DisplayRole:
+        if role != Qt.ItemDataRole.DisplayRole:
             return QVariant()
         return rows[index.row()][index.column()]
     def headerData(self, section, orientation, role):
-        if role != Qt.DisplayRole or orientation != Qt.Horizontal:
+        if role != Qt.ItemDataRole.DisplayRole or orientation != Qt.Orientation.Horizontal:
             return QVariant()
         return headers[section]