Test code for superfly to hack.

bzr-revno: 405

xml1.py

@@ -0,0 +1,86 @@
+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("song")
+        self.song_xml.appendChild(self.song)
+        self.song.setAttribute("version", "1.0")
+        
+    def add_lyrics_to_song(self):
+        # Create the main <lyrics> element
+        self.lyrics = self.song_xml.createElement("lyrics")
+        self.lyrics.setAttribute("language", "en")
+        self.song.appendChild(self.lyrics)
+        
+    def add_verse_to_lyrics(self, type, number, content): 
+        """
+        type - type of verse (Chorus, Verse , Bridge, Custom etc
+        number - number of item eg verse 1
+        content - the text to be stored
+        """
+        verse = self.song_xml.createElement("verse")
+        verse.setAttribute("type", type)
+        verse.setAttribute('label', number)        
+        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="  ")
+        
+    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 == '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
+    
+    
+