fix: If array passed to add_attrib, add each individually

pull/92/head
Hannah Ward 2017-06-16 12:25:27 +01:00
parent 72a484ca32
commit 64f1edd5b4
No known key found for this signature in database
GPG Key ID: 6F3BAD60DE190290
1 changed files with 7 additions and 3 deletions

View File

@ -593,7 +593,11 @@ class MISPEvent(object):
if not found:
raise Exception('No attribute with UUID/ID {} found.'.format(attribute_id))
def add_attribute(self, type, value, **kwargs):
def add_attribute(self, type_, value, **kwargs):
attribute = MISPAttribute(self.describe_types)
attribute.set_all_values(type=type, value=value, **kwargs)
self.attributes.append(attribute)
if isinstance(value, list):
for a in value:
self.add_attribute(type_, a, **kwargs)
else:
attribute.set_all_values(type=type_, value=value, **kwargs)
self.attributes.append(attribute)