forked from openlp/openlp
84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
from xml.dom.minidom import Document
|
|
from xml.etree.ElementTree import ElementTree, XML, dump
|
|
"""
|
|
<?xml version="1.0" encoding="UTF-8"?>
|
|
<song version="1.0">
|
|
<lyrics language="en">
|
|
<verse type="chorus" label="1">
|
|
<![CDATA[ ... ]]>
|
|
</verse>
|
|
</lyrics>
|
|
</song>
|
|
|
|
"""
|
|
class SongXMLBuilder():
|
|
def __init__(self):
|
|
# Create the minidom document
|
|
self.song_xml = Document()
|
|
|
|
def new_document(self):
|
|
# Create the <song> base element
|
|
self.song = self.song_xml.createElement(u'song')
|
|
self.song_xml.appendChild(self.song)
|
|
self.song.setAttribute(u'version', u'1.0')
|
|
|
|
def add_lyrics_to_song(self):
|
|
# Create the main <lyrics> element
|
|
self.lyrics = self.song_xml.createElement(u'lyrics')
|
|
self.lyrics.setAttribute(u'language', u'en')
|
|
self.song.appendChild(self.lyrics)
|
|
|
|
def add_verse_to_lyrics(self, type, label, content):
|
|
"""
|
|
type - type of verse (Chorus, Verse , Bridge, Custom etc
|
|
label - label of item eg verse 1
|
|
content - the text to be stored
|
|
"""
|
|
verse = self.song_xml.createElement(u'verse')
|
|
verse.setAttribute(u'type', type)
|
|
verse.setAttribute(u'label', label)
|
|
self.lyrics.appendChild(verse)
|
|
|
|
# add data as a CDATA section
|
|
cds = self.song_xml.createCDATASection(content)
|
|
verse.appendChild(cds)
|
|
|
|
def dump_xml(self):
|
|
# Debugging aid to see what we have
|
|
print self.song_xml.toprettyxml(indent=u' ')
|
|
|
|
def extract_xml(self):
|
|
# Print our newly created XML
|
|
return self.song_xml.toxml()
|
|
|
|
class SongXMLParser():
|
|
def __init__(self, xml):
|
|
self.song_xml = ElementTree(element=XML(xml))
|
|
|
|
def get_verses(self):
|
|
#return a list of verse's and attributes
|
|
iter=self.song_xml.getiterator()
|
|
verse_list = []
|
|
for element in iter:
|
|
if element.tag == u'verse':
|
|
verse_list.append([element.attrib, element.text])
|
|
return verse_list
|
|
|
|
def dump_xml(self):
|
|
# Debugging aid to see what we have
|
|
print dump(self.song_xml)
|
|
|
|
if __name__ == '__main__':
|
|
sxml=SongXMLBuilder()
|
|
sxml.new_document()
|
|
sxml.add_lyrics_to_song()
|
|
sxml.add_verse_to_lyrics(u'chorus', u'1', u'The is\n is \nsome\n text')
|
|
sxml.add_verse_to_lyrics(u'verse', u'2', u'The is\n is \nmore\n text')
|
|
sxml.dump_xml()
|
|
x1 = sxml.extract_xml()
|
|
print x1
|
|
print "================================"
|
|
spra=SongXMLParser(x1)
|
|
vl = spra.get_verses()
|
|
print vl
|