From aa32c196835ec257ffbabd06881f79f9d5279f02 Mon Sep 17 00:00:00 2001 From: Guillaume Pellerin Date: Tue, 13 Jan 2015 01:22:05 +0100 Subject: [PATCH] fix new csv keys --- telemeta/views/playlist.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/telemeta/views/playlist.py b/telemeta/views/playlist.py index 17c86ad3..0068e0df 100644 --- a/telemeta/views/playlist.py +++ b/telemeta/views/playlist.py @@ -111,9 +111,9 @@ class PlaylistView(object): elements.append(collection) if elements: - element = elements[0].to_dict_with_more() - tags = element.keys() - for e in elements: + tags = [] + element_dicts = [e.to_dict_with_more() for e in elements] + for e in element_dicts: for key in e.keys(): if not key in tags: tags.append(key) @@ -125,11 +125,13 @@ class PlaylistView(object): tags.insert(0, 'code') writer.writerow(tags) - for element in elements: + for element in element_dicts: data = [] - element = element.to_dict_with_more() for tag in tags: - data.append(element[tag]) + if tag in element.keys(): + data.append(element[tag]) + else: + data.append('') writer.writerow(data) return response -- 2.39.5