Fix (innocuous) mypy warning

Was asking user to select a track when next track selected.

app/playlistmodel.py

@@ -34,7 +34,7 @@ from helpers import (
     set_track_metadata,
 )
 from log import log
-from models import Playdates, PlaylistRows, Tracks
+from models import NoteColours, Playdates, PlaylistRows, Tracks
 
 
 HEADER_NOTES_COLUMN = 1
@@ -213,6 +213,13 @@ class PlaylistModel(QAbstractTableModel):
                 return QBrush(QColor(Config.COLOUR_BITRATE_MEDIUM))
             else:
                 return QBrush(QColor(Config.COLOUR_BITRATE_OK))
+        if column == Col.NOTE.value:
+            with Session() as session:
+                note_colour = NoteColours.get_colour(session, prd.note)
+                if note_colour:
+                    return QBrush(QColor(note_colour))
+                else:
+                    return QBrush()
 
         return QBrush()
 

app/playlists.py

@@ -1,5 +1,5 @@
 from pprint import pprint
-from typing import Callable, cast, List, Optional, TYPE_CHECKING
+from typing import Callable, cast, List, Optional, overload, TYPE_CHECKING
 
 from PyQt6.QtCore import (
     QEvent,
@@ -225,6 +225,9 @@ class PlaylistTab(QTableView):
         # edited
         self.data_model.update_track_times()
 
+        # Deselect edited line
+        self.clear_selection()
+
     def dropEvent(self, event):
         if event.source() is not self or (
             event.dropAction() != Qt.DropAction.MoveAction
@@ -248,16 +251,24 @@ class PlaylistTab(QTableView):
             self.clear_selection()
 
             # Resize rows
-            self.signals.resize_rows_signal.emit(self.playlist_id)
+            self.resizeRowsToContents()
 
             event.accept()
 
+    @overload
+    def edit(self, index: QModelIndex) -> None:
+        ...
+
+    @overload
     def edit(
         self,
         index: QModelIndex,
         trigger: QAbstractItemView.EditTrigger,
-        event: Optional[QEvent],
+        event: Optional[QEvent]
     ) -> bool:
+        ...
+
+    def edit(self, index, trigger, event):
         """
         Override QAbstractItemView.edit to catch when editing starts
 
@@ -633,11 +644,13 @@ class PlaylistTab(QTableView):
 
         row_indexes = self._selected_row_indexes()
 
-        if len(row_indexes) != 1:
+        if len(row_indexes) > 1:
             show_warning(
-                self.musicmuster, "No or multiple rows selected", "Select only one row"
+                self.musicmuster, "Multiple rows selected", "Select only one row"
             )
             return None
+        elif not row_indexes:
+            return None
 
         return row_indexes[0]