REF/ENH: move common model methods to a common view submodule, start …

…work on LivePVTableModel and NestableTableModel

superscore/tests/test_widgets.py

@@ -7,7 +7,7 @@
 
 from superscore.model import Collection, Root
 from superscore.widgets.core import DataWidget
-from superscore.widgets.tree import RootTree
+from superscore.widgets.views import RootTree
 
 
 @pytest.mark.parametrize(

superscore/widgets/page/entry.py

@@ -7,7 +7,7 @@
 from superscore.model import Collection
 from superscore.widgets.core import DataWidget, Display, NameDescTagsWidget
 from superscore.widgets.manip_helpers import insert_widget
-from superscore.widgets.tree import RootTree
+from superscore.widgets.views import RootTree
 
 
 class CollectionPage(Display, DataWidget):

superscore/widgets/page/search.py

@@ -11,6 +11,7 @@
 from superscore.model import Collection, Entry, Readback, Setpoint, Snapshot
 from superscore.widgets import ICON_MAP
 from superscore.widgets.core import Display
+from superscore.widgets.views import BaseTableEntryModel, ButtonDelegate
 
 logger = logging.getLogger(__name__)
 
@@ -150,7 +151,7 @@ def subfilter_results(self) -> None:
         self.proxy_model.invalidateFilter()
 
 
-class ResultModel(QtCore.QAbstractTableModel):
+class ResultModel(BaseTableEntryModel):
     headers: List[str] = ['Name', 'Type', 'Description', 'Created', 'Open']
 
     def __init__(self, *args, entries: List[Entry] = None, **kwargs) -> None:
@@ -198,76 +199,6 @@ def data(self, index: QtCore.QModelIndex, role: int) -> Any:
         # if nothing is found, return invalid QVariant
         return QtCore.QVariant()
 
-    def rowCount(self, index):
-        return len(self.entries)
-
-    def columnCount(self, index):
-        return len(self.headers)
-
-    def headerData(
-        self,
-        section: int,
-        orientation: QtCore.Qt.Orientation,
-        role: int
-    ) -> Any:
-        """
-        Returns the header data for the model.
-        Currently only displays horizontal header data
-        """
-        if role != QtCore.Qt.DisplayRole:
-            return
-
-        if orientation == QtCore.Qt.Horizontal:
-            return self.headers[section]
-
-    def flags(self, index: QtCore.QModelIndex) -> QtCore.Qt.ItemFlag:
-        """
-        Returns the item flags for the given ``index``.  The returned
-        item flag controls what behaviors the item supports.
-
-        Parameters
-        ----------
-        index : QtCore.QModelIndex
-            the index referring to a cell of the TableView
-
-        Returns
-        -------
-        QtCore.Qt.ItemFlag
-            the ItemFlag corresponding to the cell
-        """
-        if (index.column() == len(self.headers) - 1):
-            return QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsEnabled
-        else:
-            return QtCore.Qt.ItemIsEnabled
-
-
-class ButtonDelegate(QtWidgets.QStyledItemDelegate):
-    clicked = QtCore.Signal(QtCore.QModelIndex)
-
-    def __init__(self, *args, button_text: str = '', **kwargs):
-        self.button_text = button_text
-        super().__init__(*args, **kwargs)
-
-    def createEditor(
-        self,
-        parent: QtWidgets.QWidget,
-        option,
-        index: QtCore.QModelIndex
-    ) -> QtWidgets.QWidget:
-        button = QtWidgets.QPushButton(self.button_text, parent)
-        button.clicked.connect(
-            lambda _, index=index: self.clicked.emit(index)
-        )
-        return button
-
-    def updateEditorGeometry(
-        self,
-        editor: QtWidgets.QWidget,
-        option: QtWidgets.QStyleOptionViewItem,
-        index: QtCore.QModelIndex
-    ) -> None:
-        return editor.setGeometry(option.rect)
-
 
 class ResultFilterProxyModel(QtCore.QSortFilterProxyModel):
     """