CoverManager.py

# -*- coding: utf-8 -*-
#
# This file is part of the Rhythmbox Desktop Art plug-in
#
# Copyright © 2008 Mathias Nedrebø < mathias at nedrebo dot org >
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

#
#
# This is just a mess, dont look any further
# I only look for localy stored cover images
#
#

from gi.repository import RB
import DesktopControl
import mimetypes
import rb

from gi.repository import Gtk, GObject

from os import path, listdir
#from urllib.request import url2pathname
from desktop_rb3compat import url2pathname

IMAGE_NAMES = ['cover', 'album', 'albumart', '.folder', 'folder']
STORAGE_LOC = "~/.gnome2/rhythmbox/covers/"
RETRIES = 5

class CoverManager():
    def __init__(self, db):
        self.db = db

    def get_cover_and_song_info(self, db_entry):
        return (self.get_cover(db_entry), self.get_song_info(db_entry))

    def get_cover(self, db_entry=None):
        # Find cover in music dir
        if db_entry:
            print("dbentry")
            cover_dir = path.dirname(url2pathname(db_entry.get_playback_uri()).replace('file://', ''))
            if path.isdir(cover_dir):
                print(cover_dir)
                for f in listdir(cover_dir):
                    file_name = path.join(cover_dir, f)
                    mt = mimetypes.guess_type(file_name)[0]
                    if mt and mt.startswith('image/'):
                        if path.splitext(f)[0].lower() in IMAGE_NAMES:
                            print("spli")
                            print(file_name)
                            return file_name

            # Find cover saved by artdisplay plugin
            song_info = self.get_song_info(db_entry)
            for rb_cover_path in ('~/.gnome2/rhythmbox/covers', '~/.cache/rhythmbox/covers/'):
                for file_type in ('jpg', 'png', 'jpeg', 'gif', 'svg'):
                    cover_file = path.join(path.expanduser(rb_cover_path),
                                           '%s - %s.%s' %
                                           (song_info['artist'],
                                            song_info['album'],
                                            file_type))
                    if path.isfile(cover_file):
                        print("Found image in cache folder")
                        print(cover_file)
                        return cover_file

            key = db_entry.create_ext_db_key(RB.RhythmDBPropType.ALBUM)
            cover_db = RB.ExtDB(name='album-art')
            art_location = cover_db.lookup(key)
            
            if art_location and not isinstance(art_location, str):
                art_location = art_location[0] # hack for RB 3.2 api

            if art_location and path.exists(art_location):
                print(art_location)
                return art_location
            # Find the image from AlbumArt plugin
            #cover_art = self.db.entry_request_extra_metadata(db_entry, "rb:coverArt")

            # If image not found return
            #if cover_art==None:
            #    print "Image not found, bloody timeouts"
            #    return DesktopControl.UNKNOWN_COVER

                # Do the dirty work here
                cover_file = path.expanduser(STORAGE_LOC) + song_info['title'] + "-" + song_info['artist'] + ".jpg"
                print(cover_file)
                cover_art.save(cover_file, "jpeg", {"quality":"100"})
                print("Returning cover file")
                print(cover_file)
                return cover_file


            # No cover found
            print("no cover")
            return DesktopControl.UNKNOWN_COVER
        # Not playing
        print("not playing")
        return None

    def get_song_info(self, db_entry=None):
        song_info = {}
        if db_entry:
            song_info['title'] = db_entry.get_string( RB.RhythmDBPropType.TITLE)
            song_info['artist'] = db_entry.get_string( RB.RhythmDBPropType.ARTIST)
            song_info['album'] = db_entry.get_string( RB.RhythmDBPropType.ALBUM)
        return song_info