MISP
/
PyMISP
mirror of https://github.com/MISP/PyMISP
2
0
Fork 0
Code Issues Releases Wiki Activity

Check if objectType exists in event 

Prevent failing when f.e. an event does not have a tag.
pull/8/head
Koen Van Impe 2016-03-11 14:51:38 +01:00
parent 7bea32ea1a
commit f495c23303
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
examples/feed-generator/generate.py
View File

@ -64,14 +64,15 @@ def __cleanupEventFields(event, temp):
def __cleanupEventObjects(event, temp): def __cleanupEventObjects(event, temp):
for objectType in objectsToSave.keys(): for objectType in objectsToSave.keys():
if objectsToSave[objectType]['multiple'] is True: if objectsToSave[objectType]['multiple'] is True:
for objectInstance in temp['Event'][objectType]: if objectType in temp['Event']:
tempObject = {} for objectInstance in temp['Event'][objectType]:
for field in objectsToSave[objectType]['fields']: tempObject = {}
if field in objectInstance.keys(): for field in objectsToSave[objectType]['fields']:
tempObject[field] = objectInstance[field] if field in objectInstance.keys():
if objectType not in event['Event']: tempObject[field] = objectInstance[field]
event['Event'][objectType] = [] if objectType not in event['Event']:
event['Event'][objectType].append(tempObject) event['Event'][objectType] = []
event['Event'][objectType].append(tempObject)
else: else:
tempObject = {} tempObject = {}
for field in objectsToSave[objectType]['fields']: for field in objectsToSave[objectType]['fields']: