Assume filtered model in the edit view

[mussorgsky] / src / album_art_panel.py

diff --git a/src/album_art_panel.py b/src/album_art_panel.py
index cf3f9ed..42287ca 100644 (file)
--- a/src/album_art_panel.py
+++ b/src/album_art_panel.py
@@ -4,7 +4,7 @@ import gtk, gobject
 from album_art_spec import getCoverArtThumbFileName
 from download_dialog import MussorgskyAlbumArtDownloadDialog
 from utils import escape_html
-from aa_selection_dialog import AlbumArtSelectionDialog
+from aa_selection_dialog import AlbumArtSelectionDialog, RESPONSE_CLICK
 
 class MussorgskyAlbumArtPanel (hildon.StackableWindow):
 
@@ -31,8 +31,6 @@ class MussorgskyAlbumArtPanel (hildon.StackableWindow):
         self.treeview.set_model (self.model)
 
     def __create_view (self):
-        vbox = gtk.VBox (spacing=12, homogeneous=False)
-
         self.treeview = gtk.TreeView ()
         self.treeview.connect ("row-activated", self.row_activated_cb)
 
@@ -43,8 +41,6 @@ class MussorgskyAlbumArtPanel (hildon.StackableWindow):
         album_art = gtk.TreeViewColumn ("Album art", gtk.CellRendererPixbuf (), pixbuf=1)
         self.treeview.append_column (album_art)
 
-        #vbox.add (self.treeview)
-
         pannable_area = hildon.PannableArea ()
         pannable_area.add (self.treeview)
         self.add (pannable_area)
@@ -73,11 +69,13 @@ class MussorgskyAlbumArtPanel (hildon.StackableWindow):
         dialog.show_all ()
         
         response = dialog.run ()
-        if (response > -1):
+        if (response == RESPONSE_CLICK):
             (img, thumb) = dialog.get_selection ()
             if img and thumb:
                 pixbuf = gtk.gdk.pixbuf_new_from_file_at_size (thumb, 64, 64)
                 treeview.get_model ().set (it, 1, pixbuf)
+            else:
+                treeview.get_model ().set (it, 1, None)
         dialog.destroy ()
             
 if __name__ == "__main__":