From 1db3990cd66ff71241c6375dc392b6c7a6069c18 Mon Sep 17 00:00:00 2001
From: Keith Edmunds <kae@tiger-computing.co.uk>
Date: Wed, 29 Nov 2023 07:57:36 +0000
Subject: [PATCH] V3: add note colouring

---
 app/playlistmodel.py | 9 ++++++++-
 app/playlists.py     | 3 +++
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/app/playlistmodel.py b/app/playlistmodel.py
index cbeeb37..31693d4 100644
--- a/app/playlistmodel.py
+++ b/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()
 
diff --git a/app/playlists.py b/app/playlists.py
index fc7066b..aa7a6c1 100644
--- a/app/playlists.py
+++ b/app/playlists.py
@@ -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