Stuff

#!/usr/bin/python#RhythmDB Loader, by Thomas Folz-Donahue#This is GPL free softwareimport sys, stringfrom xml.dom import minidomimport gtkfrom library import libraryimport configfrom plugins.songsmenu import SongsMenuPluginfrom functools import partialimport urllibimport os.pathtry: 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):        print "lol"        rhythmdb = loadrhythmdb(os.path.expanduser("~/.gnome2/rhythmbox/rhythmdb.xml"))        for rdb_loc in rhythmdb:            if rdb_loc[:7] == "file://":                ql_loc = urllib.unquote(rdb_loc[7:])            else:                ql_loc = rdb_loc            if not config.get("plugins", __name__):                if ql_loc in library:                    print "%s already exists"%ql_loc                    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            print "adding %s:%s to library"%(ql_loc,libeme)            library.add_filename(ql_loc)            if ql_loc in library:                for key in libeme:                    library[ql_loc][key] = libeme[key]                    #        while True:#            try:#                print input(">")#            except:#                continue# lolxmldef 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