Das deutsche Python-Forum

Code: Alles auswählen

import sys
from lxml import etree
#import elementtree.ElementTree as etree

def walk_tree(node, level, output, key):
    
    # Keys on Level4 are for a single Treck
    if (node.tag == "key") and (level == 4):
        # Save keyTag to allocate with next stringTag
        key = node.text
    
    # We are at the bitrateTag
    if (node.tag == "integer") and (key == "Bit Rate") and (level == 4):  
        # Save Bit Rate
        output[4] = node.text
    
    #We are at a string tag
    if (node.tag == "string") and (level == 4):
        # Find out what kind of string
        output[manage_output(key)] = node.text.encode( "utf-8" )
        
    if (node.tag == "dict") and (level == 3):
        print """,output[0],"";"",output[1],"";"",output[2],"";"",output[3],"";"",output[4],"""
        for i in range(0,4):
            output[i] = ""
        
    children = node.getchildren()
    for child in children:
        walk_tree(child, level + 1,output, key)

def manage_output(out_type):
    #out=lambda *x:sys.stdout.write(" ".join(map(str,x)))

    if out_type == "Name":
        return 0
    if out_type == "Artist":
        return 1
    if out_type == "Album":
        return 2
    if out_type == "Genre":
        return 3

def test(inFileName):
    doc = etree.parse(inFileName)
    root = doc.getroot()
    #global output
    output = ["Name","Artist","Album","Genre","Bit Rate[kbit/s]"]
    key = ""
    walk_tree(root, 0, output, key)

def main():
    args = sys.argv[1:]
    if len(args) != 1:
        print 'usage: python test.py infile.xml'
        sys.exit(-1)
    test(args[0])

if __name__ == '__main__':
    main()