From b1841b02ea73db33551d20ea0d20dbd54d6ff5de Mon Sep 17 00:00:00 2001
From: Keith Edmunds <kae@tiger-computing.co.uk>
Date: Sun, 5 Mar 2023 20:05:04 +0000
Subject: [PATCH] WIP playlists refactor: set header colour when removing track

---
 app/playlists.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/app/playlists.py b/app/playlists.py
index 14aa045..fadc8f3 100644
--- a/app/playlists.py
+++ b/app/playlists.py
@@ -1513,10 +1513,8 @@ class PlaylistTab(QTableWidget):
             # Set note text in correct column for section head
             _ = self._set_item_text(row, HEADER_NOTES_COLUMN, plr.note)
 
-            note_colour = NoteColours.get_colour(session, plr.note)
-            if not note_colour:
-                note_colour = Config.COLOUR_NOTES_PLAYLIST
-            self._set_row_colour(row, QColor(note_colour))
+            _ = self._set_row_note_colour(session, row, plr.note,
+                                          section_header=True)
             self.clear_selection()
 
     def _rescan(self, row: int, track_id: int) -> None: