From f495c23303aa4e71bbec03077e674ee9d8b3268c Mon Sep 17 00:00:00 2001 From: Koen Van Impe Date: Fri, 11 Mar 2016 14:51:38 +0100 Subject: [PATCH] Check if objectType exists in event Prevent failing when f.e. an event does not have a tag. --- examples/feed-generator/generate.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/examples/feed-generator/generate.py b/examples/feed-generator/generate.py index ad3ebba..13229b9 100755 --- a/examples/feed-generator/generate.py +++ b/examples/feed-generator/generate.py @@ -64,14 +64,15 @@ def __cleanupEventFields(event, temp): def __cleanupEventObjects(event, temp): for objectType in objectsToSave.keys(): if objectsToSave[objectType]['multiple'] is True: - for objectInstance in temp['Event'][objectType]: - tempObject = {} - for field in objectsToSave[objectType]['fields']: - if field in objectInstance.keys(): - tempObject[field] = objectInstance[field] - if objectType not in event['Event']: - event['Event'][objectType] = [] - event['Event'][objectType].append(tempObject) + if objectType in temp['Event']: + for objectInstance in temp['Event'][objectType]: + tempObject = {} + for field in objectsToSave[objectType]['fields']: + if field in objectInstance.keys(): + tempObject[field] = objectInstance[field] + if objectType not in event['Event']: + event['Event'][objectType] = [] + event['Event'][objectType].append(tempObject) else: tempObject = {} for field in objectsToSave[objectType]['fields']: