annotate scripts/comment_parser.py @ 923:7a7a72880996

Merge from the default branch
author Nicholas Jillings <n.g.r.jillings@se14.qmul.ac.uk>
date Tue, 02 Jun 2015 11:24:23 +0100
parents 533d51508e93
children 070a90d1117c 4a0bfa7bef24
rev   line source
n@921 1 import xml.etree.ElementTree as ET
n@921 2 import os
n@921 3 import csv
n@921 4
n@921 5 # get every XML file in folder
n@921 6 for file in os.listdir("."): # You have to put this script in folder where output XML files are.
n@921 7 if file.endswith(".xml"):
n@921 8 tree = ET.parse(file)
n@921 9 root = tree.getroot()
n@921 10
n@921 11 # get list of all page names
n@921 12 for audioholder in root.findall("./audioholder"): # iterate over pages
n@921 13 page_name = audioholder.get('id') # get page name
n@921 14
n@921 15 # create folder [page_name] if not yet created
n@921 16 if not os.path.exists(page_name):
n@921 17 os.makedirs(page_name)
n@921 18
n@921 19 # for page [page_name], print comments related to fragment [id]
n@921 20 for audioelement in root.findall("*/[@id='"+page_name+"']/audioelement"):
n@921 21 audio_id = str(audioelement.get('id'))
n@921 22 # append to file [page_name]/[page_name]-comments-[id].csv
n@921 23 with open(page_name+'/'+page_name+'-comments-'+audio_id+'.csv', 'a') as csvfile:
n@921 24 commentstr = root.find("*/[@id='"
n@921 25 + page_name
n@921 26 + "']/audioelement/[@id='"
n@921 27 + audio_id
n@921 28 + "']/comment/response").text
n@921 29 writer = csv.writer(csvfile, delimiter=',')
n@921 30 writer.writerow([commentstr.encode("utf-8")])
n@921 31 #TODO Comma doesn't act as delimiter now!
n@921 32 # (when adding more than just a comment per line):
n@921 33 # writer.writerow([file + ',' + commentstr.encode("utf-8")])
n@921 34
n@921 35 #TODO Replace 'new line' with something else?
n@921 36
n@921 37 #TODO 'Append' means duplicate entries if run several times...
n@921 38