WIP V3

InterceptEscapeWhenEditingTableCell.py

@@ -51,7 +51,7 @@ class MyTableWidget(QTableWidget):
     def __init__(self, parent=None):
         super().__init__(parent)
         self.setItemDelegate(EscapeDelegate(self))
-        self.setEditTriggers(QAbstractItemView.EditTrigger.DoubleClicked)
+        # self.setEditTriggers(QAbstractItemView.EditTrigger.DoubleClicked)
 
 
 class MainWindow(QMainWindow):

InterceptEscapeWhenEditingTableCellInView.py

@@ -0,0 +1,99 @@
+#!/usr/bin/env python3
+
+from PyQt6.QtCore import Qt, QEvent, QObject, QVariant, QAbstractTableModel
+from PyQt6.QtWidgets import (
+    QApplication,
+    QMainWindow,
+    QMessageBox,
+    QPlainTextEdit,
+    QStyledItemDelegate,
+    QTableView,
+)
+
+from PyQt6.QtGui import QKeyEvent
+
+from typing import cast
+
+
+class EscapeDelegate(QStyledItemDelegate):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+
+    def createEditor(self, parent, option, index):
+        return QPlainTextEdit(parent)
+
+    def eventFilter(self, editor: QObject, event: QEvent):
+        """By default, QPlainTextEdit doesn't handle enter or return"""
+
+        print("EscapeDelegate event handler")
+        if event.type() == QEvent.Type.KeyPress:
+            key_event = cast(QKeyEvent, event)
+            print(key_event)
+            if key_event.key() == Qt.Key.Key_Return:
+                if key_event.modifiers() == (Qt.KeyboardModifier.ControlModifier):
+                    print("save data")
+                    self.commitData.emit(editor)
+                    self.closeEditor.emit(editor)
+                    return True
+            elif key_event.key() == Qt.Key.Key_Escape:
+                discard_edits = QMessageBox.question(
+                    self.parent(), "Abandon edit", "Discard changes?"
+                )
+                if discard_edits == QMessageBox.StandardButton.Yes:
+                    print("abandon edit")
+                    self.closeEditor.emit(editor)
+                    return True
+        return False
+
+
+class MyTableWidget(QTableView):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.setItemDelegate(EscapeDelegate(self))
+        self.setModel(MyModel())
+
+
+class MyModel(QAbstractTableModel):
+
+    def columnCount(self, index):
+        return 2
+
+    def rowCount(self, index):
+        return 2
+
+    def data(self, index, role):
+        if not index.isValid() or not (0 <= index.row() < 2):
+            return QVariant()
+
+        row = index.row()
+        column = index.column()
+        if role == Qt.ItemDataRole.DisplayRole:
+            return QVariant(f"Row {row}, Col {column}")
+        return QVariant
+
+    def flags(self, index):
+        return Qt.ItemFlag.ItemIsEnabled | Qt.ItemFlag.ItemIsSelectable | Qt.ItemFlag.ItemIsEditable
+
+
+class MainWindow(QMainWindow):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.table_widget = MyTableWidget(self)
+        # self.table_widget.setRowCount(2)
+        # self.table_widget.setColumnCount(2)
+        # for row in range(2):
+        #     for col in range(2):
+        #         item = QTableWidgetItem()
+        #         item.setText(f"Row {row}, Col {col}")
+        #         self.table_widget.setItem(row, col, item)
+        self.setCentralWidget(self.table_widget)
+
+        self.table_widget.resizeColumnsToContents()
+        self.table_widget.resizeRowsToContents()
+
+
+if __name__ == "__main__":
+    app = QApplication([])
+    window = MainWindow()
+    window.show()
+    app.exec()

app/playlistmodel.py

@@ -91,11 +91,6 @@ class PlaylistModel(QAbstractTableModel):
         super().__init__(*args, **kwargs)
 
         self.playlist_rows: dict[int, PlaylistRowData] = {}
-        # self.current_row = None
-        # self.next_row = None
-        self.previous_row = None
-        self.current_row = 2
-        self.next_row = 3
 
         self.refresh_data()
 
@@ -214,6 +209,20 @@ class PlaylistModel(QAbstractTableModel):
 
         return QVariant()
 
+    def flags(self, index: QModelIndex) -> Qt.ItemFlag:
+        """
+        Standard model flags
+        """
+
+        if not index.isValid():
+            return Qt.ItemFlag.ItemIsEnabled
+
+        default = Qt.ItemFlag.ItemIsEnabled | Qt.ItemFlag.ItemIsSelectable
+        if index.column() in [Col.TITLE.value, Col.ARTIST.value, Col.NOTE.value]:
+            return default | Qt.ItemFlag.ItemIsEditable
+
+        return default
+
     def headerData(
         self,
         section: int,
@@ -266,3 +275,13 @@ class PlaylistModel(QAbstractTableModel):
         """Standard function for view"""
 
         return len(self.playlist_rows)
+
+    def setData(
+        self, index: QModelIndex, value: QVariant, role: int = Qt.ItemDataRole.EditRole
+    ) -> bool:
+        """
+        Update model with edited data
+        """
+
+        print(f"setData({index.row()=}, {index.column()=}, {value=}, {role=})")
+        return True