X-Git-Url: https://code.delx.au/notipod/blobdiff_plain/091ffd73bb5472213fe7b2e40e85a8ac1a418891..02030014f9372d738ae6a519ec71c464dbfb0f6a:/notipod_gui.py

diff --git a/notipod_gui.py b/notipod_gui.py
index 9b299ea..39d911f 100644
--- a/notipod_gui.py
+++ b/notipod_gui.py
@@ -48,16 +48,23 @@ class PlaylistModel(NSObject):
 			return len(playlist.children)
 
 	def outlineView_objectValueForTableColumn_byItem_(self, _, col, playlist):
-		col = col.identifier() if col else "playlist"
+		if not col:
+			return
+		col = col.identifier()
 
 		if col == "selected":
 			selected = NSApp.delegate().playlists()
 			return playlist.pid in selected
-		if col == None or col == "playlist":
+		if col == "icon":
+			return NSImage.imageNamed_("playlist-" + playlist.ptype)
+		if col == "playlist":
 			return playlist.name
 
 	def outlineView_setObjectValue_forTableColumn_byItem_(self, _, v, col, playlist):
-		col = col.identifier() if col else "playlist"
+		if not col:
+			return
+		col = col.identifier()
+
 
 		if col != "selected":
 			return