~thelama/+junk/do-plugins

##

## Extract pickled data from the QuodLibet songs database

## to an easy to parse XML file.

##

## By Mathieu Cadet <mathieu cadet gmail>

##

 

import os,sys,cPickle

import codecs

 

sys.path += ["/usr/local/share/quodlibet",

             "/usr/share/quodlibet"]

 

def quit(msg=None):

    if msg:

        print msg

    sys.exit (1)

 

user = os.environ.get("USER")

home = os.environ.get("HOME")

if not user or not home:

    quit("USER and/or HOME environnement variable not defined")

 

try:

    f = open("%s/.quodlibet/songs" % home)

    try:

        d = cPickle.load(f)

    finally:

        f.close()

except IOError:

    print "Unable to load QuodLibet database."

    raise

 

try:

    xml = codecs.open("/tmp/quodlibet-songs-%s" % user, "wt","utf-8")

    try:

        xml.write("<songs>\n")

 

        for s in d:

            if len(s.get("title", '')) > 0 and \

            len(s.get("artist", '')) > 0 or len(s.get("album", '')) > 0:

                xml.write("<song>\n")

                xml.write("<title><![CDATA[%s]]></title>\n" % s.get("title"))

                xml.write("<artist><![CDATA[%s]]></artist>\n" % s.get("artist", "Unknown artist").replace("\n", ","))

                xml.write("<album><![CDATA[%s]]></album>\n" % s.get("album", ""))

                xml.write("<filename><![CDATA[%s]]></filename>\n" % s.get("~filename","").decode("utf-8", 'replace'))

                xml.write("</song>\n")

 

        xml.write("</songs>\n")

    finally:

        xml.close()

except IOError:

    print "Unable to write the XML song database."

    raise