vcs.maemo.org Git - mussorgsky/blob - src/aa_selection_dialog.py

   1 import hildon
   2 import gtk
   3 import gobject
   4 from album_art import MussorgskyAlbumArt
   5
   6 class AlbumArtSelectionDialog (gtk.Dialog):
   7
   8     def __init__ (self, parent, artist, album, size, downloader=None):
   9         """
  10         parent window, amount of images to offer
  11         Optionally downloader (for testing porpouses)
  12         """
  13         gtk.Dialog.__init__ (self,
  14                              "Select album art", parent,
  15                              gtk.DIALOG_DESTROY_WITH_PARENT,
  16                              (gtk.STOCK_CANCEL, gtk.RESPONSE_REJECT))
  17         self.artist = artist
  18         self.album = album
  19         self.size = size
  20         self.paths = []
  21         self.__create_view (size)
  22
  23         if (downloader):
  24             self.downloader = downloader
  25         else:
  26             self.downloader = MussorgskyAlbumArt ()
  27
  28         gobject.idle_add (self.__get_alternatives_async)
  29         self.selection_img = None
  30         self.selection_thumb = None
  31         hildon.hildon_gtk_window_set_progress_indicator (self, 1)
  32
  33
  34     def __create_view (self, size):
  35         hbox = gtk.HBox (homogeneous=True)
  36
  37         self.images = []
  38         self.event_boxes = []
  39         for i in range (0, size):
  40             img = gtk.Image ()
  41             img.set_size_request (124, 124)
  42             self.images.append (img)
  43
  44             event_box = gtk.EventBox ()
  45             event_box.add (img)
  46             event_box.connect ("button-release-event", self.click_on_img, i)
  47             event_box.set_sensitive (False)
  48             self.event_boxes.append (event_box)
  49
  50             hbox.pack_start (event_box, expand=False, fill=True)
  51
  52         self.vbox.add (hbox)
  53
  54     def __get_alternatives_async (self):
  55         counter = 0
  56         for (path, thumb) in self.downloader.get_alternatives (self.album, self.artist, self.size):
  57             self.paths.insert (counter, (path, thumb))
  58             self.images[counter].set_from_file (thumb)
  59             self.event_boxes [counter].set_sensitive (True)
  60             counter += 1
  61             while (gtk.events_pending()):
  62                 gtk.main_iteration()
  63
  64         while (counter < self.size):
  65                 self.images[counter].set_from_stock (gtk.STOCK_CDROM, gtk.ICON_SIZE_DIALOG)
  66                 counter += 1
  67
  68         hildon.hildon_gtk_window_set_progress_indicator (self, 0)
  69
  70
  71     def click_on_img (self, widget, event, position):
  72         img, thumb = self.paths[position]
  73         self.selection_img, self.selection_thumb = self.downloader.save_alternative (self.artist,
  74                                                                                      self.album,
  75                                                                                      img, thumb)
  76         self.response (position)
  77
  78     def get_selection (self):
  79         return (self.selection_img, self.selection_thumb)
  80
  81
  82
  83 if __name__ == "__main__":
  84
  85     import time
  86     class MockDownloader:
  87         def __init__ (self):
  88             self.alt = [("../hendrix.jpeg", "../hendrix-thumb.jpeg"),
  89                         ("../hoover.jpeg", "../hoover-thumb.jpeg"),
  90                         ("../backbeat.jpeg", "../backbeat-thumb.jpeg"),
  91                         ("../dylan.jpeg", "../dylan-thumb.jpeg")]
  92         def get_alternatives (self, album, artist, amount):
  93             for a in self.alt:
  94                 time.sleep (1)
  95                 yield a
  96         def save_alternative (self, artist, album, img, thumb):
  97             return ("/home/user/.cache/media-art/" + img, "/home/user/.thumbnails/normal/" + thumb)
  98
  99
 100     def clicked_button (self):
 101         aadd = AlbumArtSelectionDialog (w, "rory gallagher", "irish tour", 4, MockDownloader ())
 102         aadd.show_all ()
 103         response = aadd.run ()
 104         if response == gtk.RESPONSE_CLOSE or response == gtk.RESPONSE_DELETE_EVENT or response == gtk.RESPONSE_REJECT:
 105             print "Noooo"
 106         else:
 107             print "Selected", aadd.get_selection ()
 108         aadd.hide ()
 109
 110     w = gtk.Window ()
 111     w.connect ("destroy", gtk.main_quit)
 112     box = gtk.VBox ()
 113
 114     button = gtk.Button ("click")
 115     button.connect ("clicked", clicked_button)
 116     box.add (button)
 117
 118     w.add (box)
 119     w.show_all ()
 120
 121
 122     gtk.main ()
 123