[chore] Add generics methods for models

Kastakin · Dec 5, 2023 · d0b61c3 · d0b61c3
1 parent e39cd8a
commit d0b61c3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/main/python/pyes/viewmodels/models.py b/src/main/python/pyes/viewmodels/models.py
@@ -4,6 +4,7 @@
 import re
 import string
 from itertools import permutations
+from typing import Any
 
 import pandas as pd
 from commands import ComponentsCellEdit, SpeciesCellEdit
@@ -126,6 +127,12 @@ def rowCount(self, index=QModelIndex()):
     def columnCount(self, index=QModelIndex()):
         return self._data.columns.size
 
+    def data(self, index, role: Qt.ItemDataRole = Qt.ItemDataRole.DisplayRole):
+        raise NotImplementedError()
+
+    def setData(self, index: QModelIndex, value: Any, role) -> bool:
+        raise NotImplementedError
+
 
 class ConcentrationsModel(GenericModel):
     def __init__(self, data: pd.DataFrame, undo_stack: QUndoStack):