Tons of fixes/tweaks/changes in general

[jamaendo] / jamaui / albumlist.py

diff --git a/jamaui/albumlist.py b/jamaui/albumlist.py
index e851cdb..ce1671a 100644 (file)
--- a/jamaui/albumlist.py
+++ b/jamaui/albumlist.py
@@ -66,5 +66,36 @@ class TrackList(gtk.TreeView):
 
     def get_track_id(self, path):
         treeiter = self.__store.get_iter(path)
 
     def get_track_id(self, path):
         treeiter = self.__store.get_iter(path)

+        _, _, _id = self.__store.get(treeiter, 0, 1, 2)
+        return _id
+
+class RadioList(gtk.TreeView):
+    def __init__(self):
+        gtk.TreeView.__init__(self)
+        self.__store = gtk.ListStore(str, int)
+        self.set_model(self.__store)
+
+        col = gtk.TreeViewColumn('Name')
+        self.append_column(col)
+        cell = gtk.CellRendererText()
+        col.pack_start(cell, True)
+        col.add_attribute(cell, 'text', 0)
+
+        self.set_search_column(0)
+        col.set_sort_column_id(0)
+
+    def add_radio(self, radio):
+        self.__store.append([self.radio_name(radio), radio.ID])
+
+    def get_radio_id(self, path):
+        treeiter = self.__store.get_iter(path)

         _, _id = self.__store.get(treeiter, 0, 1)
         return _id
         _, _id = self.__store.get(treeiter, 0, 1)
         return _id

+
+    def radio_name(self, radio):
+        if radio.idstr:
+            return radio.idstr.capitalize()
+        elif radio.name:
+            return radio.name
+        else:
+            return "Radio #%s" % (radio.ID)