Simplfy MISPEvent.add_attribute

pull/98/head
Raphaël Vinot 2017-07-12 11:25:41 +02:00
parent 481c8e4790
commit 9fc631cc85
1 changed files with 3 additions and 3 deletions

View File

@ -593,11 +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)
if isinstance(value, list):
for a in value:
self.add_attribute(type_, a, **kwargs)
self.add_attribute(type, a, **kwargs)
else:
attribute.set_all_values(type=type_, value=value, **kwargs)
attribute.set_all_values(type=type, value=value, **kwargs)
self.attributes.append(attribute)