home *** CD-ROM | disk | FTP | other *** search
/ Mac Easy 2010 May / Mac Life Ubuntu.iso / casper / filesystem.squashfs / usr / lib / rhythmbox / plugins / artdisplay / CoverArtDatabase.py < prev    next >
Encoding:
Python Source  |  2009-04-07  |  8.3 KB  |  257 lines
  1. # -*- Mode: python; coding: utf-8; tab-width: 8; indent-tabs-mode: t; -*- 
  2. #
  3. # Copyright (C) 2006 - Gareth Murphy, Martin Szulecki, 
  4. # Ed Catmur <ed@catmur.co.uk>
  5. #
  6. # This program is free software; you can redistribute it and/or modify
  7. # it under the terms of the GNU General Public License as published by
  8. # the Free Software Foundation; either version 2, or (at your option)
  9. # any later version.
  10. #
  11. # The Rhythmbox authors hereby grant permission for non-GPL compatible
  12. # GStreamer plugins to be used and distributed together with GStreamer
  13. # and Rhythmbox. This permission is above and beyond the permissions granted
  14. # by the GPL license by which Rhythmbox is covered. If you modify this code
  15. # you may extend this exception to your version of the code, but you are not
  16. # obligated to do so. If you do not wish to do so, delete this exception
  17. # statement from your version.
  18. # 
  19. # This program is distributed in the hope that it will be useful,
  20. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  21. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  22. # GNU General Public License for more details.
  23. #
  24. # You should have received a copy of the GNU General Public License
  25. # along with this program; if not, write to the Free Software
  26. # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301  USA.
  27.  
  28. import rhythmdb, rb
  29. import os
  30. import gtk
  31. import itertools
  32. import gobject
  33.  
  34. from PodcastCoverArtSearch import PodcastCoverArtSearch
  35. from AmazonCoverArtSearch import AmazonCoverArtSearch
  36.  
  37. from urllib import unquote
  38.  
  39. try:
  40.     # try to use the gio implementation, fall back to gnome-vfs if that
  41.     # isn't available.
  42.     import gio
  43.     if gio.pygio_version > (2,15,2):    # probably
  44.         from LocalCoverArtSearchGIO import LocalCoverArtSearch
  45. except:
  46.     from LocalCoverArtSearch import LocalCoverArtSearch
  47.  
  48. ART_SEARCHES_LOCAL = [LocalCoverArtSearch]
  49. ART_SEARCHES_REMOTE = [PodcastCoverArtSearch, AmazonCoverArtSearch]
  50. OLD_ART_FOLDER = '~/.gnome2/rhythmbox/covers'
  51.  
  52. ART_FOLDER = os.path.join(rb.user_cache_dir(), 'covers')
  53. ART_CACHE_EXTENSION_JPG = 'jpg'
  54. ART_CACHE_EXTENSION_PNG = 'png'
  55. ART_CACHE_FORMAT_JPG = 'jpeg'
  56. ART_CACHE_FORMAT_PNG = 'png'
  57. ART_CACHE_SETTINGS_JPG = {"quality": "100"}
  58. ART_CACHE_SETTINGS_PNG = {"compression": "9"}
  59.  
  60. class TicketSystem:
  61.     def __init__ (self):
  62.         self.counter = itertools.count ()
  63.         self.hash = {}
  64.         self.dead = set ()
  65.  
  66.     def get (self, item):
  67.         ticket = self.counter.next ()
  68.         self.hash.setdefault (item, set ()).add (ticket)
  69.         return ticket
  70.  
  71.     def bury (self, ticket):
  72.         try:
  73.             self.dead.remove (ticket)
  74.             return True
  75.         except KeyError:
  76.             return False
  77.  
  78.     def forget (self, item, ticket):
  79.         try:
  80.             self.hash[item].remove (ticket)
  81.             return True
  82.         except KeyError:
  83.             self.dead.remove (ticket)
  84.             return False
  85.  
  86.     def purge (self, item):
  87.         self.dead.update (self.hash.pop (item, set ()))
  88.  
  89.     def release (self, item, ticket):
  90.         try:
  91.             self.dead.update (self.hash.pop (item) - set([ticket]))
  92.             return True
  93.         except KeyError:
  94.             self.dead.remove (ticket)
  95.             return False
  96.  
  97. class CoverArtDatabase (object):
  98.     def __init__ (self):
  99.         self.ticket = TicketSystem ()
  100.  
  101.     def build_art_cache_filename (self, db, entry, extension):
  102.         artist = db.entry_get (entry, rhythmdb.PROP_ARTIST)
  103.         album = db.entry_get (entry, rhythmdb.PROP_ALBUM)
  104.         art_folder = os.path.expanduser (ART_FOLDER)
  105.         old_art_folder = os.path.expanduser (OLD_ART_FOLDER)
  106.         if not os.path.exists (art_folder) and os.path.exists (old_art_folder):
  107.             parent = os.path.dirname(os.path.abspath(art_folder))
  108.             if not os.path.exists (parent):
  109.                 os.makedirs (parent)
  110.             os.rename (old_art_folder, art_folder)
  111.         if not os.path.exists (art_folder):
  112.             os.makedirs (art_folder)
  113.  
  114.         # FIXME: the following block of code is messy and needs to be redone ASAP
  115.         return art_folder + '/%s - %s.%s' % (artist.replace ('/', '-'), album.replace ('/', '-'), extension)    
  116.  
  117.     def engines (self, blist):
  118.         for Engine in ART_SEARCHES_LOCAL:
  119.             yield Engine (), Engine.__name__, False
  120.         for Engine in ART_SEARCHES_REMOTE:
  121.             if Engine.__name__ not in blist:
  122.                 yield Engine (), Engine.__name__, True
  123.     
  124.     def set_pixbuf_from_uri (self, db, entry, uri, callback):
  125.         def loader_cb (data):
  126.             self.set_pixbuf (db, entry, self.image_data_load (data), callback)
  127.  
  128.         l = rb.Loader()
  129.         l.get_url (str (uri), loader_cb)
  130.  
  131.     def set_pixbuf (self, db, entry, pixbuf, callback):
  132.         if entry is None or pixbuf is None:
  133.             return
  134.         if pixbuf.get_has_alpha():
  135.             art_location = self.build_art_cache_filename (db, entry, ART_CACHE_EXTENSION_PNG)
  136.             art_cache_format = ART_CACHE_FORMAT_PNG
  137.             art_cache_settings = ART_CACHE_SETTINGS_PNG
  138.         else:
  139.             art_location = self.build_art_cache_filename (db, entry, ART_CACHE_EXTENSION_JPG)
  140.             art_cache_format = ART_CACHE_FORMAT_JPG
  141.             art_cache_settings = ART_CACHE_SETTINGS_JPG
  142.         self.ticket.purge (entry)
  143.         pixbuf.save (art_location, art_cache_format, art_cache_settings)
  144.         callback (entry, pixbuf, art_location)
  145.         for Engine in ART_SEARCHES_LOCAL:
  146.             try:
  147.                 Engine ().save_pixbuf (db, entry, pixbuf)
  148.             except AttributeError:
  149.                 pass
  150.  
  151.     def cancel_get_pixbuf (self, entry):
  152.         self.ticket.purge (entry)
  153.   
  154.     def get_pixbuf (self, db, entry, callback):
  155.         if entry is None:
  156.             callback (entry, None, None)
  157.             return
  158.             
  159.         st_artist = db.entry_get (entry, rhythmdb.PROP_ARTIST) or _("Unknown")
  160.         st_album = db.entry_get (entry, rhythmdb.PROP_ALBUM) or _("Unknown")
  161.  
  162.         # replace quote characters
  163.         # don't replace single quote: could be important punctuation
  164.         for char in ["\""]:
  165.             st_artist = st_artist.replace (char, '')
  166.             st_album = st_album.replace (char, '')
  167.  
  168.         rb.Coroutine (self.image_search, db, st_album, st_artist, entry, callback).begin ()
  169.  
  170.     def image_search (self, plexer, db, st_album, st_artist, entry, callback):
  171.         art_location_jpg = self.build_art_cache_filename (db, entry, ART_CACHE_EXTENSION_JPG)
  172.         art_location_png = self.build_art_cache_filename (db, entry, ART_CACHE_EXTENSION_PNG)
  173.         blist_location = self.build_art_cache_filename (db, entry, "rb-blist")
  174.  
  175.         art_location = None
  176.         if os.path.exists (art_location_jpg):
  177.             art_location = art_location_jpg
  178.         if os.path.exists (art_location_png):
  179.             art_location = art_location_png
  180.  
  181.         # Check local cache
  182.         if art_location:
  183.             self.ticket.purge (entry)
  184.             pixbuf = gtk.gdk.pixbuf_new_from_file (art_location)    
  185.             callback (entry, pixbuf, art_location)
  186.             return
  187.  
  188.         blist = self.read_blist (blist_location)
  189.         ticket = self.ticket.get (entry)
  190.         for engine, engine_name, engine_remote in self.engines (blist):
  191.             plexer.clear ()
  192.             engine.search (db, entry, plexer.send ())
  193.             while True:
  194.                 yield None
  195.                 _, (engine, entry, results) = plexer.receive ()
  196.                 if not results:
  197.                     break
  198.                 for url in engine.get_best_match_urls (results):
  199.                     if str(url) == "":
  200.                         print "got empty url from engine %s." % (engine)
  201.                         continue
  202.  
  203.                     l = rb.Loader()
  204.                     yield l.get_url (str (url), plexer.send ())
  205.                     _, (data, ) = plexer.receive ()
  206.                     pixbuf = self.image_data_load (data)
  207.                     if pixbuf:
  208.                         if self.ticket.release (entry, ticket):
  209.                             if engine_remote:
  210.                                 if pixbuf.get_has_alpha ():
  211.                                     pixbuf.save (art_location_png, ART_CACHE_FORMAT_PNG, ART_CACHE_SETTINGS_PNG)
  212.                                 else:
  213.                                     pixbuf.save (art_location_jpg, ART_CACHE_FORMAT_JPG, ART_CACHE_SETTINGS_JPG)
  214.                                 uri = art_location
  215.                             else:
  216.                                 uri = unquote (str (url))
  217.                             callback (entry, pixbuf, uri)
  218.                         self.write_blist (blist_location, blist)
  219.                         return
  220.                 if not engine.search_next ():
  221.                     if engine_remote:
  222.                         blist.append (engine_name)
  223.                     break
  224.                 if self.ticket.bury (ticket):
  225.                     self.write_blist (blist_location, blist)
  226.                     return
  227.             if self.ticket.bury (ticket):
  228.                 self.write_blist (blist_location, blist)
  229.                 return
  230.         if self.ticket.forget (entry, ticket):
  231.             callback (entry, None, None)
  232.         self.write_blist (blist_location, blist)
  233.  
  234.     def read_blist (self, blist_location):
  235.         if os.path.exists (blist_location):
  236.             return [line.strip () for line in file (blist_location)]
  237.         else:
  238.             return []
  239.  
  240.     def write_blist (self, blist_location, blist):
  241.         if blist:
  242.             blist_file = file (blist_location, 'w')
  243.             blist_file.writelines (blist)
  244.             blist_file.close ()
  245.         elif os.path.exists (blist_location):
  246.             os.unlink (blist_location)
  247.  
  248.     def image_data_load (self, data):
  249.         if data and len (data) >= 1000:
  250.             pbl = gtk.gdk.PixbufLoader ()
  251.             try:
  252.                 if pbl.write (data) and pbl.close ():
  253.                     return pbl.get_pixbuf ()
  254.             except gobject.GError:
  255.                 pass
  256.         return None
  257.