diff scripts/comment_parser.py @ 1030:057c6b039f4e

Comment and rating scripts: robust against audioholders without id or audioelements; fixes issue where plots were not cleared before adding new content; showing legend optional; robust against non-UTF8 characters in legend names
author Brecht De Man <BrechtDeMan@users.noreply.github.com>
date Mon, 15 Jun 2015 13:09:46 +0100
parents 4a0bfa7bef24
children ab79d76c6f85
line wrap: on
line diff
--- a/scripts/comment_parser.py	Fri Jun 12 12:01:14 2015 +0100
+++ b/scripts/comment_parser.py	Mon Jun 15 13:09:46 2015 +0100
@@ -11,6 +11,9 @@
         # get list of all page names
         for audioholder in root.findall("./audioholder"):   # iterate over pages
             page_name = audioholder.get('id')               # get page name
+            
+            if page_name is None: # ignore 'empty' audio_holders
+                break
 
             # create folder [page_name] if not yet created
             if not os.path.exists(page_name):
@@ -18,21 +21,26 @@
 
             # for page [page_name], print comments related to fragment [id]
             for audioelement in root.findall("*/[@id='"+page_name+"']/audioelement"):
-                audio_id = str(audioelement.get('id'))
-                # append to file [page_name]/[page_name]-comments-[id].csv
-                with open(page_name+'/'+page_name+'-comments-'+audio_id+'.csv', 'a') as csvfile:
-                    commentstr = root.find("*/[@id='"
-                                           + page_name
-                                           + "']/audioelement/[@id='"
-                                           + audio_id
-                                           + "']/comment/response").text
-                    writer = csv.writer(csvfile, delimiter=',')
-                    writer.writerow([commentstr.encode("utf-8")])
-                    #TODO Comma doesn't act as delimiter now!
-                    # (when adding more than just a comment per line):
-                    # writer.writerow([file + ',' + commentstr.encode("utf-8")])
+                if audioelement is not None: # Check it exists
+                    audio_id = str(audioelement.get('id'))
 
-                    #TODO Replace 'new line' with something else?
+                    # append to file [page_name]/[page_name]-comments-[id].csv
+                    with open(page_name+'/'+page_name+'-comments-'+audio_id+'.csv', 'a') as csvfile:
+                        writer = csv.writer(csvfile, delimiter=',')
+                        commentstr = root.find("*/[@id='"
+                                               + page_name
+                                               + "']/audioelement/[@id='"
+                                               + audio_id
+                                               + "']/comment/response").text
+                        if commentstr is None:
+                            writer.writerow([''])
+                        else:
+                            writer.writerow([commentstr.encode("utf-8")])
+                        #TODO Comma doesn't act as delimiter now!
+                        # (when adding more than just a comment per line):
+                        # writer.writerow([file + ',' + commentstr.encode("utf-8")])
 
-                    #TODO 'Append' means duplicate entries if run several times...
+                        #TODO Replace 'new line' with something else?
 
+                        #TODO 'Append' means duplicate entries if run several times...
+