Plugins: rhythmdbloader.py

#!/usr/bin/python
# RhythmDB Loader by Thomas Folz-Donahue
# This is GPL free software

import sys, string
from xml.dom import minidom
import gtk
from library import library
import config
import util.uri
from plugins.songsmenu import SongsMenuPlugin
from functools import partial
import urllib
import os.path

try: config.get("plugins", __name__)
except: config.set("plugins", __name__, False)


def passfun(key, value, name=None, pre=False, kind="string"):
    if name != None:
        key = name
    elif pre:
        key = "~#"+key.encode('ascii')
    else:
        key = key.encode('ascii')
    if kind=="int":
        value = int(value)
    elif kind=="float":
        value = float(value)
    return (key,value)

song_import = {u'album':passfun,
               u'rating':partial(passfun,pre=True,kind="float"),
               u'artist':passfun,
               u'first-seen':partial(passfun,name='~#added'),
               u'title':passfun,
               u'play-count':partial(passfun,name='~#playcount',kind="int"),
               u'last-played':partial(passfun,name='~#lastplayed', kind='int'),
               u'mtime':partial(passfun,pre=True,kind='int'),
               u'genre':passfun,
               }




class rhythmdbloader(SongsMenuPlugin):
    PLUGIN_ID = "rhythmdbloader"
    PLUGIN_NAME = _("RhythmDB Loader")
    PLUGIN_DESC = _("Import your preexisting RhythmDB database")
    PLUGIN_VERSION = "0.1"
    PLUGIN_ICON = 'gtk-open'


    @classmethod 
    def PluginPreferences(self, parent):
        overwrite_button = gtk.CheckButton(_("overwrite QuodLibet data with RhythmDB data in case of conflicts"))
        overwrite_button.connect('toggled', self.set_overwrite_mode)
        justdoit_button = gtk.Button(_("import all songs now"))
        justdoit_button.connect('pressed', self.import_all)
        vbox = gtk.VBox()
        vbox.add(overwrite_button)
        vbox.add(justdoit_button)
        return vbox

    @classmethod
    def set_overwrite_mode(self, button):
        config.set("plugins", __name__, button.get_active())

    
    @classmethod
    def import_all(self, foo):
        rhythmdb = loadrhythmdb(os.path.expanduser("~/.gnome2/rhythmbox/rhythmdb.xml"))
        for rdb_loc in rhythmdb:
            if rdb_loc.startswith("file://"):
                try:
                    ql_loc = util.uri.URI(rdb_loc).filename
                except ValueError:
                    print "bad uri- %s"%rdb_loc
                    continue
            else:
                ql_loc = rdb_loc
            if not os.path.isfile(ql_loc):
                print "%s - not a file, ignoring"%rdb_loc
                continue
            if not config.get("plugins", __name__):
                if ql_loc in library:
                    continue
            libeme = {}
            for key in rhythmdb[rdb_loc]:
                try:
                    k,v = song_import[key](key,rhythmdb[rdb_loc][key])
                    libeme[k] = v
                except KeyError:
                    continue
            library.add_filename(ql_loc)
            if ql_loc in library:
                for key in libeme:
                    library[ql_loc][key] = libeme[key]
        
def loadrhythmdb(rhythmdbfile):
    rhythmdb = minidom.parse(rhythmdbfile)
    library = {}
    entries = rhythmdb.getElementsByTagName("entry")
    for entry in entries:
        if entry.getAttribute("type") != u"song":
            continue # not supporting ignore, iradio yet
        libeme = {}
        for child in entry.childNodes:
            
            if child.nodeType != minidom.Node.ELEMENT_NODE:
                continue
            try:
                libeme[child.tagName] = child.firstChild.data
            except AttributeError:
                continue
        try:
            location = libeme[u"location"]
        except KeyError:
            continue
        del libeme[u"location"]
        library[location] = libeme
    return library